论文

MedCBR:面向医学视觉语言模型能力边界对齐的两阶段框架

Aligning Medical Vision Language Models at the Capability Boundary with Entropy Adaptive Rubric Grounding

精选理由

这篇论文针对医学影像诊断模型瞎猜答案的问题,用熵门控的强化学习把训练火力集中在模型最拿不准的病例上,六个基准涨到 61.2%。

论文提出 MedCBR,一个针对医学视觉语言模型的两阶段对齐框架。第一阶段用基于经验通过率的 DBPO,在模型能力边界处构建反事实偏好对,把学习集中在难以区分的鉴别诊断样本上。第二阶段用 EGPO 依据预测熵激活从边界轨迹提炼的临床评分标准,同时奖励答案正确性、解剖学定位和推理保真度。在六个医学基准上平均准确率达到 61.2%,消融实验显示两阶段组合的增益超过各自单独提升之和。

原文 · arXiv cs.AI

Aligning Medical Vision Language Models at the Capability Boundary with Entropy Adaptive Rubric Grounding

Medical diagnostic reasoning requires verifiable links between visual evidence and clinical knowledge, yet current medical vision language models are aligned with objectives that reward the wrong behavior. Uniform supervised fine tuning lets easy cases dominate updates while clinically critical boundary cases saturate prematurely. Sparse rewards based only on final-answer correctness cannot assess anatomical grounding or clinical logic, inviting reward hacking through answer shortcuts. We propose MedCBR, a two-stage alignment framework coupling boundary-aware preference optimization with rubric-grounded reinforcement learning. First, we construct counterfactual boundary preferences at the model's capability frontier using empirical pass rates and preference pairs that share reasoning prefixes but diverge at decisive diagnostic branches. Difficulty Weighted Boundary Preference Optimization (DBPO) concentrates learning on this calibrated boundary region. Second, we distill reusable clinical rubrics from boundary trajectories and activate them according to prediction entropy. Entropy Gated Grounded Policy Optimization (EGPO) rewards answer correctness, anatomical grounding, and reasoning fidelity while suppressing rubric noise on reliable samples. A reference policy anchored to the boundary-aligned model couples population-level boundary priors with instance-level uncertainty, making diagnostic reasoning more accurate and clinically grounded. Experiments on six medical benchmarks show consistent improvements over strong general, specialized, and reinforcement learning baselines, reaching 61.2% average accuracy with the largest gains on challenging differential diagnosis tasks. Ablations show super-additive gains: combining the two stages exceeds the sum of their individual improvements.