End-to-End Clustering Enhanced Contrastive Learning for Radiology Reports Generation

IF 5.3 3区 计算机科学 Q1 COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCE
Xinyao Liu;Junchang Xin;Qi Shen;Chuangang Li;Zhihong Huang;Zhiqiong Wang
{"title":"End-to-End Clustering Enhanced Contrastive Learning for Radiology Reports Generation","authors":"Xinyao Liu;Junchang Xin;Qi Shen;Chuangang Li;Zhihong Huang;Zhiqiong Wang","doi":"10.1109/TETCI.2024.3449876","DOIUrl":null,"url":null,"abstract":"With the rapid growth of medical imaging data, radiologists must dedicate a significant amount of time to report writing. Automated generation of radiology reports not only alleviates the heavy workload of physicians but, more importantly, can reduce mistakes and oversights caused by insufficient experience. However, due to substantial data bias in medical data, prior studies using typical cross-entropy loss in encoder-decoder architectures often result in generalized descriptions of normal tissues and may overlook crucial clinical abnormalities. Therefore, we propose a clustering enhanced contrastive learning model named CECL to generate more diverse radiology reports, and it is worth noting that our method is end-to-end trainable. Specifically, an adaptive alignment fusion encoder-decoder network (AAF) is constructed by fusing the image features and text semantic features from the transformer decoder, eliminating information redundancy across different modalities. Moreover, a label-guided contrastive learning (LCL) module is proposed. In detail, clustering is performed on the fused features using Gaussian competition. Supervised contrastive learning is conducted based on the clustering results to enhance feature representation ability. We evaluate the CECL on two widely used publicly available datasets, IU X-ray and MIMIC-CXR, using NLG and CE metrics. The experimental results demonstrate that CECL can produce fluent reports with more descriptions of anomalies, outperforming other state-of-the-art methods with higher clinical correctness.","PeriodicalId":13135,"journal":{"name":"IEEE Transactions on Emerging Topics in Computational Intelligence","volume":"9 2","pages":"1780-1794"},"PeriodicalIF":5.3000,"publicationDate":"2024-09-02","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"IEEE Transactions on Emerging Topics in Computational Intelligence","FirstCategoryId":"94","ListUrlMain":"https://ieeexplore.ieee.org/document/10663478/","RegionNum":3,"RegionCategory":"计算机科学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q1","JCRName":"COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCE","Score":null,"Total":0}
引用次数: 0

Abstract

With the rapid growth of medical imaging data, radiologists must dedicate a significant amount of time to report writing. Automated generation of radiology reports not only alleviates the heavy workload of physicians but, more importantly, can reduce mistakes and oversights caused by insufficient experience. However, due to substantial data bias in medical data, prior studies using typical cross-entropy loss in encoder-decoder architectures often result in generalized descriptions of normal tissues and may overlook crucial clinical abnormalities. Therefore, we propose a clustering enhanced contrastive learning model named CECL to generate more diverse radiology reports, and it is worth noting that our method is end-to-end trainable. Specifically, an adaptive alignment fusion encoder-decoder network (AAF) is constructed by fusing the image features and text semantic features from the transformer decoder, eliminating information redundancy across different modalities. Moreover, a label-guided contrastive learning (LCL) module is proposed. In detail, clustering is performed on the fused features using Gaussian competition. Supervised contrastive learning is conducted based on the clustering results to enhance feature representation ability. We evaluate the CECL on two widely used publicly available datasets, IU X-ray and MIMIC-CXR, using NLG and CE metrics. The experimental results demonstrate that CECL can produce fluent reports with more descriptions of anomalies, outperforming other state-of-the-art methods with higher clinical correctness.
求助全文
约1分钟内获得全文 求助全文
来源期刊
CiteScore
10.30
自引率
7.50%
发文量
147
期刊介绍: The IEEE Transactions on Emerging Topics in Computational Intelligence (TETCI) publishes original articles on emerging aspects of computational intelligence, including theory, applications, and surveys. TETCI is an electronics only publication. TETCI publishes six issues per year. Authors are encouraged to submit manuscripts in any emerging topic in computational intelligence, especially nature-inspired computing topics not covered by other IEEE Computational Intelligence Society journals. A few such illustrative examples are glial cell networks, computational neuroscience, Brain Computer Interface, ambient intelligence, non-fuzzy computing with words, artificial life, cultural learning, artificial endocrine networks, social reasoning, artificial hormone networks, computational intelligence for the IoT and Smart-X technologies.
×
引用
GB/T 7714-2015
复制
MLA
复制
APA
复制
导出至
BibTeX EndNote RefMan NoteFirst NoteExpress
×
提示
您的信息不完整,为了账户安全,请先补充。
现在去补充
×
提示
您因"违规操作"
具体请查看互助需知
我知道了
×
提示
确定
请完成安全验证×
copy
已复制链接
快去分享给好友吧!
我知道了
右上角分享
点击右上角分享
0
联系我们:info@booksci.cn Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。 Copyright © 2023 布克学术 All rights reserved.
京ICP备2023020795号-1
ghs 京公网安备 11010802042870号
Book学术文献互助
Book学术文献互助群
群 号:481959085
Book学术官方微信