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

• 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 .

Authors

Institutions

Publication Details

Journal
Biomedical Signal Processing and Control
Published
2024-11-08
DOI
https://doi.org/10.1016/j.bspc.2024.107185
Citations
9
Primary Topic
Advanced MRI Techniques and Applications
Type
article
Field-Weighted Citation Impact
3.64

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

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

Gary F. Egan, Mehrtash Harandi, Zhaolin Chen, Zhifeng Chen et al.
9 citations
Biomedical Signal Processing and Control
Advanced MRI Techniques and Applications
3.64
article

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

Gary F. Egan, Mehrtash Harandi, Zhaolin Chen, Zhifeng Chen, Mevan Ekanayake
article en
9 citations

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 .

Biomedical Signal Processing and ControlVol. 100
Monash University (AU)
Australian Research Council
Openalex Percentile: Top 6%
Advanced MRI Techniques and Applications
3.64
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.