Quantifying and Leveraging Predictive Uncertainty for Medical Image Assessment

07/08/2020
by   Florin C. Ghesu, et al.
27

The interpretation of medical images is a challenging task, often complicated by the presence of artifacts, occlusions, limited contrast and more. Most notable is the case of chest radiography, where there is a high inter-rater variability in the detection and classification of abnormalities. This is largely due to inconclusive evidence in the data or subjective definitions of disease appearance. An additional example is the classification of anatomical views based on 2D Ultrasound images. Often, the anatomical context captured in a frame is not sufficient to recognize the underlying anatomy. Current machine learning solutions for these problems are typically limited to providing probabilistic predictions, relying on the capacity of underlying models to adapt to limited information and the high degree of label noise. In practice, however, this leads to overconfident systems with poor generalization on unseen data. To account for this, we propose a system that learns not only the probabilistic estimate for classification, but also an explicit uncertainty measure which captures the confidence of the system in the predicted output. We argue that this approach is essential to account for the inherent ambiguity characteristic of medical images from different radiologic exams including computed radiography, ultrasonography and magnetic resonance imaging. In our experiments we demonstrate that sample rejection based on the predicted uncertainty can significantly improve the ROC-AUC for various tasks, e.g., by 8 of different abnormalities in chest radiographs. In addition, we show that using uncertainty-driven bootstrapping to filter the training data, one can achieve a significant increase in robustness and accuracy.

READ FULL TEXT

page 18

page 20

page 23

research
06/18/2019

Quantifying and Leveraging Classification Uncertainty for Chest Radiograph Assessment

The interpretation of chest radiographs is an essential task for the det...
research
01/04/2022

Self-supervised Learning from 100 Million Medical Images

Building accurate and robust artificial intelligence systems for medical...
research
02/11/2020

Finding novelty with uncertainty

Medical images are often used to detect and characterize pathology and d...
research
11/10/2022

Improving Uncertainty-based Out-of-Distribution Detection for Medical Image Segmentation

Deep Learning models are easily disturbed by variations in the input ima...
research
10/23/2021

"One-Shot" Reduction of Additive Artifacts in Medical Images

Medical images may contain various types of artifacts with different pat...
research
01/31/2020

Unsupervised deep clustering for predictive texture pattern discovery in medical images

Predictive marker patterns in imaging data are a means to quantify disea...
research
06/02/2022

Rashomon Capacity: A Metric for Predictive Multiplicity in Probabilistic Classification

Predictive multiplicity occurs when classification models with nearly in...

Please sign up or login with your details

Forgot password? Click here to reset