Beyond N-grams: A Hierarchical Reward Learning Framework for Clinically-Aware Medical Report Generation

Authors

  • Yuan Wang Zhejiang University
  • Shujian Gao Fudan University
  • Jiaxiang Liu Zhejiang University;Guangdong Institute of Intelligence Science and Technology
  • Songtao Jiang Zhejiang University
  • Xia Haoxiang Zhejiang University
  • Xiaotian Zhang Zhejiang University
  • Zhaolu Kang Peking University
  • Yemin Wang Zhejiang University
  • Zuozhu Liu Zhejiang University

DOI:

https://doi.org/10.1609/aaai.v40i40.40662

Abstract

Automatic medical report generation can greatly reduce the workload of doctors, but it is often unreliable for real-world deployment. Current methods can write formally fluent sentences but may be factually flawed, introducing serious medical errors known as clinical hallucinations, which make them untrustworthy for diagnosis. To bridge this gap, we introduce HiMed-RL, a Hierarchical Medical Reward Learning Framework designed to explicitly prioritize clinical quality. HiMed-RL moves beyond simple text matching by deconstructing reward learning into three synergistic levels: it first ensures linguistic fluency at the token-level, then enforces factual grounding at the concept-level by aligning key medical terms with expert knowledge, and finally assesses high-level diagnostic consistency at the semantic-level using a specialized LLM verifier. This hierarchical reward is implemented via a Human-inspired Dynamic Reward Adjustment, a strategy which first teaches the model to learn basic facts before progressing to more complex diagnostic reasoning. Experimentally, HiMed-3B achieves state-of-the-art performance on both in-domain and out-of-domain benchmarks, particularly on the latter, with an improvement of 10.8% over the second-best baseline. Our work provides a robust paradigm for generating reports that not only improve fluency but clinical fine-grained quality.

Downloads

Published

2026-03-14

How to Cite

Wang, Y., Gao, S., Liu, J., Jiang, S., Haoxiang, X., Zhang, X., … Liu, Z. (2026). Beyond N-grams: A Hierarchical Reward Learning Framework for Clinically-Aware Medical Report Generation. Proceedings of the AAAI Conference on Artificial Intelligence, 40(40), 33719–33727. https://doi.org/10.1609/aaai.v40i40.40662

Issue

Section

AAAI Technical Track on Natural Language Processing V