arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.04550cs.CV
收藏

NPB-REC:一种用于欠采样MRI重建并具备不确定性估计的非参数贝叶斯深度学习方法

NPB-REC: A Non-parametric Bayesian Deep-learning Approach for Undersampled MRI Reconstruction with Uncertainty Estimation

Technion – Israel Institute of Technology(以色列理工学院)

Samah Khawaled, Moti Freiman

AI总结:

本文提出全贝叶斯框架NPB-REC,利用随机梯度Langevin动力学建模网络参数后验,在fastMRI上提升欠采样MRI重建质量、不确定性相关性及跨解剖分布泛化能力。

摘要:

从欠采样MRI数据重建高质量图像的能力,对于提高MRI时间分辨率和缩短采集时间至关重要。已有研究提出用于该任务的深度学习方法,但由于缺乏经过验证的方法来量化重建图像中的不确定性,其临床应用受到阻碍。我们提出“NPB-REC”,这是一个非参数全贝叶斯框架,用于从欠采样数据进行MRI重建并估计不确定性。我们在训练期间使用随机梯度Langevin动力学来刻画网络参数的后验分布。这使我们既能够提高重建图像的质量,又能够量化重建图像中的不确定性。我们在来自fastMRI挑战赛的多线圈MRI数据集上证明了本方法的有效性,并将其与基线端到端变分网络(End-to-End Variational Network,E2E-VarNet)进行比较。在重建准确性方面,本方法通过PSNR和SSIM衡量优于基线(34.55、0.908 对比 33.08、0.897,p<0.01,加速率R=8),并且提供与重建误差相关性更好的不确定性度量(Pearson相关,R=0.94 对比 R=0.91)。此外,本方法对解剖分布偏移表现出更好的泛化能力(PSNR和SSIM为32.38、0.849 对比 31.63、0.836,p<0.01;在脑部数据上训练、在膝部数据上推理,加速率R=8)。NPB-REC有潜力促进基于深度学习的方法安全用于欠采样数据的MRI重建。代码和训练后的模型可在 https://github.com/samahkh/NPB-REC 获取。

英文摘要:

The ability to reconstruct high-quality images from undersampled MRI data is vital in improving MRI temporal resolution and reducing acquisition times. Deep learning methods have been proposed for this task, but the lack of verified methods to quantify the uncertainty in the reconstructed images hampered clinical applicability. We introduce "NPB-REC", a non-parametric fully Bayesian framework, for MRI reconstruction from undersampled data with uncertainty estimation. We use Stochastic Gradient Langevin Dynamics during training to characterize the posterior distribution of the network parameters. This enables us to both improve the quality of the reconstructed images and quantify the uncertainty in the reconstructed images. We demonstrate the efficacy of our approach on a multi-coil MRI dataset from the fastMRI challenge and compare it to the baseline End-to-End Variational Network (E2E-VarNet). Our approach outperforms the baseline in terms of reconstruction accuracy by means of PSNR and SSIM ($34.55$, $0.908$ vs. $33.08$, $0.897$, $p<0.01$, acceleration rate $R=8$) and provides uncertainty measures that correlate better with the reconstruction error (Pearson correlation, $R=0.94$ vs. $R=0.91$). Additionally, our approach exhibits better generalization capabilities against anatomical distribution shifts (PSNR and SSIM of $32.38$, $0.849$ vs. $31.63$, $0.836$, $p<0.01$, training on brain data, inference on knee data, acceleration rate $R=8$). NPB-REC has the potential to facilitate the safe utilization of deep learning-based methods for MRI reconstruction from undersampled data. Code and trained models are available at \url{https://github.com/samahkh/NPB-REC}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑