Consistency Models for Fast MRI Reconstruction Using Regularization by Denoising
Authors: Merve Gülle, Junno Yun, Yaşar Utku Alçalar, Mehmet Akçakaya
Organizations: Department of Electrical and Computer Engineering, University of Minnesota, MN, USA · Center for Magnetic Resonance Research, University of Minnesota, MN, USA
Diffusion models (DMs) have emerged as powerful generative priors for MRI reconstruction with promising results. Yet DM-based methods require extensive iterative refinement, limiting their practical deployment. Consistency models (CMs) provide a compelling alternative, aiming to map out the diffusion trajectory in a single pass, enabling faster generation. In this work, we propose CM-RED, a novel MRI reconstruction method that integrates a pretrained CM into the regularization by denoising (RED) scheme. Our method builds on accelerated proximal gradient RED (RED-APG), and further incorporates controlled noise injection during the update steps to enhance generative diversity and accelerate convergence. Extensive experiments on the fastMRI knee and brain datasets demonstrate that CM-RED achieves high-quality reconstructions across multiple anatomies, contrast weights, acceleration factors, and undersampling patterns, using only 4 network function evaluations (NFEs). The proposed method consistently outperforms existing DM- and CM-based approaches in both quantitative metrics and visual fidelity, and exhibits strong robustness to hyperparameter variations, highlighting CM-RED as an efficient and effective generative framework for accelerated MRI reconstruction. The source code and pretrained models are publicly available at https://github.com/MerveGulle/CM-RED.
Figures & tables
Figure 1: Illustration of DM and CM trajectories on MRI data.
Figure 2: Illustration of CM-RED steps with N=4 , as described in Alg. 1 .
Acc.
iN
γ
δn
κ
ρ
μ(0)
4 ×
50
0.1
(0.4,3.0,3.0,2.5)
(0.5,5.0)
–2.0
0.9
8 ×
50
0.1
(0.5,7.0,6.0,3.5)
(1.0,2.5)
–3.0
0.9
Table 1: Complete set of hyperparameters for our four-step sampling setup, including noise schedule, regularization parameters, CM out scale parameter, and momentum coefficients.
Figure 3: Representative reconstructions of DM- and CM-based methods on fastMRI knee and brain data with equidistant R=4 and R=8 undersampling. The number of NFEs used by each method is shown in parentheses. Yellow boxes indicate magnified regions, and red arrows highlight residual artifacts. Best viewed when zoomed in.
Figure 4: Representative reconstructions for Gaussian undersampling. Number of NFEs for by each method is provided in parentheses. Yellow boxes indicate magnified regions, and red arrows highlight residual artifacts. Best viewed zoomed in.
Method
NFEs ↓
R
Cor PD
Cor PD-FS
Axial T1-Pre
Axial T1-Post
Axial T2
Axial FLAIR
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
DPS
1000
×4
34.82
0.886
32.37
0.781
37.70
0.928
38.25
0.932
34.49
0.911
34.37
0.907
×8
31.79
0.832
30.30
0.727
33.75
0.896
33.89
0.899
30.14
0.868
30.23
0.862
DDS
100
×4
34.86
0.901
31.40
0.781
38.24
0.958
38.17
0.955
34.22
0.928
35.29
0.936
×8
31.63
0.837
29.67
0.720
34.82
0.927
34.63
0.923
30.88
0.884
31.35
0.887
CM4IR
4
×4
29.96
0.766
24.32
0.528
36.83
0.918
37.35
0.917
31.43
0.856
34.47
0.915
Table 2: Quantitative comparison of reconstruction methods on fastMRI Knoll et al. (2020b) knee and brain datasets with R=4 and 8 using 1D equidistant undersampling. Best: BOLD , second-best: Underlined .
Method
NFEs ↓
R
Cor PD
Cor PD-FS
Axial T1-Pre
Axial T1-Post
Axial T2
Axial FLAIR
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
DPS
1000
×4
35.19
0.894
32.81
0.795
35.36
0.915
35.88
0.918
33.37
0.903
32.92
0.901
×8
33.64
0.862
31.65
0.757
33.87
0.906
34.16
0.909
32.22
0.893
31.92
0.887
DDS
100
×4
35.10
0.909
31.94
0.796
37.77
0.956
38.00
0.956
34.15
0.928
35.47
0.940
×8
33.14
0.871
30.47
0.745
35.63
0.939
35.68
0.936
32.19
0.903
33.14
0.912
CM4IR
4
×4
32.71
0.839
26.58
0.598
36.62
0.927
37.63
0.934
32.31
0.879
33.33
0.900
Table 3: Quantitative comparison of reconstruction methods on fastMRI Knoll et al. (2020b) knee and brain datasets with R=4 and 8 using 1D Gaussian undersampling. Best: BOLD , second-best: Underlined .
Figure 5: Hyperparameter sensitivity analysis for CM-RED with R=8 Gaussian undersampling. Violin plots show the per-slice difference in PSNR (left) and SSIM (right) between the nominal configuration and reconstructions obtained with randomly perturbed hyperparameters.
Noise Inj.
Momentum
Cor PD Knee
Ax T2 Brain
✗
✗
35.59 / 0.917
34.96 / 0.936
✗
✓
36.09 / 0.923
35.25 / 0.937
✓
✗
35.71 / 0.917
35.23 / 0.937
✓
✓
36.58 / 0.927
36.13 / 0.945
Table 4: PSNR/SSIM on coronal PD knee and axial T2 brain R=4 equidistant undersampling with/without momentum and noise injection using 4 NFEs.
Figure 6: Effect of noise injection and momentum on CM-RED convergence for fastMRI axial T1-post brain data with R=8 equidistant undersampling. PSNR and SSIM are reported between 4 to 30 NFEs.
Chandra Family Department of Electrical and Computer Engineering The University of Texas at Austin Austin, TX, USA · Department of Imaging Physics MD Anderson Cancer Center Houston, TX, USA · Dell Medical School Department of Neurology The University of Texas at Austin Austin, TX, USA +3
Department of Radiology & Biomedical Imaging, Yale School of Medicine, New Haven, CT 06510 USA · Department of Biomedical Engineering, Rensselaer Polytechnic Institute, Troy, NY 12180 USA