Accelerated Multi-Modal MR Imaging with Transformers

Accelerated multi-modal magnetic resonance (MR) imaging is a new and effective solution for fast MR imaging, providing superior performance in restoring the target modality from its undersampled counterpart with guidance from an auxiliary modality. However, existing works simply combine the auxiliary modality as prior information, lacking in-depth investigations on the potential mechanisms for fusing different modalities. Further, they usually rely on the convolutional neural networks (CNNs), which is limited by the intrinsic locality in capturing the long-distance dependency. To this end, we propose a multimodal transformer (MTrans), which is capable of transferring multi-scale features from the target modality to the auxiliary modality, for accelerated MR imaging. To capture deep multi-modal information, our MTrans utilizes an improved multi-head attention mechanism, named cross attention module, which absorbs features from the auxiliary modality that contribute to the target modality. Our framework provides three appealing benefits: (i) Our MTrans is the first attempt at using improved transformers for multimodal MR imaging, affording more global information compared with existing CNN-based methods. (ii) A new cross attention module is proposed to exploit the useful information in each modality at different scales. It affords both distinct structural information and subtle pixel-level information, which supplement the target modality effectively. (iii) We evaluate MTrans with various accelerated multimodal MR imaging tasks, e.g., MR image reconstruction and super-resolution, where MTrans outperforms state-ofthe-art methods on fastMRI and real-world clinical datasets.

[1]  Xuanqin Mou,et al.  Spectral CT Reconstruction via Low-Rank Representation and Region-Specific Texture Preserving Markov Random Field Regularization , 2020, IEEE Transactions on Medical Imaging.

[2]  Mohammad Shahdloo,et al.  Prior-Guided Image Reconstruction for Accelerated Multi-Contrast MRI via Generative Adversarial Networks , 2020, IEEE Journal of Selected Topics in Signal Processing.

[3]  Yan Wang,et al.  TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation , 2021, ArXiv.

[4]  Mathews Jacob,et al.  MoDL: Model-Based Deep Learning Architecture for Inverse Problems , 2017, IEEE Transactions on Medical Imaging.

[5]  Daniel Rueckert,et al.  A Deep Cascade of Convolutional Neural Networks for Dynamic MR Image Reconstruction , 2017, IEEE Transactions on Medical Imaging.

[6]  Michael Unser,et al.  Deep Convolutional Neural Network for Inverse Problems in Imaging , 2016, IEEE Transactions on Image Processing.

[7]  Jaejun Yoo,et al.  Deep Residual Learning for Accelerated MRI Using Magnitude and Phase Networks , 2018, IEEE Transactions on Biomedical Engineering.

[8]  Ling Shao,et al.  Hi-Net: Hybrid-Fusion Network for Multi-Modal MR Image Synthesis , 2020, IEEE Transactions on Medical Imaging.

[9]  Shu Liao,et al.  Learning MRI k-Space Subsampling Pattern Using Progressive Weight Pruning , 2020, MICCAI.

[10]  Yong Chen,et al.  Deep-Learning-Based Multi-Modal Fusion for Fast MR Reconstruction , 2019, IEEE Transactions on Biomedical Engineering.

[11]  Zongben Xu,et al.  ADMM-CSNet: A Deep Learning Approach for Image Compressive Sensing , 2020, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[12]  Dong Liang,et al.  DeepcomplexMRI: Exploiting deep residual network for fast parallel MR imaging with complex convolution. , 2020, Magnetic resonance imaging.

[13]  Kyoung Mu Lee,et al.  Enhanced Deep Residual Networks for Single Image Super-Resolution , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[14]  Guang Yang,et al.  DAGAN: Deep De-Aliasing Generative Adversarial Networks for Fast Compressed Sensing MRI Reconstruction , 2018, IEEE Transactions on Medical Imaging.

[15]  Yu Yang,et al.  Simultaneous single- and multi-contrast super-resolution for brain MRI images based on a convolutional neural network , 2018, Comput. Biol. Medicine.

[16]  Ge Wang,et al.  Super-resolution MRI through Deep Learning , 2018, 1810.06776.

[17]  Yudong Zhang,et al.  Exponential Wavelet Iterative Shrinkage Thresholding Algorithm for compressed sensing magnetic resonance imaging , 2015, Inf. Sci..

[18]  Leslie Ying,et al.  Accelerating magnetic resonance imaging via deep learning , 2016, 2016 IEEE 13th International Symposium on Biomedical Imaging (ISBI).

[19]  Jong Chul Ye,et al.  Deep learning for tomographic image reconstruction , 2020, Nature Machine Intelligence.

[20]  Ling Shao,et al.  DONet: Dual-Octave Network for Fast MR Image Reconstruction , 2021, IEEE transactions on neural networks and learning systems.

[21]  Zhi-Pei Liang,et al.  Accelerated High-Dimensional MR Imaging With Sparse Sampling Using Low-Rank Tensors , 2016, IEEE Transactions on Medical Imaging.

[22]  Georg Heigold,et al.  An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale , 2021, ICLR.

[23]  Feng Shi,et al.  Brain MRI super resolution using 3D deep densely connected neural networks , 2018, 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018).

[24]  Hao Zheng,et al.  Cascaded Dilated Dense Network with Two-step Data Consistency for MRI Reconstruction , 2019, NeurIPS.

[25]  Di Guo,et al.  Sparse MRI reconstruction using multi-contrast image guided graph representation. , 2017, Magnetic resonance imaging.

[26]  Ling Shao,et al.  Dual-Octave Convolution for Accelerated Parallel MR Image Reconstruction , 2021, AAAI.

[27]  Nicolas Usunier,et al.  End-to-End Object Detection with Transformers , 2020, ECCV.

[28]  Jingwei Zhuo,et al.  P‐LORAKS: Low‐rank modeling of local k‐space neighborhoods with parallel imaging data , 2016, Magnetic resonance in medicine.

[29]  Morteza Mardani,et al.  Deep Generative Adversarial Neural Networks for Compressive Sensing MRI , 2019, IEEE Transactions on Medical Imaging.

[30]  M. Jacob,et al.  Deep Generalization of Structured Low-Rank Algorithms (Deep-SLR) , 2019, IEEE Transactions on Medical Imaging.

[31]  Vivek K Goyal,et al.  Multi‐contrast reconstruction with Bayesian compressed sensing , 2011, Magnetic resonance in medicine.

[32]  Justin P. Haldar,et al.  Compressed-Sensing MRI With Random Encoding , 2011, IEEE Transactions on Medical Imaging.

[33]  John W. Paisley,et al.  A Deep Information Sharing Network for Multi-Contrast Compressed Sensing MRI Reconstruction , 2018, IEEE Transactions on Image Processing.

[34]  Pascal Vincent,et al.  fastMRI: An Open Dataset and Benchmarks for Accelerated MRI , 2018, ArXiv.

[35]  Bruce R. Rosen,et al.  Image reconstruction by domain-transform manifold learning , 2017, Nature.

[36]  Yonina C. Eldar,et al.  Coupled Dictionary Learning for Multi-Contrast MRI Reconstruction , 2018, 2018 25th IEEE International Conference on Image Processing (ICIP).

[37]  Konstantinos Kamnitsas,et al.  Context-Sensitive Super-Resolution for Fast Fetal Magnetic Resonance Imaging , 2017, CMMI/RAMBO/SWITCH@MICCAI.

[38]  Fahad Shahbaz Khan,et al.  Transformers in Vision: A Survey , 2021, ACM Comput. Surv..

[39]  Hongming Shan,et al.  Multi-Contrast Super-Resolution MRI Through a Progressive Network , 2019, IEEE Transactions on Medical Imaging.

[40]  Xu Zhou,et al.  Accuracy of 3-T MRI using susceptibility-weighted imaging to detect meniscal tears of the knee , 2014, Knee Surgery, Sports Traumatology, Arthroscopy.

[41]  Aaron Defazio,et al.  GrappaNet: Combining Parallel Imaging With Deep Learning for Multi-Coil MRI Reconstruction , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[42]  Chi-Hieu Pham,et al.  Brain MRI super-resolution using deep 3D convolutional networks , 2017, 2017 IEEE 14th International Symposium on Biomedical Imaging (ISBI 2017).

[43]  Qiegen Liu,et al.  Model-based Convolutional De-Aliasing Network Learning for Parallel MR Imaging , 2019, MICCAI.

[44]  Minghui Zhang,et al.  Highly undersampled magnetic resonance imaging reconstruction using autoencoding priors , 2020, Magnetic resonance in medicine.

[45]  Won-Ki Jeong,et al.  Compressed Sensing MRI Reconstruction Using a Generative Adversarial Network With a Cyclic Loss , 2017, IEEE Transactions on Medical Imaging.

[46]  Daniel Rueckert,et al.  Super-resolution reconstruction of cardiac MRI using coupled dictionary learning , 2014, 2014 IEEE 11th International Symposium on Biomedical Imaging (ISBI).

[47]  Wenxuan Wang,et al.  TransBTS: Multimodal Brain Tumor Segmentation Using Transformer , 2021, MICCAI.