用去噪正则化整合一致性模型,4次前向传播完成高精度MRI重建。
Consistency Models for Fast MRI Reconstruction Using Regularization by Denoising

- 将预训练一致性模型嵌入去噪正则化框架,单步生成扩散轨迹。
- 仅需4次网络函数评估(NFEs),在膝关节与脑部数据上均达高质量重建。
- 对超参数不敏感,适合临床快速部署的加速MRI重建场景。
扩散模型(DMs)作为强大的生成先验在MRI重建中表现优异,但其需要大量迭代优化,限制了实际应用。一致性模型(CMs)提供了一种替代方案,可在单次前向传播中映射扩散轨迹,实现更快生成。本文提出CM-RED,将预训练的CM融入去噪正则化(RED)框架。方法基于加速近端梯度RED(RED-APG),并在更新步骤中引入可控噪声注入,以增强生成多样性并加速收敛。在fastMRI膝关节和脑部数据集上的大量实验表明,CM-RED在多种解剖结构、对比权重、加速度因子和欠采样模式下均实现高质量重建,仅需4次网络函数评估(NFEs)。所提方法在定量指标和视觉保真度上持续优于现有基于DM和CM的方法,且对超参数变化具有强鲁棒性,展现出高效可靠的生成式加速MRI重建能力。源代码与预训练模型已公开于https://github.com/MerveGulle/CM-RED。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。