arXiv:2606.27023cs.LGcs.CL2026-06被引 1

让医疗视觉问答模型说出更靠谱的不确定判断。

Just how sure are you? Improving Verbalized Uncertainty Calibration in Medical VQA

论文配图:Just how sure are you? Improving Verbalized Uncertainty Calibration in Medical VQA
图 1 · 摘自论文原文
  • 用多信号联合损失函数训练,提升模型对自身信心的准确度。
  • 在三个医疗VQA数据集上,校准误差降低60%以上,区分能力提升26%。
  • 适合需要可信推理的医疗AI场景,尤其关注模型自信程度的开发者。

将多模态大语言模型(MLLM)用于医疗视觉问答(VQA)时,模型常过度自信,而现有仅针对文本的置信度校准方法未考虑医学图像理解的多模态特性。本文提出一种基于训练的框架,通过组合式损失函数微调MLLM:包含类Brier校准项、防止置信度坍缩的锚定正则项、对比图像-文本对齐项,以及基于KL散度的模型稳定性项。对齐信号来自一个2×2因子扰动设计,交叉测试图像存在性与文本完整性,以探测模型对视觉输入与语言先验的依赖程度。最后采用顶部K KL散度正则项保护模型回答能力。在三个医疗VQA基准和两种架构(MedGemma 4B IT与Qwen2 VL 7B Instruct)上,该方法使校准误差降低60%或更多,区分能力提升26%或更多,同时保持预测准确率。平均表现优于提示、采样及训练类方法,消融实验表明损失各组件均对校准提升有关键作用。所有代码已公开。

原文摘要 · Abstract (English)

Multimodal large language models (MLLMs) applied to Medical Visual Question Answering (VQA) tend to produce overconfident outputs regardless of actual correctness, and existing verbalized confidence calibration methods, developed primarily for text only LLMs, do not account for the multimodal nature of medical image understanding. This work proposes a training based framework that finetunes MLLMs to improve their calibration using a composite loss function combining a Brier style calibration term, an anchor regularizer that prevents confidence collapse toward extreme values, a contrastive image text alignment term, and a KL based model stabilization term. The alignment signal is derived from a $2 \times 2$ factorial perturbation design that crosses image presence with text integrity, probing the reliance of the model on visual modality input versus language priors. Finally, a top K KL divergence regularizer is used to protect the answering ability of the model during finetuning. Across three Medical VQA benchmarks and two architectures (MedGemma 4B IT and Qwen2 VL 7B Instruct), our method reduces calibration error by 60% or more, and improves discrimination by 26% or more, while preserving predictive accuracy. On average across benchmarks, the technique outperforms prompting based, sampling based, and training based approaches, and ablation experiments confirm that each component of the loss function is indeed necessary for improving the calibration. All code for the experiments is publicly available.

医疗AI置信度校准多模态VQA

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。