Hierarchical medical image report adversarial generation with hybrid discriminator
Junsan Zhang,Ming Cheng,Qiaoqiao Cheng,Xiuxuan Shen,Yao Wan,Jie Zhu,Mengxuan Liu
DOI: https://doi.org/10.1016/j.artmed.2024.102846
IF: 7.011
2024-05-01
Artificial Intelligence in Medicine
Abstract:BACKGROUND AND OBJECTIVES: Generating coherent reports from medical images is an important task for reducing doctors' workload. Unlike traditional image captioning tasks, the task of medical image report generation faces more challenges. Current models for generating reports from medical images often fail to characterize some abnormal findings, and some models generate reports with low quality. In this study, we propose a model to generate high-quality reports from medical images.METHODS: In this paper, we propose a model called Hybrid Discriminator Generative Adversarial Network (HDGAN), which combines Generative Adversarial Network (GAN) with Reinforcement Learning (RL). The HDGAN model consists of a generator, a one-sentence discriminator, and a one-word discriminator. Specifically, the RL reward signals are judged on the one-sentence discriminator and one-word discriminator separately. The one-sentence discriminator can better learn sentence-level structural information, while the one-word discriminator can learn word diversity information effectively.RESULTS: Our approach performs better on the IU-X-ray and COV-CTR datasets than the baseline models. For the ROUGE metric, our method outperforms the state-of-the-art model by 0.36 on the IU-X-ray, 0.06 on the MIMIC-CXR and 0.156 on the COV-CTR.CONCLUSIONS: The compositional framework we proposed can generate more accurate medical image reports at different levels.
engineering, biomedical,computer science, artificial intelligence,medical informatics