Quantifying Sources of Uncertainty in Deep Learning-Based Image Reconstruction

Riccardo Barbano, Zeljko Kereta, Chen Zhang, Andreas Hauptmann, Simon Arridge, Bangti Jin

Keywords: Image Reconstruction, Uncertainty Quantification, Uncertainty Decoupling, BNNs, VI

TL;DR: We present a Bayesian deep gradient descent method for quantifying aleatoric and epistemic uncertainties

Abstract: Image reconstruction methods based on deep neural networks have shown outstanding performance, equalling or exceeding the state-of-the-art of conventional approaches, but often do not provide uncertainty information of the reconstruction. In this work we propose a scalable and efficient framework to simultaneously quantify aleatoric and epistemic uncertainties in learned iterative image reconstruction. We build on a Bayesian deep gradient descent method for quantifying epistemic uncertainty, and incorporate heteroscedastic variance of the noise to account for the aleatoric uncertainty. We show that it exhibits competitive performance with respect to conventional benchmarks for computed tomography with both sparse view and limited angle data. The estimated uncertainty captures the variability in the reconstructions, caused by the restricted measurement model, and by missing information, due to the limited angle geometry.