arXiv:2505.03498cs.CVphysics.med-ph2025-05被引 2

用扩散模型快速修复脑部MRI运动伪影,仅需4步反向生成。

Res-MoCoDiff: Residual-guided diffusion models for motion artifact correction in brain MRI

  • 通过残差误差转移机制模拟噪声演化,提升去伪影效率。
  • 在真实和仿真数据上均达到最高SSIM与最低NMSE,PSNR达41.91 dB。
  • 每批两幅图像仅需0.37秒,比传统方法快近百倍,适合临床实时应用。

脑部MRI中的运动伪影主要由头部刚性移动引起,会降低图像质量并影响后续分析。传统缓解方法如重复扫描或运动追踪会增加流程负担。本文提出Res-MoCoDiff,一种专为MRI运动伪影校正设计的高效去噪扩散概率模型。该模型在前向扩散过程中引入新颖的残差误差转移机制,使噪声分布更贴近受损数据分布,从而实现仅需4步即可完成逆向生成。采用以Swin Transformer块替代注意力层的U-net主干网络,增强多尺度鲁棒性;训练中结合l1+l2损失函数,提升图像清晰度并减少像素级误差。模型在基于真实运动模拟框架生成的仿真数据集及in-vivo MR-ART数据集上进行评估,并与CycleGAN、Pix2pix及基于视觉变换器的扩散模型对比,使用PSNR、SSIM、NMSE等定量指标。结果表明,该方法在轻度、中度、重度失真下均表现优异,持续获得最高SSIM与最低NMSE,轻度失真时PSNR达41.91±2.94 dB。平均采样时间降至每批两幅图像0.37秒,相较传统方法的101.74秒显著缩短。

原文摘要 · Abstract (English)

Objective. Motion artifacts in brain MRI, mainly from rigid head motion, degrade image quality and hinder downstream applications. Conventional methods to mitigate these artifacts, including repeated acquisitions or motion tracking, impose workflow burdens. This study introduces Res-MoCoDiff, an efficient denoising diffusion probabilistic model specifically designed for MRI motion artifact correction.Approach.Res-MoCoDiff exploits a novel residual error shifting mechanism during the forward diffusion process to incorporate information from motion-corrupted images. This mechanism allows the model to simulate the evolution of noise with a probability distribution closely matching that of the corrupted data, enabling a reverse diffusion process that requires only four steps. The model employs a U-net backbone, with attention layers replaced by Swin Transformer blocks, to enhance robustness across resolutions. Furthermore, the training process integrates a combined l1+l2 loss function, which promotes image sharpness and reduces pixel-level errors. Res-MoCoDiff was evaluated on both an in-silico dataset generated using a realistic motion simulation framework and an in-vivo MR-ART dataset. Comparative analyses were conducted against established methods, including CycleGAN, Pix2pix, and a diffusion model with a vision transformer backbone, using quantitative metrics such as PSNR, SSIM, and NMSE.Main results. The proposed method demonstrated superior performance in removing motion artifacts across minor, moderate, and heavy distortion levels. Res-MoCoDiff consistently achieved the highest SSIM and the lowest NMSE values, with a PSNR of up to 41.91+-2.94 dB for minor distortions. Notably, the average sampling time was reduced to 0.37 seconds per batch of two image slices, compared with 101.74 seconds for conventional approaches.

MRI修复扩散模型运动伪影Swin Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。