학술논문

Bayesian Uncertainty Estimation of Learned Variational MRI Reconstruction
Document Type
Periodical
Source
IEEE Transactions on Medical Imaging IEEE Trans. Med. Imaging Medical Imaging, IEEE Transactions on. 41(2):279-291 Feb, 2022
Subject
Bioengineering
Computing and Processing
Magnetic resonance imaging
Image reconstruction
Uncertainty
Bayes methods
Optimal control
Inverse problems
Estimation
undersampled MRI
total deep variation
convolutional neural network
image reconstruction
epistemic uncertainty estimation
Bayes’ theorem
Language
ISSN
0278-0062
1558-254X
Abstract
Recent deep learning approaches focus on improving quantitative scores of dedicated benchmarks, and therefore only reduce the observation-related (aleatoric) uncertainty. However, the model-immanent (epistemic) uncertainty is less frequently systematically analyzed. In this work, we introduce a Bayesian variational framework to quantify the epistemic uncertainty. To this end, we solve the linear inverse problem of undersampled MRI reconstruction in a variational setting. The associated energy functional is composed of a data fidelity term and the total deep variation (TDV) as a learned parametric regularizer. To estimate the epistemic uncertainty we draw the parameters of the TDV regularizer from a multivariate Gaussian distribution, whose mean and covariance matrix are learned in a stochastic optimal control problem. In several numerical experiments, we demonstrate that our approach yields competitive results for undersampled MRI reconstruction. Moreover, we can accurately quantify the pixelwise epistemic uncertainty, which can serve radiologists as an additional resource to visualize reconstruction reliability.