Related papers: Uncertainty Estimation in Medical Image Denoising with Bayesian Deep Image Prior

Uncertainty Estimation in Medical Image Denoising with Bayesian Deep Image Prior

URL: http://arxiv.org/abs/2008.08837v1
Date: Thu, 20 Aug 2020 08:34:51 GMT
Title: Uncertainty Estimation in Medical Image Denoising with Bayesian Deep Image Prior
Authors: Max-Heinrich Laves and Malte T\"olle and Tobias Ortmaier
Abstract summary: Uncertainty in inverse medical imaging tasks with deep learning has received little attention. Deep models trained on large data sets tend to hallucinate and create artifacts in the reconstructed output that are not present. We use a randomly convolutional network as parameterization of the reconstructed image and perform gradient descent to match the observation, which is known as deep image prior.
Score: 2.0303656145222857
License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
Abstract: 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 output that are not anatomically present. We use a randomly initialized convolutional network as parameterization of the reconstructed image and perform gradient descent to match the observation, which is known as deep image prior. In this case, the reconstruction does not suffer from hallucinations as no prior training is performed. We extend this to a Bayesian approach with Monte Carlo dropout to quantify both aleatoric and epistemic uncertainty. The presented method is evaluated on the task of denoising different medical imaging modalities. The experimental results show that our approach yields well-calibrated uncertainty. That is, the predictive uncertainty correlates with the predictive error. This allows for reliable uncertainty estimates and can tackle the problem of hallucinations and artifacts in inverse medical imaging tasks.