arXiv:2507.09279cs.CVcs.AI2025-07ICCV被引 6

用强化学习优化提示词,让医疗多模态模型的自信更贴近真实准确率。

Prompt4Trust: A Reinforcement Learning Prompt Augmentation Framework for Clinically-Aligned Confidence Calibration in Multimodal Large Language Models

  • 用轻量级LLM生成上下文感知提示,引导主模型输出可信度匹配准确率的回答。
  • 在PMC-VQA上达到当前最优医学视觉问答性能,同时显著提升置信度校准效果。
  • 小模型训练后可零样本适配大模型,适合医疗等高安全场景的低成本部署。

多模态大语言模型(MLLMs)在医疗领域有巨大应用潜力,但其在高风险场景中的应用受限于两大问题:对提示设计敏感,且常以高置信度生成错误回答。由于临床医生可能依赖模型的置信度判断结果可靠性,因此当模型表达高置信时,其准确性也应高。我们提出Prompt4Trust,首个针对MLLM置信度校准的强化学习提示增强框架。通过训练一个轻量级LLM生成上下文感知的辅助提示,引导下游任务MLLM输出置信度更贴合实际准确率的回答。与传统校准方法不同,Prompt4Trust聚焦于临床决策中至关重要的校准特性。除了由临床导向目标驱动的改进外,该方法还提升了任务准确率,在包含多种医学影像模态的多项选择题数据集PMC-VQA上达到当前最优表现。此外,使用小型下游模型训练的框架在实验中展现出对更大模型的优异零样本泛化能力,表明无需高昂计算成本即可实现可扩展校准。本工作展示了自动化且符合人类意图的提示工程在提升高危场景下MLLM可信度方面的潜力。代码库地址:https://github.com/xingbpshen/prompt4trust。

原文摘要 · Abstract (English)

Multimodal large language models (MLLMs) hold considerable promise for applications in healthcare. However, their deployment in safety-critical settings is hindered by two key limitations: (i) sensitivity to prompt design, and (ii) a tendency to generate incorrect responses with high confidence. As clinicians may rely on a model's stated confidence to gauge the reliability of its predictions, it is especially important that when a model expresses high confidence, it is also highly accurate. We introduce Prompt4Trust, the first reinforcement learning (RL) framework for prompt augmentation targeting confidence calibration in MLLMs. A lightweight LLM is trained to produce context-aware auxiliary prompts that guide a downstream task MLLM to generate responses in which the expressed confidence more accurately reflects predictive accuracy. Unlike conventional calibration techniques, Prompt4Trust specifically prioritizes aspects of calibration most critical for safe and trustworthy clinical decision-making. Beyond improvements driven by this clinically motivated calibration objective, our proposed method also improves task accuracy, achieving state-of-the-art medical visual question answering (VQA) performance on the PMC-VQA benchmark, which is composed of multiple-choice questions spanning diverse medical imaging modalities. Moreover, our framework trained with a small downstream task MLLM showed promising zero-shot generalization to larger MLLMs in our experiments, suggesting the potential for scalable calibration without the associated computational costs. This work demonstrates the potential of automated yet human-aligned prompt engineering for improving the the trustworthiness of MLLMs in safety critical settings. Our codebase can be found at https://github.com/xingbpshen/prompt4trust.

多模态医疗AI置信度校准强化学习

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