Estimating Uncertainty in Deep Learning MRI Reconstruction using a Pixel Classification Image Reconstruction Framework
Kamlesh Pawar1,2, Gary F Egan1,2,3, and Zhaolin Chen1,4
1Monash Biomedical Imaging, Monash University, Melbourne, Australia, 2School of Psychological Sciences, Monash University, Melbourne, Australia, 3ARC Centre of Excellence for Integrative Brain Function, Monash University, Melbourne, Australia, 4Department of Electrical and Computer Systems Engineering, Monash University, Melbourne, Australia
Image reconstruction
was modeled as a classification problem and the predicted probability of the reconstructed
pixel was used to estimate the uncertainty map. The predicted uncertainties correlate
with the actual errors, providing a tool for risk assessment of DL image
reconstruction.
Figure 1: A image obtained from the undersampled k-space data was used as input and a fully
sampled image quantized to 8-bit (256 grey levels) was used as target image to
the network for training. The DL Unet network classifies each pixel in the reconstructed image and the output of the network was probability for each pixel belonging to one
of the 256 distinct quantized grey levels. A weighted linear combination of the
predicted probability forms the reconstructed image and standard deviation of a
Gaussian fitted curve to the predicted probability distribution from the uncertainty
maps.
Figure 2: (a)
Input image from undersampled (factor of 4) k-space data; (b, d) are the
predicted probabilities at spatial location as pointed out by the red, blue and
green dots in the reconstructed image respectively; (c) Reconstructed image
obtained using weighted linear combination of the predicted probabilities; (e)
fully sampled reference image; ; (f) error image obtained by subtracting
reference and output image; (c) Uncertainty maps obtained from standard
deviation of the fitted Gaussian curve to the predicted probabilities.