Work

CL-MRI: Self-Supervised contrastive learning to improve the accuracy of undersampled MRI reconstruction

Year: 2024

Type: article

Abstract: • A novel contrastive learning pretraining framework to improve MRI reconstruction. • Maximizes mutual information among various undersampled representations of an image. • Optimizes the information content at the input of the reconstruction model. • Generalizable over a wide range of reconstruction models and acceleration factors. • Robust to adversarial input conditions and transferable between diverse datasets. Deep learning (DL) methods have emerged as the state-of-the-art for Magnetic Resonance Imaging (MRI) reconstruction. DL methods typically involve training deep neural networks to take undersampled MRI images as input and transform them into high-quality MRI images through data-driven processes. However, deep learning models often fail with higher levels of undersampling due to the insufficient information in the input, which is crucial for producing high-quality MRI images. Thus, optimizing the information content at the input of a DL reconstruction model could significantly improve reconstruction accuracy. 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. Code available here .

Source: Biomedical Signal Processing and Control

DOI: 10.1016/j.bspc.2024.107185

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

Institution: Monash University

Language: English

Topic: Advanced MRI Techniques and Applications

Subfield: Radiology, Nuclear Medicine and Imaging

Field: Medicine

Domain: Health Sciences

SDG: Good health and well-being

Open Access status: hybrid

Estimated APC paid: $2,510