arXiv:2505.02007cs.CV2025-05ICML被引 5

提出高效计算MRI重建中像素级噪声方差的方法,解决深度学习模型忽略噪声传播的难题。

Efficient Noise Calculation in Deep Learning-based MRI Reconstructions

  • 利用雅可比矩阵近似推导出无偏估计,实现内存高效的噪声方差计算
  • 在膝关节和脑部MRI数据上,计算效率提升十倍以上,结果接近蒙特卡洛模拟
  • 适用于不同噪声水平、加速因子和采样方案,适合医学影像质量评估与模型部署

加速MRI重建涉及一个病态逆问题,采集数据中的噪声会传播到重建图像中。噪声分析对评估重建结果保真度及指导新方法设计至关重要。然而,由于分析与计算上的固有挑战,基于深度学习(DL)的重建方法常忽略噪声传播。本文提出一种理论严谨、内存高效的像素级方差计算技术,用于量化加速MRI重建中因采集噪声带来的不确定性。该方法通过近似网络雅可比矩阵来估算噪声协方差,其对角线即为像素级方差;并推导出无偏估计器,结合雅可比矩阵压缩技术高效实现。我们在膝关节和脑部MRI数据集上,对监督与非监督训练的数据驱动和物理驱动网络进行了评估。结果表明,相比蒙特卡洛模拟的实证参考值,本方法性能接近,同时计算与内存开销降低一个数量级或更多。此外,方法在不同输入噪声水平、加速因子和多样化欠采样方案下均表现稳健,凸显其广泛适用性。本工作将准确且高效的噪声分析重新确立为重建算法的核心要素,有望重塑深度学习MRI重建的评估与应用范式。代码将在论文录用后公开。

原文摘要 · Abstract (English)

Accelerated MRI reconstruction involves solving an ill-posed inverse problem where noise in acquired data propagates to the reconstructed images. Noise analyses are central to MRI reconstruction for providing an explicit measure of solution fidelity and for guiding the design and deployment of novel reconstruction methods. However, deep learning (DL)-based reconstruction methods have often overlooked noise propagation due to inherent analytical and computational challenges, despite its critical importance. This work proposes a theoretically grounded, memory-efficient technique to calculate voxel-wise variance for quantifying uncertainty due to acquisition noise in accelerated MRI reconstructions. Our approach approximates noise covariance using the DL network's Jacobian, which is intractable to calculate. To circumvent this, we derive an unbiased estimator for the diagonal of this covariance matrix (voxel-wise variance) and introduce a Jacobian sketching technique to efficiently implement it. We evaluate our method on knee and brain MRI datasets for both data- and physics-driven networks trained in supervised and unsupervised manners. Compared to empirical references obtained via Monte Carlo simulations, our technique achieves near-equivalent performance while reducing computational and memory demands by an order of magnitude or more. Furthermore, our method is robust across varying input noise levels, acceleration factors, and diverse undersampling schemes, highlighting its broad applicability. Our work reintroduces accurate and efficient noise analysis as a central tenet of reconstruction algorithms, holding promise to reshape how we evaluate and deploy DL-based MRI. Our code will be made publicly available upon acceptance.

MRI重建噪声分析深度学习雅可比矩阵

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。