High-Frequency Space Diffusion Models for Accelerated MRI

Chentao Cao,Zhuo-Xu Cui,Yue Wang,Shaonan Liu,Taijin Chen,Hairong Zheng,Dong Liang,Yanjie Zhu
DOI: https://doi.org/10.1109/TMI.2024.3351702
2024-01-20
Abstract: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 <a class="link-external link-https" href="https://github.com/Aboriginer/HFS-SDE" rel="external noopener nofollow">this https URL</a>.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
### What problems does this paper attempt to solve? This paper aims to solve the problems faced by fast imaging techniques in magnetic resonance imaging (MRI). Specifically, the paper focuses on the following two main challenges: 1. **Consistency problem in the low - frequency region**: - In fast MRI imaging, the low - frequency region of k - space data is usually fully sampled, while the existing diffusion models operate on the entire image or k - space, which may lead to increased uncertainty in the reconstruction of the low - frequency region. Therefore, how to ensure that the low - frequency region of the reconstructed image is consistent with the actually collected low - frequency region is a key issue. 2. **Problem of slow convergence**: - Existing diffusion models usually require a large number of iterations to converge, resulting in a long - lasting reconstruction process. This is an important bottleneck for fast MRI imaging in clinical applications. To solve these problems, the authors propose a new stochastic differential equation (SDE), called the high - frequency frequency - domain diffusion model (HFS - SDE). This method ensures the determinacy of the low - frequency region by performing the diffusion process in the high - frequency frequency - domain and accelerates the sampling process of the reverse diffusion. Experimental results show that HFS - SDE is superior to traditional parallel imaging methods, supervised deep learning methods, and existing diffusion models in terms of reconstruction accuracy and stability. In addition, theoretical and experimental verifications also confirm that HFS - SDE has a faster convergence speed.