ViLU通过融合多模态文本表示,提升视觉语言模型的失败预测能力。
ViLU: Learning Vision-Language Uncertainties for Failure Prediction
- 利用视觉、预测文本与图像条件化文本表示构建不确定性感知的多模态特征
- 在ImageNet-1k、CC12M等数据集上显著优于现有失败预测方法
- 无需模型访问,适用于事后分析场景,适合部署于不可见模型
可靠的风险量化与失败预测仍是视觉语言模型的开放挑战。我们提出ViLU,一种新的视觉语言不确定性量化框架,通过整合所有任务相关的文本表示来上下文化不确定性估计。ViLU通过交叉注意力将视觉嵌入、预测文本嵌入与图像条件化文本表示融合,构建不确定性感知的多模态表示。不同于基于损失预测的传统方法,ViLU训练一个二分类器作为不确定性预测器,使用加权二元交叉熵损失区分正确与错误预测,实现对损失的无关性。特别地,该方法适用于仅能获取视觉和文本嵌入的后处理场景。在多样数据集上的大量实验表明,本方法显著优于当前最先进的失败预测方法。我们将其应用于ImageNet-1k、CC12M及LAION-400M等标准分类与图文数据集。消融研究凸显了架构与训练策略在有效不确定性量化中的关键作用。代码已公开:https://github.com/ykrmm/ViLU。
原文摘要 · Abstract (English)
Reliable Uncertainty Quantification (UQ) and failure prediction remain open challenges for Vision-Language Models (VLMs). We introduce ViLU, a new Vision-Language Uncertainty quantification framework that contextualizes uncertainty estimates by leveraging all task-relevant textual representations. ViLU constructs an uncertainty-aware multi-modal representation by integrating the visual embedding, the predicted textual embedding, and an image-conditioned textual representation via cross-attention. Unlike traditional UQ methods based on loss prediction, ViLU trains an uncertainty predictor as a binary classifier to distinguish correct from incorrect predictions using a weighted binary cross-entropy loss, making it loss-agnostic. In particular, our proposed approach is well-suited for post-hoc settings, where only vision and text embeddings are available without direct access to the model itself. Extensive experiments on diverse datasets show the significant gains of our method compared to state-of-the-art failure prediction methods. We apply our method to standard classification datasets, such as ImageNet-1k, as well as large-scale image-caption datasets like CC12M and LAION-400M. Ablation studies highlight the critical role of our architecture and training in achieving effective uncertainty quantification. Our code is publicly available and can be found here: https://github.com/ykrmm/ViLU.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。