CL-MRI: Self-Supervised Contrastive Learning to Improve the Accuracy of Undersampled MRI Reconstruction

Mevan Ekanayake,Zhifeng Chen,Mehrtash Harandi,Gary Egan,Zhaolin Chen

2024-05-30

Abstract:In Magnetic Resonance Imaging (MRI), image acquisitions are often undersampled in the measurement domain to accelerate the scanning process, at the expense of image quality. However, image quality is a crucial factor that influences the accuracy of clinical diagnosis; hence, high-quality image reconstruction from undersampled measurements has been a key area of research. Recently, deep learning (DL) methods have emerged as the state-of-the-art for MRI reconstruction, typically involving deep neural networks to transform undersampled MRI images into high-quality MRI images through data-driven processes. Nevertheless, there is clear and significant room for improvement in undersampled DL MRI reconstruction to meet the high standards required for clinical diagnosis, in terms of eliminating aliasing artifacts and reducing image noise. In this paper, we introduce a self-supervised pretraining procedure using contrastive learning to improve the accuracy of undersampled DL MRI reconstruction. We use contrastive learning to transform the MRI image representations into a latent space that maximizes mutual information among different undersampled representations and optimizes the information content at the input of the downstream DL reconstruction models. Our experiments demonstrate improved reconstruction accuracy across a range of acceleration factors and datasets, both quantitatively and qualitatively. Furthermore, our extended experiments validate the proposed framework's robustness under adversarial conditions, such as measurement noise, different k-space sampling patterns, and pathological abnormalities, and also prove the transfer learning capabilities on MRI datasets with completely different anatomy. Additionally, we conducted experiments to visualize and analyze the properties of the proposed MRI contrastive learning latent space.

Image and Video Processing,Artificial Intelligence,Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

The paper focuses on improving the accuracy of reconstructing high-quality images from undersampled data in magnetic resonance imaging (MRI). Due to the need for fast scanning, MRI images are often undersampled in the measurement domain, which results in decreased image quality and impacts the accuracy of clinical diagnosis. In recent years, deep learning methods have emerged as the latest technology for MRI reconstruction, but challenges such as artifact removal and noise reduction still remain.

The paper proposes a self-supervised contrastive learning pre-training method (CL-MRI) aiming to improve the accuracy of undersampled deep learning MRI reconstruction. Through contrastive learning, the representation of MRI images is transformed into a latent space, maximizing the mutual information between different undersampled representations, and optimizing the informational content of the input for the downstream reconstruction model. Experimental results demonstrate that this approach enhances both quantitative and qualitative reconstruction accuracy across various acceleration factors and datasets. It also exhibits robustness under adversarial conditions such as noise, different k-space sampling patterns, and pathological abnormalities. Furthermore, the paper conducts visualizations and analyses of the contrastive learning MRI latent space. In summary, this paper aims to enhance MRI reconstruction accuracy through contrastive learning to address the limitations of current deep learning methods in achieving satisfactory reconstruction results at high acceleration factors.