O212 - Well-Calibrated Regression Uncertainty in Medical Imaging with Deep Learning

Max-Heinrich Laves, Sontje Ihler, Jacob F. Fast, Lüder A. Kahrs, Tobias Ortmaier

Show abstract - Show schedule - Proceedings - PDF - Reviews - Teaser

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 to use $ \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, surpassing more complex calibration methods. 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.
Hide abstract

Oral Session #1 - Uncertainty - 8:30 - 9:30 UTC-4 (Monday)
Poster Session #1 - 9:30 - 11:00 UTC-4 (Monday)
Hide schedule

Access paper channel

Oral presentation

Can't display slides, your browser doesn't support embedding PDFs. You can still download the slides:

Download slides