Benchmarking Scalable Epistemic Uncertainty Quantification in Organ Segmentation
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 26 of its 38 references that are in the map. First posted on arXiv 15 Aug 2023.
Builds on
26 of 38 references.
References · 26
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,385
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
- 2016Variational Inference: A Review for Statisticianscited by 1,472
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,209
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 913
- 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 560
- 2020Bayesian Deep Learning and a Probabilistic Perspective of Generalizationcited by 480
- 2019Deep Ensembles: A Loss Landscape Perspectivecited by 443
- 2019A large annotated medical image dataset for the development and evaluation of segmentation algorithmscited by 373
- 2020BatchEnsemble: An Alternative Approach to Efficient Ensemble and Lifelong Learningcited by 371
14 more
- 2014Stochastic Gradient Hamiltonian Monte Carlocited by 343
- 2018Aleatoric uncertainty estimation with test-time augmentation for medical image segmentation with convolutional neural networkscited by 280
- 2017Concrete Dropoutcited by 227
- 2021Trustworthy AI: From Principles to Practicescited by 207
- 2020Efficient and Scalable Bayesian Neural Nets with Rank-1 Factorscited by 171
- 2019Confidence Calibration and Predictive Uncertainty Estimation for Deep Medical Image Segmentationcited by 163
- 2020Uncertainty Quantification and Deep Ensemblescited by 139
- 2020Better Aggregation in Test-Time Augmentationcited by 122
- 2021Understanding Failures in Out-of-Distribution Detection with Deep Generative Modelscited by 89
- 2022Shifts 2.0: Extending The Dataset of Real Distributional Shiftscited by 24
- 2020Estimating Uncertainty in Neural Networks for Cardiac MRI Segmentation: A Benchmark Studycited by 19
- 2020Encoding the latent posterior of Bayesian Neural Networks for uncertainty quantificationcited by 18
- 2023Fully Bayesian VIB-DeepSSMcited by 11
- 2019The MBPEP: a deep ensemble pruning algorithm providing high quality uncertainty predictioncited by 10
Built on by
Citing papers per year: 2024: 4.
4 papers.
- 2024Comparative Benchmarking of Failure Detection Methods in Medical Image Segmentation: Unveiling the Role of Confidence Aggregationcited by 12
- 2024Weakly Supervised Bayesian Shape Modeling from Unsegmented Medical Imagescited by 4
- 2024Estimation and Analysis of Slice Propagation Uncertainty in 3D Anatomy Segmentationcited by 2
- 2024Dimensionality Reduction and Nearest Neighbors for Improving Out-of-Distribution Detection in Medical Image Segmentationcited by 1
- 2024Dimensionality Reduction and Nearest Neighbors for Improving Out-of-Distribution Detection in Medical Image Segmentation5 Aug 2024
- 2024Comparative Benchmarking of Failure Detection Methods in Medical Image Segmentation: Unveiling the Role of Confidence Aggregation5 Jun 2024
- 2024Weakly Supervised Bayesian Shape Modeling from Unsegmented Medical Images15 May 2024
- 2024Estimation and Analysis of Slice Propagation Uncertainty in 3D Anatomy Segmentation18 Mar 2024