论文标题
加速MRI的高频空间扩散模型
High-Frequency Space Diffusion Models for Accelerated MRI
论文作者
论文摘要
具有连续随机微分方程(SDE)的扩散模型在图像产生中表现出了出色的性能。在解决磁共振(MR)重建中的逆问题之前,它可以用作深层生成性。但是,$ K $ - 空间数据的低频区域通常在快速MR成像中完全采样,而现有的扩散模型则在整个图像或$ K $ -SPACE中执行,不可避免地会在低频率区域的重建中引入不确定性。此外,现有的扩散模型通常需要进行大量迭代以收敛,从而导致耗时的重建。为了应对这些挑战,我们提出了一种专门针对MR重建的新型SDE,并在高频空间中的扩散过程(称为HFS-SDE)。这种方法确保了完全采样的低频区域的确定性,并加速了反向扩散的采样程序。在公开可用的FastMRI数据集上进行的实验表明,在重建精度和稳定性方面,提出的HFS-SDE方法优于传统的平行成像方法,监督深度学习和现有扩散模型。快速收敛性能也通过理论和实验验证确认。我们的代码和权重可从https://github.com/aboriginer/hfs-sde获得。
Diffusion models with continuous stochastic differential equations (SDEs) have shown superior performances in image generation. It can serve as a deep generative prior to solving the inverse problem in magnetic resonance (MR) reconstruction. However, low-frequency regions of $k$-space data are typically fully sampled in fast MR imaging, while existing diffusion models are performed throughout the entire image or $k$-space, inevitably introducing uncertainty in the reconstruction of low-frequency regions. Additionally, existing diffusion models often demand substantial iterations to converge, resulting in time-consuming reconstructions. To address these challenges, we propose a novel SDE tailored specifically for MR reconstruction with the diffusion process in high-frequency space (referred to as HFS-SDE). This approach ensures determinism in the fully sampled low-frequency regions and accelerates the sampling procedure of reverse diffusion. Experiments conducted on the publicly available fastMRI dataset demonstrate that the proposed HFS-SDE method outperforms traditional parallel imaging methods, supervised deep learning, and existing diffusion models in terms of reconstruction accuracy and stability. The fast convergence properties are also confirmed through theoretical and experimental validation. Our code and weights are available at https://github.com/Aboriginer/HFS-SDE.