arXiv:2510.16321eess.IVcs.AI2025-10NeurIPS被引 9

用时间嵌入提升MRI重建,减少伪影且不增加计算量

Time-Embedded Algorithm Unrolling for Computational MRI

  • 将迭代步数作为时间变量嵌入网络,动态调整正则化操作
  • 在fastMRI数据上实现当前最佳重建效果,加速率高达8倍
  • 适用于各类算法翻滚方法,无需额外算力开销

算法翻滚方法在求解计算磁共振成像(MRI)中的正则化最小二乘问题上表现优异。这类方法将迭代算法展开为固定步数的结构,通常交替执行基于神经网络的正则化近端算子、数据保真操作及可学习参数的辅助更新。尽管优化方法要求近端算子网络在各轮间共享,但可能导致伪影或模糊。实践中发现使用独立网络有益,却显著增加可学习参数,易过拟合。受近似消息传递(AMP)中变化阈值的近端算子与扩散模型中时间嵌入成功的启发,本文提出一种用于逆问题的时间嵌入式算法翻滚方案。具体而言,我们将向量AMP(VAMP)中的迭代依赖近端操作及后续的Onsager校正视为时间嵌入神经网络;同时,数据保真操作及其关联的Onsager校正中的标量权重也设为时间相关可学习参数。在fastMRI数据集上的大量实验表明,该方法有效降低混叠伪影并抑制噪声放大,在不同加速率下均达当前最优性能。此外,该时间嵌入策略可无缝扩展至现有算法翻滚方法,显著提升重建质量而几乎不增加计算复杂度。

原文摘要 · Abstract (English)

Algorithm unrolling methods have proven powerful for solving the regularized least squares problem in computational magnetic resonance imaging (MRI). These approaches unfold an iterative algorithm with a fixed number of iterations, typically alternating between a neural network-based proximal operator for regularization, a data fidelity operation and auxiliary updates with learnable parameters. While the connection to optimization methods dictate that the proximal operator network should be shared across unrolls, this can introduce artifacts or blurring. Heuristically, practitioners have shown that using distinct networks may be beneficial, but this significantly increases the number of learnable parameters, making it challenging to prevent overfitting. To address these shortcomings, by taking inspirations from proximal operators with varying thresholds in approximate message passing (AMP) and the success of time-embedding in diffusion models, we propose a time-embedded algorithm unrolling scheme for inverse problems. Specifically, we introduce a novel perspective on the iteration-dependent proximal operation in vector AMP (VAMP) and the subsequent Onsager correction in the context of algorithm unrolling, framing them as a time-embedded neural network. Similarly, the scalar weights in the data fidelity operation and its associated Onsager correction are cast as time-dependent learnable parameters. Our extensive experiments on the fastMRI dataset, spanning various acceleration rates and datasets, demonstrate that our method effectively reduces aliasing artifacts and mitigates noise amplification, achieving state-of-the-art performance. Furthermore, we show that our time-embedding strategy extends to existing algorithm unrolling approaches, enhancing reconstruction quality without increasing the computational complexity significantly.

MRI重建算法翻滚时间嵌入图像恢复

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