Deep learning (DL) models are capable of successfully exploiting latent representations in MR data and have become state-of-the-art for accelerated MRI reconstruction. However, undersampling the measurements in k-space as well as the over- or under-parameterized and non-transparent nature of DL make these models exposed to uncertainty. Consequently, uncertainty estimation has become a major issue in DL MRI reconstruction. To estimate uncertainty, Monte Carlo (MC) inference techniques have become a common practice where multiple reconstructions are utilized to compute the variance in reconstruction as a measurement of uncertainty. However, these methods demand high computational costs as they require multiple inferences through the DL model. To this end, we introduce a method to estimate uncertainty during MRI reconstruction using a pixel classification framework. The proposed method, PixCUE (stands for Pixel Classification Uncertainty Estimation) produces the reconstructed image along with an uncertainty map during a single forward pass through the DL model. We demonstrate that this approach generates uncertainty maps that highly correlate with the reconstruction errors with respect to various MR imaging sequences and under numerous adversarial conditions. We also show that the estimated uncertainties are correlated to that of the conventional MC method. We further provide an empirical relationship between the uncertainty estimations using PixCUE and well-established reconstruction metrics such as NMSE, PSNR, and SSIM. We conclude that PixCUE is capable of reliably estimating the uncertainty in MRI reconstruction with a minimum additional computational cost.
翻译:摘要:深度学习(DL)模型能够成功利用磁共振数据中的潜在表征,已成为加速MRI重建领域的先进技术。然而,k空间测量数据的欠采样以及深度学习模型过参数化或欠参数化且不透明的特性,使这些模型面临不确定性。因此,不确定性估计已成为深度学习MRI重建中的关键问题。为估计不确定性,蒙特卡洛(MC)推理技术成为常用方法,该方法通过多次重建计算重建方差作为不确定性度量。但这些方法需要多次通过深度学习模型进行推理,计算成本高昂。为此,我们提出一种基于像素分类框架在MRI重建过程中估计不确定性的方法。所提方法PixCUE(像素分类不确定性估计)可在单次前向传播中生成重建图像及相应不确定性图。我们证明,该方法生成的不确定性图与多种MR成像序列及多种对抗条件下的重建误差高度相关。同时,该方法估计的不确定性与传统MC方法的结果具有相关性。我们还进一步建立了PixCUE不确定性估计与NMSE、PSNR、SSIM等成熟重建指标之间的经验关系。结论表明,PixCUE能以极低的额外计算成本可靠估计MRI重建中的不确定性。