arXiv:2506.00805cs.CVcs.CL2025-06ACL被引 3

用自对比奖励提升医疗视觉语言模型对齐精度,仅需2000条数据。

HSCR: Hierarchical Self-Contrastive Rewarding for Aligning Medical Vision Language Models

  • 利用模型自身生成低质量回答,通过日志概率变化识别错位词元。
  • 生成高质量反向数据,使模型在2000条样本下实现零样本性能提升。
  • 多层级优化捕捉细微语义差异,适合临床可信度要求高的场景。

医疗视觉语言模型(Med-VLMs)已在多种任务中取得成功,但现有方法常忽视模态错位问题,导致临床响应不可靠。本文提出分层自对比奖励(HSCR),解决两个关键挑战:1)低成本生成高质量偏好数据;2)捕捉精细且上下文感知的偏好以提升对齐效果。HSCR首先利用Med-VLMs在采样时更易生成不理想响应的特性,通过分析视觉标记删除后的输出概率偏移,识别引发错位的模态耦合标记,并推导出隐式对齐奖励函数。该函数指导解码过程中用幻觉标记替换这些标记,生成高质量的反向数据。此外,HSCR引入多层级偏好优化策略,超越传统相邻层级优化,通过利用反向数据中的相对质量信息,捕捉细微对齐线索,实现更精确、上下文敏感的优化。在多个医学任务(包括Med-VQA、医学图像描述生成和指令遵循)上的大量实验表明,HSCR不仅显著提升零样本性能,还在仅使用2,000条训练样本的情况下大幅改善模态对齐与可信度。

原文摘要 · Abstract (English)

Medical Vision-Language Models (Med-VLMs) have achieved success across various tasks, yet most existing methods overlook the modality misalignment issue that can lead to untrustworthy responses in clinical settings. In this paper, we propose Hierarchical Self-Contrastive Rewarding (HSCR), a novel approach that addresses two critical challenges in Med-VLM alignment: 1) Cost-effective generation of high-quality preference data; 2) Capturing nuanced and context-aware preferences for improved alignment. HSCR first leverages the inherent capability of Med-VLMs to generate dispreferred responses with higher sampling probability. By analyzing output logit shifts after visual token dropout, we identify modality-coupled tokens that induce misalignment and derive an implicit alignment reward function. This function guides token replacement with hallucinated ones during decoding, producing high-quality dispreferred data. Furthermore, HSCR introduces a multi-level preference optimization strategy, which extends beyond traditional adjacent-level optimization by incorporating nuanced implicit preferences, leveraging relative quality in dispreferred data to capture subtle alignment cues for more precise and context-aware optimization. Extensive experiments across multiple medical tasks, including Med-VQA, medical image captioning and instruction following, demonstrate that HSCR not only enhances zero-shot performance but also significantly improves modality alignment and trustworthiness with just 2,000 training entries.

医疗AI视觉语言模型对齐优化奖励机制

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