Calibrating ensembles for scalable uncertainty quantification in deep learning-based medical image segmentationShow others and affiliations
2023 (English)In: Computers in Biology and Medicine, ISSN 0010-4825, E-ISSN 1879-0534, Vol. 163, article id 107096Article in journal (Refereed) Published
Abstract [en]
Uncertainty quantification in automated image analysis is highly desired in many applications. Typically, machine learning models in classification or segmentation are only developed to provide binary answers; however, quantifying the uncertainty of the models can play a critical role for example in active learning or machine human interaction. Uncertainty quantification is especially difficult when using deep learning-based models, which are the state-of-the-art in many imaging applications. The current uncertainty quantification approaches do not scale well in high-dimensional real-world problems. Scalable solutions often rely on classical techniques, such as dropout, during inference or training ensembles of identical models with different random seeds to obtain a posterior distribution. In this paper, we present the following contributions. First, we show that the classical approaches fail to approximate the classification probability. Second, we propose a scalable and intuitive framework for uncertainty quantification in medical image segmentation that yields measurements that approximate the classification probability. Third, we suggest the usage of k-fold cross-validation to overcome the need for held out calibration data. Lastly, we motivate the adoption of our method in active learning, creating pseudo-labels to learn from unlabeled images and human–machine collaboration.
Place, publisher, year, edition, pages
Elsevier Ltd , 2023. Vol. 163, article id 107096
Keywords [en]
Deep learning, Segmentation, Uncertainty quantification
National Category
Computer graphics and computer vision Radiology, Nuclear Medicine and Medical Imaging
Identifiers
URN: urn:nbn:se:kth:diva-331437DOI: 10.1016/j.compbiomed.2023.107096ISI: 001028875900001PubMedID: 37302375Scopus ID: 2-s2.0-85161555981OAI: oai:DiVA.org:kth-331437DiVA, id: diva2:1781597
Note
QC 20230710
2023-07-102023-07-102025-02-01Bibliographically approved