用视觉对齐校准多模态大模型的置信度,避免盲目自信。
Calibrating Uncertainty Quantification of Multi-Modal LLMs using Grounding
- 结合文本与图像一致性,提升置信度准确性
- 在医疗问答和视觉问答任务中显著改善校准效果
- 适合需要可靠置信度的医疗、安全等高风险场景
我们提出一种针对多模态大语言模型(LLMs)的不确定性量化(UQ)校准新方法。现有先进方法依赖模型在不同设置下生成多个回答的一致性,但常在模型一致错误时仍给出高置信度,导致置信度与准确率不匹配。为此,我们引入跨模态一致性,将文本回答与视觉输入对齐,并利用对齐模型的置信度来校准整体置信度。由于对齐模型本身存在不确定性,我们采用温度缩放这一通用参数化校准技术,对对齐模型的置信度进行再校准。我们在多个多模态任务上评估,包括医学问答(Slake)和视觉问答(VQAv2),使用LLaVA-Med和LLaVA等模型。实验表明,该框架在两项任务上均实现显著改进的校准性能。
原文摘要 · Abstract (English)
We introduce a novel approach for calibrating uncertainty quantification (UQ) tailored for multi-modal large language models (LLMs). Existing state-of-the-art UQ methods rely on consistency among multiple responses generated by the LLM on an input query under diverse settings. However, these approaches often report higher confidence in scenarios where the LLM is consistently incorrect. This leads to a poorly calibrated confidence with respect to accuracy. To address this, we leverage cross-modal consistency in addition to self-consistency to improve the calibration of the multi-modal models. Specifically, we ground the textual responses to the visual inputs. The confidence from the grounding model is used to calibrate the overall confidence. Given that using a grounding model adds its own uncertainty in the pipeline, we apply temperature scaling - a widely accepted parametric calibration technique - to calibrate the grounding model's confidence in the accuracy of generated responses. We evaluate the proposed approach across multiple multi-modal tasks, such as medical question answering (Slake) and visual question answering (VQAv2), considering multi-modal models such as LLaVA-Med and LLaVA. The experiments demonstrate that the proposed framework achieves significantly improved calibration on both tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。