arXiv:2607.10147cs.CV2026-07

解决胸部X光报告生成中强化学习不稳定的难题。

REVA-PO: Stabilizing Reinforcement Learning for Chest X-ray Report Generation

论文配图:REVA-PO: Stabilizing Reinforcement Learning for Chest X-ray Report Generation
图 1 · 摘自论文原文
  • 动态调整惩罚权重,根据输出质量自动松紧约束。
  • 定期重置策略,使探索空间更大,提升长期训练稳定性。
  • 适合医疗文本生成、强化学习优化的研究者和开发者。

自动化胸部X光报告生成近年来受益于强化学习(RL)和大语言模型。然而,传统RL训练常因固定KL正则化和静态参考策略导致不稳定或探索受限。本文提出响应加权与验证锚定策略优化(REVA-PO),通过响应加权正则化(RER)动态调节每条输出的KL权重:高质量输出放松约束,低质量输出收紧约束;同时采用验证锚定策略重置(VAPR),定期将当前策略与最佳验证检查点对齐,清除累积的正则压力,扩大可行探索空间。为确保良好起点,采用三阶段流程:预热训练、分类器引导的监督微调和强化学习。在MIMIC-CXR和IU-Xray数据集上的大量实验表明,REVA-PO在语言质量和临床准确性上均达到新最优。其中,MIMIC-CXR上BLEU-4提升5.1%,IU-Xray上提升3.6%;CheXpert F1与RadGraph F1分别提高4.5%和12.8%。代码已公开于https://github.com/LiGuo12/REVA_PO/。

原文摘要 · Abstract (English)

Automated chest X-ray report generation has recently benefited from reinforcement learning (RL) and large language models. However, RL training often suffers from instability or limited exploration due to fixed Kullback-Leibler (KL) regularization and a static reference policy that accumulates KL pressure over time. We propose Response-Weighted and Validation-Anchored Policy Optimization (REVA-PO), a RL framework that stabilizes long-term training via Response-Weighted Regularization (RER) and Validation-Anchored Policy Reset (VAPR). RER dynamically adjusts per-response KL weights based on advantage and reference-policy entropy, relaxing constraints for high-quality responses while tightening them for low-quality ones. Complementarily, VAPR periodically synchronizes the reference and current policies to the best validation checkpoint, resetting accumulated regularization pressure to expand the viable exploration space. To ensure a robust starting point, we employ a three-stage pipeline consisting of warm-up training, classifier-guided supervised fine-tuning, and RL. Extensive evaluations on MIMIC-CXR and IU-Xray demonstrate that REVA-PO sets new state-of-the-art benchmarks in both linguistic quality and clinical accuracy. Notably, BLEU-4 improves by 5.1% on MIMIC-CXR and 3.6% on IU-Xray, while CheXpert F1 and RadGraph F1 scores increase by 4.5% and 12.8%, respectively, over prior leading methods. The code is publicly available at https://github.com/LiGuo12/REVA_PO/.

医学报告生成强化学习策略优化

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