Selective "Selective Prediction": Reducing Unnecessary Abstention in Vision-Language Reasoning

Tejas Srinivasan, Jack Hessel, Tanmay Gupta, Bill Yuchen Lin, Yejin Choi +2 more
2/23/2024
cs.CL

Official code: tejas1995/ReCoVERR (3 stars, last commit 2024-06-14).

Abstract

Selective prediction minimizes incorrect predictions from vision-language models (VLMs) by allowing them to abstain from answering when uncertain. However, when deploying a vision-language system with low tolerance for inaccurate predictions, selective prediction may be over-cautious and abstain too frequently, even on many correct predictions. We introduce ReCoVERR, an inference-time algorithm to reduce the over-abstention of a selective vision-language system without increasing the error rate of the system's predictions. When the VLM makes a low-confidence prediction, instead of abstaining ReCoVERR tries to find relevant clues in the image that provide additional evidence for the prediction. ReCoVERR uses an LLM to pose related questions to the VLM, collects high-confidence evidences, and if enough evidence confirms the prediction the system makes a prediction instead of abstaining. ReCoVERR enables three VLMs (BLIP2, InstructBLIP, and LLaVA-1.5) to answer up to 20% more questions on the VQAv2 and A-OKVQA tasks without decreasing system accuracy, thus improving overall system reliability. Our code is available at https://github.com/tejas1995/ReCoVERR.

View on arXivView PDF

Code Implementations(1)

tejas1995/ReCoVERROfficial100%
30Python, Jupyter NotebookJun 14, 20242 years ago

Similar Papers

Cite this paper

@article{srinivasan2024selective,
  title  = {Selective "Selective Prediction": Reducing Unnecessary Abstention in Vision-Language Reasoning},
  author = {Tejas Srinivasan and Jack Hessel and Tanmay Gupta and Bill Yuchen Lin and Yejin Choi and Jesse Thomason and Khyathi Raghavi Chandu},
  year   = {2024},
  eprint = {2402.15610},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.15610v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2402.15610.svg)](https://codeofpaper.com/paper/2402.15610)

Discussion