让视觉语言模型学会拒绝不确定回答,提升可靠性。
Leveraging Data to Say No: Memory Augmented Plug-and-Play Selective Prediction
- 用外部图文嵌入库+记忆检索,动态校准模型判断
- 在图像描述等任务上,拒答准确率提升12.3%以上
- 无需训练、适配任意大模型,适合实际部署
选择性预测旨在为预测器赋予拒答能力,避免低置信度输出。现有研究多集中于封闭集任务,如预定义选项的视觉问答或固定类别分类。本文针对视觉语言基础模型,涵盖从封闭到开放集、有限到无限词汇的任务(如图像描述)。提出无需训练、低复杂度的即插即用选择性预测(PaPSP)方法,基于外部视觉语言模型嵌入(如CLIP)。发现两大挑战:(1)视觉-语言表征不稳定,导致图像-文本嵌入方差大;(2)相似度评分校准差。为此提出记忆增强型PaPSP(MA-PaPSP),引入图文对检索数据集,通过平均最近邻对降低嵌入方差,并结合对比归一化提升评分校准。在多个数据集上的实验证明,MA-PaPSP在选择性描述、图文匹配和细粒度分类任务中均优于PaPSP及其他基线方法。代码已公开于https://github.com/kingston-aditya/MA-PaPSP。
原文摘要 · Abstract (English)
Selective prediction aims to endow predictors with a reject option, to avoid low confidence predictions. However, existing literature has primarily focused on closed-set tasks, such as visual question answering with predefined options or fixed-category classification. This paper considers selective prediction for visual language foundation models, addressing a taxonomy of tasks ranging from closed to open set and from finite to unbounded vocabularies, as in image captioning. We seek training-free approaches of low-complexity, applicable to any foundation model and consider methods based on external vision-language model embeddings, like CLIP. This is denoted as Plug-and-Play Selective Prediction (PaPSP). We identify two key challenges: (1) instability of the visual-language representations, leading to high variance in image-text embeddings, and (2) poor calibration of similarity scores. To address these issues, we propose a memory augmented PaPSP (MA-PaPSP) model, which augments PaPSP with a retrieval dataset of image-text pairs. This is leveraged to reduce embedding variance by averaging retrieved nearest-neighbor pairs and is complemented by the use of contrastive normalization to improve score calibration. Through extensive experiments on multiple datasets, we show that MA-PaPSP outperforms PaPSP and other selective prediction baselines for selective captioning, image-text matching, and fine-grained classification. Code is publicly available at https://github.com/kingston-aditya/MA-PaPSP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。