Trust it or not: Confidence-guided automatic radiology report generation

被引:4
|
作者
Wang, Yixin [1 ]
Lin, Zihao [2 ]
Xu, Zhe [3 ]
Dong, Haoyu [4 ]
Luo, Jie [5 ]
Tian, Jiang [6 ]
Shi, Zhongchao [6 ]
Huang, Lifu [2 ]
Zhang, Yang [7 ]
Fan, Jianping [6 ,8 ]
He, Zhiqiang [7 ]
机构
[1] Stanford Univ, Dept Bioengn, Stanford, CA USA
[2] Virginia Tech, Dept Comp Sci, Blacksburg, VA USA
[3] Chinese Univ Hong Kong, Dept Biomed Engn, Shatin, Hong Kong, Peoples R China
[4] Duke Univ, Dept Elect & Comp Engn, Durham, NC USA
[5] Harvard Med Sch, Brigham & Womens Hosp, Boston, MA USA
[6] AI Lab Lenovo Res, Beijing, Peoples R China
[7] Lenovo Ltd, Lenovo Corp Res & Dev, Beijing, Peoples R China
[8] Univ North Carolina, Dept Comp Sci, Charlotte, NC USA
关键词
Radiology report generation; Uncertainty estimation; Image captioning; UNCERTAINTY;
D O I
10.1016/j.neucom.2024.127374
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Medical imaging plays a pivotal role in diagnosis and treatment in clinical practice. Inspired by the significant progress in automatic image captioning, various deep learning (DL) -based methods have been proposed to generate radiology reports for medical images. Despite promising results, previous works overlook the uncertainties of their models and are thus unable to provide clinicians with the reliability/confidence of the generated radiology reports to assist their decision -making. In this paper, we propose a novel method to explicitly quantify both the visual uncertainty and the textual uncertainty for DL -based radiology report generation. Such multi -modal uncertainties can sufficiently capture the model's confidence degree at both the report level and sentence level, which can be further leveraged to weight the losses for more comprehensive model optimization. Experimental results have demonstrated that the proposed method for model uncertainty characterization and estimation can produce more reliable confidence scores for radiology report generation, and the modified loss function, which takes into account the uncertainties, leads to better model performance on two public radiology report datasets based on both standard evaluation metrics and experienced radiologists.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] CONFIDENCE-GUIDED LEARNING FOR BREAST DENSITY CLASSIFICATION
    Park, Sun Young
    Sargent, Dustin
    Richmond, David
    2021 IEEE 18TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2021, : 1623 - 1627
  • [2] Confidence-guided path planning for mobile sensors
    Turgut, Damla
    Kreidl, O. Patrick
    Dutta, Ayan
    Boloni, Ladislau
    IEEE CONFERENCE ON GLOBAL COMMUNICATIONS, GLOBECOM, 2023, : 5925 - 5930
  • [3] Large Deformation Registration with A Confidence-Guided Network
    Wu, Xiaolong
    Hu, Bo
    Xu, Ruikang
    Zhou, Shenglong
    Gao, Ruisheng
    Xiong, Zhiwei
    BIOMEDICAL IMAGE REGISTRATION, WBIR 2024, 2025, 15249 : 34 - 44
  • [4] Confidence-Guided Local Structure Prediction with HHfrag
    Kalev, Ivan
    Habeck, Michael
    PLOS ONE, 2013, 8 (10):
  • [5] CAME: Confidence-guided Adaptive Memory Efficient Optimization
    Luo, Yang
    Ren, Xiaozhe
    Zheng, Zangwei
    Jiang, Zhuo
    Jiang, Xin
    You, Yang
    PROCEEDINGS OF THE 61ST ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS, ACL 2023, VOL 1, 2023, : 4442 - 4453
  • [6] A Confidence-Guided Evaluation for Log Parsers Inner Quality
    Xie, Xueshuo
    Wang, Zhi
    Xiao, Xuhang
    Lu, Ye
    Huang, Shenwei
    Li, Tao
    MOBILE NETWORKS & APPLICATIONS, 2021, 26 (04): : 1638 - 1649
  • [7] Towards Confidence-guided Shape Completion for Robotic Applications
    Rosasco, Andrea
    Berti, Stefano
    Bottarel, Fabrizio
    Colledanchise, Michele
    Natale, Lorenzo
    2022 IEEE-RAS 21ST INTERNATIONAL CONFERENCE ON HUMANOID ROBOTS (HUMANOIDS), 2022, : 580 - 586
  • [8] ConfidentMix: Confidence-Guided Mixup for Learning With Noisy Labels
    Higashimoto, Ryota
    Yoshida, Soh
    Muneyasu, Mitsuji
    IEEE ACCESS, 2024, 12 : 58519 - 58531
  • [9] A Confidence-Guided Evaluation for Log Parsers Inner Quality
    Xueshuo Xie
    Zhi Wang
    Xuhang Xiao
    Ye Lu
    Shenwei Huang
    Tao Li
    Mobile Networks and Applications, 2021, 26 : 1638 - 1649
  • [10] Confidence-Guided Self Refinement for Action Prediction in Untrimmed Videos
    Hou, Jingyi
    Wu, Xinxiao
    Wang, Ruiqi
    Luo, Jiebo
    Jia, Yunde
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 : 6017 - 6031