cs.CVMar 4, 2026

Rethinking the Efficiency and Effectiveness of Reinforcement Learning for Radiology Report Generation

Authors: Zilin Lu, Ruifeng Yuan, Weiwei Cao, Wanxing Chang, Zhongyu Wei, Sinuo Wang, Yong Xia, Ling Zhang, 

Organizations: DAMO Academy, Alibaba Group · Ningbo No.2 Hospital, China · Hupan Lab, 310023, China · Fudan University, China · 

Abstract

Radiologists highly desire fully automated AI for radiology report generation (R2G), yet existing approaches fall short in clinical utility. Reinforcement learning (RL) holds potential to address these shortcomings, but its adoption in this task remains underexplored. In this paper, we revisit RL in terms of data efficiency and optimization effectiveness for R2G tasks. First, we explore the impact of data quantity and quality on the performance of RL in medical contexts, revealing that data quality plays a more critical role than quantity. To this end, we propose a diagnostic diversity-based data sampling strategy that enables comparable performance with fewer samples. Second, we observe that the majority of tokens in radiology reports are template-like and diagnostically uninformative, whereas the low frequency of clinically critical tokens heightens the risk of being overlooked during optimization. To tackle this, we introduce Diagnostic Token-weighted Policy Optimization (DiTPO), which directly optimizes for clinical accuracy by using a diagnostic F1 score as the reward signal. Unlike standard RL approaches that treat all tokens equally, DiTPO explicitly models the varying importance of different tokens through rule- or gradient-based mechanisms to prioritize clinically relevant content. Extensive experiments on the MIMIC-CXR, IU-Xray, and CheXpert Plus datasets demonstrate that our framework achieves state-of-the-art (SOTA) performance while requiring substantially fewer training samples in RL. Notably, on MIMIC-CXR, our framework attains an F1 score of 0.516 using only 20% of the RL training samples.

Figures & tables

Explore similar work

CardsList
  1. Enhancing Reinforcement Learning for Radiology Report Generation with Evidence-aware Rewards and Self-correcting Preference Learning

    Apr 15, 2026Qin Zhou, Guoyan Liang, Qianyi Yang +4Radiology Report GenerationRL for Generative Models

  2. Precision Recall Controllable Radiology Report Generation via Hybrid Natural Language and Clinical Reward Learning

    Jun 19, 2026Ling Chen, Ruinan Jin, Jun Luo +6Radiology Report GenerationControllable Text Generation

  3. RL-ACRGNet: Reinforcement Learning-Based Chest Radiology Report Generation Network

    Jun 1, 2026Yogesh Kumar Meena, Saurabh Agarwal, K. V. AryaRadiology Report Generation