Benchmarking Scalable Epistemic Uncertainty Quantification in Organ Segmentation

被引:4
作者
Adams, Jadie [1 ,2 ]
Elhabian, Shireen Y. [1 ,2 ]
机构
[1] Univ Utah, Sci Comp & Imaging Inst, Salt Lake City, UT 84132 USA
[2] Univ Utah, Sch Comp, Salt Lake City, UT USA
来源
UNCERTAINTY FOR SAFE UTILIZATION OF MACHINE LEARNING IN MEDICAL IMAGING, UNSURE 2023 | 2023年 / 14291卷
基金
美国国家卫生研究院;
关键词
D O I
10.1007/978-3-031-44336-7_6
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning based methods for automatic organ segmentation have shown promise in aiding diagnosis and treatment planning. However, quantifying and understanding the uncertainty associated with model predictions is crucial in critical clinical applications. While many techniques have been proposed for epistemic or model-based uncertainty estimation, it is unclear which method is preferred in the medical image analysis setting. This paper presents a comprehensive benchmarking study that evaluates epistemic uncertainty quantification methods in organ segmentation in terms of accuracy, uncertainty calibration, and scalability. We provide a comprehensive discussion of the strengths, weaknesses, and out-of-distribution detection capabilities of each method as well as recommendations for future improvements. These findings contribute to the development of reliable and robust models that yield accurate segmentations while effectively quantifying epistemic uncertainty.
引用
收藏
页码:53 / 63
页数:11
相关论文
共 36 条
  • [11] Gal Y, 2016, PR MACH LEARN RES, V48
  • [12] Hendrycks D, 2018, Arxiv, DOI [arXiv:1610.02136, 10.48550/arXiv.1610.02136, DOI 10.48550/ARXIV.1610.02136]
  • [13] The MBPEP: a deep ensemble pruning algorithm providing high quality uncertainty prediction
    Hu, Ruihan
    Huang, Qijun
    Chang, Sheng
    Wang, Hao
    He, Jin
    [J]. APPLIED INTELLIGENCE, 2019, 49 (08) : 2942 - 2955
  • [14] Aleatoric and epistemic uncertainty in machine learning: an introduction to concepts and methods
    Huellermeier, Eyke
    Waegeman, Willem
    [J]. MACHINE LEARNING, 2021, 110 (03) : 457 - 506
  • [15] Izmailov P., 2020, Advances in Neural Information Processing Systems
  • [16] Izmailov P, 2019, Arxiv, DOI arXiv:1803.05407
  • [17] Kendall Alex, 2017, Advances in Neural Information Processing Systems, V30
  • [18] Kerfoot E., 2019, Revised Selected Papers, V11395, P371, DOI [10.1007/978-3-030-12029-040, DOI 10.1007/978-3-030-12029-040]
  • [19] King DB, 2015, ACS SYM SER, V1214, P1
  • [20] Simpson AL, 2019, Arxiv, DOI arXiv:1902.09063