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
相关论文
共 71 条
  • [1] A review of uncertainty quantification in deep learning: Techniques, applications and challenges
    Abdar, Moloud
    Pourpanah, Farhad
    Hussain, Sadiq
    Rezazadegan, Dana
    Liu, Li
    Ghavamzadeh, Mohammad
    Fieguth, Paul
    Cao, Xiaochun
    Khosravi, Abbas
    Acharya, U. Rajendra
    Makarenkov, Vladimir
    Nahavandi, Saeid
    [J]. INFORMATION FUSION, 2021, 76 : 243 - 297
  • [2] Andersen J.S., 2022, ARXIV
  • [3] SPICE: Semantic Propositional Image Caption Evaluation
    Anderson, Peter
    Fernando, Basura
    Johnson, Mark
    Gould, Stephen
    [J]. COMPUTER VISION - ECCV 2016, PT V, 2016, 9909 : 382 - 398
  • [4] Expert-validated estimation of diagnostic uncertainty for deep neural networks in diabetic retinopathy detection
    Ayhan, Murat Seckin
    Kuhlewein, Laura
    Aliyeva, Gulnar
    Inhoffen, Werner
    Ziemssen, Focke
    Berens, Philipp
    [J]. MEDICAL IMAGE ANALYSIS, 2020, 64
  • [5] Banerjee S., 2005, P ACL WORKSH INTR EX, P65
  • [6] Uncertainty-aware domain alignment for anatomical structure segmentation
    Bian, Cheng
    Yuan, Chenglang
    Wang, Jiexiang
    Li, Meng
    Yang, Xin
    Yu, Shuang
    Ma, Kai
    Yuan, Jin
    Zheng, Yefeng
    [J]. MEDICAL IMAGE ANALYSIS, 2020, 64
  • [7] Boag W., 2020, PMLR, V116, P126
  • [8] Bojanowski P., 2017, Transactions of the Association for Computational Linguistic, V5, P135, DOI [DOI 10.1162/TACLA00051, 10.1162/tacl_a_00051]
  • [9] Reactions to uncertainty and the accuracy of diagnostic mammography
    Carney, Patricia A.
    Yi, Joyce P.
    Abraham, Linn A.
    Miglioretti, Diana L.
    Aiello, Erin J.
    Gerrity, Martha S.
    Reisch, Lisa
    Berns, Eric A.
    Sickles, Edward A.
    Elmore, Joann G.
    [J]. JOURNAL OF GENERAL INTERNAL MEDICINE, 2007, 22 (02) : 234 - 241
  • [10] Chang HS, 2021, Arxiv, DOI arXiv:2103.15330