Enhancing Reinforcement Learning for Radiology Report Generation with Evidence-aware Rewards and Self-correcting Preference Learning
Authors: Qin Zhou, Guoyan Liang, Qianyi Yang, Jingyuan Chen, Sai Wu, Chang Yao, Zhe Wang
Organizations: Department of Computer Science and Engineering, ECUST, China · Key Laboratory of Smart Manufacturing in Energy Chemical Process, Ministry of Education, P. R. China · Zhejiang University, Hangzhou, China · Hangzhou High-Tech Zone (Binjiang) Institute of Blockchain and Data Security
Recent reinforcement learning (RL) approaches have advanced radiology report generation (RRG), yet two core limitations persist: (1) report-level rewards offer limited evidence-grounded guidance for clinical faithfulness; and (2) current methods lack an explicit self-improving mechanism to align with clinical preference. We introduce clinically aligned Evidence-aware Self-Correcting Reinforcement Learning (ESC-RL), comprising two key components. First, a Group-wise Evidence-aware Alignment Reward (GEAR) delivers group-wise, evidence-aware feedback. GEAR reinforces consistent grounding for true positives, recovers missed findings for false negatives, and suppresses unsupported content for false positives. Second, a Self-correcting Preference Learning (SPL) strategy automatically constructs a reliable, disease-aware preference dataset from multiple noisy observations and leverages an LLM to synthesize refined reports without human supervision. ESC-RL promotes clinically faithful, disease-aligned reward and supports continual self-improvement during training. Extensive experiments on two public chest X-ray datasets demonstrate consistent gains and state-of-the-art performance.
Figures & tables
Figure 1: Overview of the proposed Evidence-aware Self-Correcting Reinforcement Learning (ESC-RL) framework.
Figure 2: Illustration of the Self-correcting Preference Learning (SPL) module.
Model
Year
Lexical Metrics
Radiology Metrics
BLEU-1 ↑
BLEU-4 ↑
ROUGE ↑
BERTScore ↑
RadCliQ ↓
RadGraphF1 ↑
ChexbertF1 ↑
GREEN ↑
R2Gen
ACL 2020
0.353
0.103
0.277
0.886
2.89
0.195
0.276
0.306
R2GenCMN
ACL 2021
0.353
0.106
0.278
0.867
2.87
0.199
0.278
0.308
RGRG
CVPR 2023
0.373
0.126
0.264
0.873
2.85
0.221
0.447
0.313
MiniGPT-Med
-
0.191
0.012
-
0.636
2.95
0.164
0.172
0.211
PromptMRG
AAAI 2024
0.398
0.112
0.268
0.857
2.77
0.227
0.476
0.289
Table 1: Comparison with existing RRG methods on the MIMIC-CXR dataset. The best and second-best results are highlighted in bold and blue , respectively. ‘-’ indicates not reported.
Model
Year
Lexical Metrics
Radiology Metrics
BLEU-1 ↑
BLEU-4 ↑
ROUGE ↑
BERTScore ↑
RadCliQ ↓
RadGraphF1 ↑
CheXbertF1 ↑
GREEN ↑
R2Gen
ACL 2020
0.289
0.052
0.243
0.861
2.79
0.187
0.145
0.482
R2GenCMN
ACL 2021
0.289
0.055
0.246
0.864
2.78
0.190
0.147
0.483
RGRG
CVPR 2023
0.266
0.063
0.180
0.867
2.71
0.189
0.180
0.481
PromptMRG
AAAI 2024
0.401
0.098
0.281
0.871
2.60
0.274
0.211
0.457
MedVersa
-
0.247
0.047
-
0.884
2.71
0.209
0.217
0.516
Table 2: Comparison with existing RRG methods on the IU-Xray dataset. The best and second-best results are highlighted in bold and blue , respectively. ‘-’ indicates not reported.
Figure 3: Qualitative comparison of reports generated by R2GenRL, PromptMRG, REVTAF, and our method. Sentences are color-coded to indicate their corresponding descriptions. Incorrect statements are marked with × , and unmarked sentences denote descriptions not mentioned in the ground-truth report.
Model
MIMIC-CXR
IU-Xray
Precision
Recall
F1
Precision
Recall
F1
R2Gen
0.333
0.273
0.276
0.151
0.145
0.145
R2GenCMN
0.334
0.275
0.278
0.154
0.147
0.147
RGRG
0.461
0.475
0.447
0.183
0.187
0.180
PromptMRG
0.501
0.509
0.476
0.213
0.229
0.211
REVTAF
0.628
0.613
0.592
0.286
0.282
0.273
Table 3: Clinical Efficacy (CE) comparison of 14 diseases on the MIMIC-CXR and IU-Xray datasets. Best values are highlighted in bold and second-best in blue .
Setting
RL
GEAR
SPL
Lexical Metrics
Radiology Metrics
BLEU-1 ↑
BLEU-4 ↑
ROUGE ↑
BERTScore ↑
RadCliQ ↓
RadGraphF1 ↑
CheXbertF1 ↑
GREEN ↑
Baseline
0.465
0.182
0.336
0.887
2.48
0.279
0.514
0.344
(a)
✓
0.472
0.187
0.335
0.889
2.45
0.282
0.517
0.349
(b)
✓
0.471
0.191
0.334
0.891
2.42
0.285
0.518
0.358
(c)
✓
0.475
0.190
0.339
0.892
2.40
0.288
0.517
0.356
(d)
✓
✓
0.481
0.194
0.345
0.890
2.40
0.289
0.517
0.359
Table 4: Effectiveness analysis of each component on MIMIC-CXR test set.
Figure 4: An example of re-integrating multiple observations into a refined report. Sentences are color-coded to indicate their corresponding descriptions. Incorrect statements are highlighted with × , and unmarked sentences denote content that is not mentioned in the ground-truth report.
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
Model
Lexical Metrics
Radiology Metrics
BLEU-1 ↑
BLEU-4 ↑
ROUGE ↑
BERTScore ↑
RadCliQ ↓
RadGraphF1 ↑
CheXbertF1 ↑
GREEN ↑
Baseline
0.465
0.182
0.336
0.887
2.48
0.279
0.514
0.344
MedKLIP
0.485
0.199
0.349
0.897
2.39
0.301
0.521
0.392
MAVL
0.487
0.199
0.352
0.898
2.39
0.304
0.608
0.394
Appendix
Table 5: Influence of different vision–language grounding models for ESC-RL on the MIMIC-CXR test set. The best results are highlighted in bold.
Model
Lexical Metrics
Radiology Metrics
BLEU-1 ↑
BLEU-4 ↑
ROUGE ↑
BERTScore ↑
RadCliQ ↓
RadGraphF1 ↑
CheXbertF1 ↑
GREEN ↑
TP-MSE & FN-MSE
0.456
0.177
0.346
0.859
2.53
0.264
0.591
0.335
TP-IoU-based & FN-IoU-based
0.451
0.171
0.342
0.865
2.59
0.254
0.590
0.327
TP-IoU-based &FN-MSE (Ours)
0.487
0.199
0.352
0.898
2.39
0.304
0.608
0.394
Appendix
Table 6: Influence of different TP/FN/FP loss for ESC-RL on the MIMIC-CXR test set. The best results are highlighted in bold.
Figure 5: Ablation study. (a) The influence of the different reward weight γ . (b) The influence of different LLMs for report re-integration.
Model
Lexical Metrics
Radiology Metrics
BLEU-1 ↑
BLEU-4 ↑
ROUGE ↑
BERTScore ↑
RadCliQ ↓
RadGraphF1 ↑
CheXbertF1 ↑
GREEN ↑
CheXbert
0.487
0.199
0.352
0.898
2.39
0.304
0.608
0.394
Gemini-2.0-flash
0.471
0.180
0.342
0.865
2.57
0.272
0.592
0.374
GPT-5
0.485
0.193
0.350
0.886
2.40
0.296
0.605
0.391
Appendix
Table 7: Influence of different disease-status extractor on the MIMIC-CXR test set. The best results are highlighted in bold.
Model
Lexical Metrics
Radiology Metrics
BLEU-1 ↑
BLEU-4 ↑
ROUGE ↑
BERTScore ↑
RadCliQ ↓
RadGraphF1 ↑
CheXbertF1 ↑
GREEN ↑
w LLM-refinement
0.487
0.199
0.352
0.898
2.39
0.304
0.608
0.394
wo LLM-refinement
0.484
0.196
0.349
0.895
2.41
0.301
0.608
0.393
Appendix
Table 8: Influence of LLM refinement on the MIMIC-CXR test set. The best results are highlighted in bold.
Figure 6: The prompt template used for re-integrating multiple observations into a refined report within the ESC-RL framework using LLMs.