IEEE Trans Med Imaging. 2023 Dec;42(12):3540-3554. doi: 10.1109/TMI.2023.3293826. Epub 2023 Nov 30.
In recent times, model-driven deep learning has evolved an iterative algorithm into a cascade network by replacing the regularizer's first-order information, such as the (sub)gradient or proximal operator, with a network module. This approach offers greater explainability and predictability compared to typical data-driven networks. However, in theory, there is no assurance that a functional regularizer exists whose first-order information matches the substituted network module. This implies that the unrolled network output may not align with the regularization models. Furthermore, there are few established theories that guarantee global convergence and robustness (regularity) of unrolled networks under practical assumptions. To address this gap, we propose a safeguarded methodology for network unrolling. Specifically, for parallel MR imaging, we unroll a zeroth-order algorithm, where the network module serves as a regularizer itself, allowing the network output to be covered by a regularization model. Additionally, inspired by deep equilibrium models, we conduct the unrolled network before backpropagation to converge to a fixed point and then demonstrate that it can tightly approximate the actual MR image. We also prove that the proposed network is robust against noisy interferences if the measurement data contain noise. Finally, numerical experiments indicate that the proposed network consistently outperforms state-of-the-art MRI reconstruction methods, including traditional regularization and unrolled deep learning techniques.
近年来,模型驱动的深度学习通过用网络模块替代正则化器的一阶信息(如梯度或近端算子),将迭代算法演变为级联网络。与典型的数据驱动网络相比,这种方法提供了更大的可解释性和可预测性。然而,从理论上讲,不能保证存在一个功能正则化器,其一阶信息与替代的网络模块匹配。这意味着展开的网络输出可能与正则化模型不一致。此外,在实际假设下,很少有成熟的理论可以保证展开网络的全局收敛性和鲁棒性(正则性)。为了解决这个差距,我们提出了一种网络展开的保护方法。具体来说,对于并行磁共振成像,我们展开了一个零阶算法,其中网络模块本身就是一个正则化器,允许网络输出被正则化模型覆盖。此外,受深度平衡模型的启发,我们在反向传播之前对展开的网络进行操作,以收敛到一个固定点,然后证明它可以紧密逼近实际的磁共振图像。我们还证明,如果测量数据包含噪声,那么所提出的网络对噪声干扰具有鲁棒性。最后,数值实验表明,所提出的网络始终优于最先进的 MRI 重建方法,包括传统正则化和展开的深度学习技术。