Well-Calibrated Regression Uncertainty in Medical Imaging with Deep Learning
The consideration of predictive uncertainty in medical imaging with deep learning is of utmost importance. We apply estimation of predictive uncertainty by variational Bayesian inference with Monte Carlo dropout to regression tasks and show why predictive uncertainty is systematically underestimated. We suggest using $ \sigma $ scaling with a single scalar value; a simple, yet effective calibration method for both aleatoric and epistemic uncertainty. The performance of our approach is evaluated on a variety of common medical regression data sets using different state-of-the-art convolutional network architectures. In all experiments, $ \sigma $ scaling is able to reliably recalibrate predictive uncertainty. It is easy to implement and maintains the accuracy. Well-calibrated uncertainty in regression allows robust rejection of unreliable predictions or detection of out-of-distribution samples.
Code (2)
Tasks
Bayesian InferenceDeep LearningregressionSimilar Papers 제목 키워드 기반
Recalibration of Aleatoric and Epistemic Regression Uncertainty in Medical Imaging
The consideration of predictive uncertainty in medical imaging with deep learning is of utmost importance. We apply estimation of both aleatoric and epistemic uncertainty by variational Bayesian inference with Monte Carl…
Bayesian InferenceregressionLikelihood Annealing: Fast Calibrated Uncertainty for Regression
Recent advances in deep learning have shown that uncertainty estimation is becoming increasingly important in applications such as medical imaging, natural language processing, and autonomous systems. However, accurately…
DenoisingImage Super-ResolutionProperty Predictionregression+1How Reliable is Your Regression Model's Uncertainty Under Real-World Distribution Shifts?
Many important computer vision applications are naturally formulated as regression problems. Within medical imaging, accurate regression models have the potential to automate various tasks, helping to lower costs and imp…
Out-of-Distribution DetectionregressionUncertainty Estimation in Medical Image Denoising with Bayesian Deep Image Prior
Uncertainty quantification in inverse medical imaging tasks with deep learning has received little attention. However, deep models trained on large data sets tend to hallucinate and create artifacts in the reconstructed …
DenoisingImage DenoisingMedical Image DenoisingUncertainty QuantificationMedSymmFlow: Bridging Generative Modeling and Classification in Medical Imaging through Symmetrical Flow Matching
Reliable medical image classification requires accurate predictions and well-calibrated uncertainty estimates, especially in high-stakes clinical settings. This work presents MedSymmFlow, a generative-discriminative hybr…
Medical Image Classification