Singh Nalini M, Iglesias Juan Eugenio, Adalsteinsson Elfar, Dalca Adrian V, Golland Polina
Computer Science and Artificial Intelligence Laboratory, MIT, Cambridge, MA, USA.
Dept. of Health Sciences & Technology, MIT, Cambridge, MA, USA.
J Mach Learn Biomed Imaging. 2022 Jun;2022. Epub 2022 Jun 23.
We propose neural network layers that explicitly combine frequency and image feature representations and show that they can be used as a versatile building block for reconstruction from frequency space data. Our work is motivated by the challenges arising in MRI acquisition where the signal is a corrupted Fourier transform of the desired image. The proposed joint learning schemes enable both correction of artifacts native to the frequency space and manipulation of image space representations to reconstruct coherent image structures at every layer of the network. This is in contrast to most current deep learning approaches for image reconstruction that treat frequency and image space features separately and often operate exclusively in one of the two spaces. We demonstrate the advantages of joint convolutional learning for a variety of tasks, including motion correction, denoising, reconstruction from undersampled acquisitions, and combined undersampling and motion correction on simulated and real world multicoil MRI data. The joint models produce consistently high quality output images across all tasks and datasets. When integrated into a state of the art unrolled optimization network with physics-inspired data consistency constraints for undersampled reconstruction, the proposed architectures significantly improve the optimization landscape, which yields an order of magnitude reduction of training time. This result suggests that joint representations are particularly well suited for MRI signals in deep learning networks. Our code and pretrained models are publicly available at https://github.com/nalinimsingh/interlacer.
我们提出了明确结合频率和图像特征表示的神经网络层,并表明它们可以用作从频率空间数据进行重建的通用构建块。我们的工作受到MRI采集过程中出现的挑战的推动,在MRI采集中,信号是所需图像的损坏傅里叶变换。所提出的联合学习方案能够校正频率空间固有的伪影,并操纵图像空间表示,以在网络的每一层重建连贯的图像结构。这与当前大多数用于图像重建的深度学习方法形成对比,这些方法分别处理频率和图像空间特征,并且通常仅在两个空间之一中运行。我们展示了联合卷积学习在各种任务中的优势,包括运动校正、去噪、欠采样采集重建以及对模拟和真实世界多线圈MRI数据进行欠采样和运动校正的组合。联合模型在所有任务和数据集上都能持续产生高质量的输出图像。当将其集成到具有物理启发的数据一致性约束的最先进的展开优化网络中用于欠采样重建时,所提出的架构显著改善了优化格局,从而使训练时间减少了一个数量级。这一结果表明,联合表示在深度学习网络中特别适合MRI信号。我们的代码和预训练模型可在https://github.com/nalinimsingh/interlacer上公开获取。