SPIRiT-Diffusion: Self-Consistency Driven Diffusion Model for Accelerated MRI

2304.05060

YC

0

Reddit

0

Published 4/23/2024 by Zhuo-Xu Cui, Chentao Cao, Yue Wang, Sen Jia, Jing Cheng, Xin Liu, Hairong Zheng, Dong Liang, Yanjie Zhu

📈

Abstract

Diffusion models have emerged as a leading methodology for image generation and have proven successful in the realm of magnetic resonance imaging (MRI) reconstruction. However, existing reconstruction methods based on diffusion models are primarily formulated in the image domain, making the reconstruction quality susceptible to inaccuracies in coil sensitivity maps (CSMs). k-space interpolation methods can effectively address this issue but conventional diffusion models are not readily applicable in k-space interpolation. To overcome this challenge, we introduce a novel approach called SPIRiT-Diffusion, which is a diffusion model for k-space interpolation inspired by the iterative self-consistent SPIRiT method. Specifically, we utilize the iterative solver of the self-consistent term (i.e., k-space physical prior) in SPIRiT to formulate a novel stochastic differential equation (SDE) governing the diffusion process. Subsequently, k-space data can be interpolated by executing the diffusion process. This innovative approach highlights the optimization model's role in designing the SDE in diffusion models, enabling the diffusion process to align closely with the physics inherent in the optimization model, a concept referred to as model-driven diffusion. We evaluated the proposed SPIRiT-Diffusion method using a 3D joint intracranial and carotid vessel wall imaging dataset. The results convincingly demonstrate its superiority over image-domain reconstruction methods, achieving high reconstruction quality even at a substantial acceleration rate of 10.

Create account to get full access

or

If you already have an account, we'll log you in

Overview

  • Diffusion models have proven successful in magnetic resonance imaging (MRI) reconstruction, but existing methods are susceptible to inaccuracies in coil sensitivity maps (CSMs).
  • k-space interpolation methods can address this issue, but conventional diffusion models are not readily applicable in k-space interpolation.
  • The paper introduces a novel approach called SPIRiT-Diffusion, a diffusion model for k-space interpolation inspired by the iterative self-consistent SPIRiT method.

Plain English Explanation

Diffusion models are a type of machine learning technique that have been used successfully to generate images, including in the field of magnetic resonance imaging (MRI) reconstruction. However, the existing diffusion-based MRI reconstruction methods have a limitation - they work in the image domain, which means they can be affected by inaccuracies in the coil sensitivity maps (CSMs), which are used to interpret the MRI data.

To address this issue, the researchers explored using a different approach called k-space interpolation. K-space is the mathematical representation of the MRI data, and interpolation methods can be used to fill in missing or incomplete data in k-space. Traditional diffusion models, however, are not well-suited for working directly in k-space.

To overcome this challenge, the researchers developed a new method called SPIRiT-Diffusion. This method combines the strengths of diffusion models and the SPIRiT (Iterative Self-consistent Parallel Imaging Reconstruction) method, which is a technique for reconstructing MRI data in k-space. The key innovation is the way the researchers designed the mathematical equations that govern the diffusion process, aligning them closely with the physical properties of the MRI data represented in k-space.

By doing this, the SPIRiT-Diffusion method is able to achieve high-quality MRI reconstruction, even at high acceleration rates (i.e., when only a small fraction of the full MRI data is available). This is a significant advancement over previous diffusion-based MRI reconstruction methods that were limited by their reliance on the image domain.

Technical Explanation

The researchers introduce a novel approach called SPIRiT-Diffusion, which is a diffusion model for k-space interpolation inspired by the iterative self-consistent SPIRiT method. Specifically, they utilize the iterative solver of the self-consistent term (i.e., k-space physical prior) in SPIRiT to formulate a novel stochastic differential equation (SDE) governing the diffusion process. This enables the diffusion process to align closely with the physics inherent in the optimization model, a concept referred to as model-driven diffusion.

To evaluate the proposed SPIRiT-Diffusion method, the researchers used a 3D joint intracranial and carotid vessel wall imaging dataset. The results demonstrate the superiority of SPIRiT-Diffusion over image-domain reconstruction methods, achieving high reconstruction quality even at a substantial acceleration rate of 10.

Critical Analysis

The paper presents a novel and promising approach to MRI reconstruction using diffusion models in the k-space domain. By aligning the diffusion process with the physical properties of the MRI data, the SPIRiT-Diffusion method is able to overcome the limitations of previous diffusion-based approaches that were constrained to the image domain.

However, the paper does not extensively discuss the computational complexity or training time requirements of the SPIRiT-Diffusion method compared to other k-space interpolation techniques. Additionally, the evaluation is limited to a specific 3D imaging dataset, and further testing on a broader range of MRI applications would be valuable to assess the generalizability of the approach.

It would also be interesting to see how SPIRiT-Diffusion compares to other state-of-the-art diffusion-based MRI reconstruction methods, such as those discussed in related papers like Learning Domain-Specific Inverse NUFFT for Accelerated Spiral MRI, Rethinking Diffusion Models for Multi-Contrast MRI Super-Resolution, Self-Supervised MRI Reconstruction with Unrolled Diffusion Models, and DiffGEPCI: 3D MRI Synthesis from MGRE Signals. Comparing the performance, computational efficiency, and broader applicability of these different diffusion-based approaches would provide valuable insights for the field.

Conclusion

The SPIRiT-Diffusion method represents an innovative approach to MRI reconstruction that leverages the strengths of diffusion models and k-space interpolation. By aligning the diffusion process with the physical properties of the MRI data in k-space, the researchers have developed a technique that can achieve high-quality reconstruction even at substantial acceleration rates.

This work highlights the importance of considering the underlying optimization model when designing diffusion-based methods, as this can lead to significant performance improvements. The success of SPIRiT-Diffusion suggests that further exploration of "model-driven diffusion" approaches may yield valuable advances in MRI reconstruction and other domains that rely on complex physical or mathematical models.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Related Papers

Learning the Domain Specific Inverse NUFFT for Accelerated Spiral MRI using Diffusion Models

Learning the Domain Specific Inverse NUFFT for Accelerated Spiral MRI using Diffusion Models

Trevor J. Chan, Chamith S. Rajapakse

YC

0

Reddit

0

Deep learning methods for accelerated MRI achieve state-of-the-art results but largely ignore additional speedups possible with noncartesian sampling trajectories. To address this gap, we created a generative diffusion model-based reconstruction algorithm for multi-coil highly undersampled spiral MRI. This model uses conditioning during training as well as frequency-based guidance to ensure consistency between images and measurements. Evaluated on retrospective data, we show high quality (structural similarity > 0.87) in reconstructed images with ultrafast scan times (0.02 seconds for a 2D image). We use this algorithm to identify a set of optimal variable-density spiral trajectories and show large improvements in image quality compared to conventional reconstruction using the non-uniform fast Fourier transform. By combining efficient spiral sampling trajectories, multicoil imaging, and deep learning reconstruction, these methods could enable the extremely high acceleration factors needed for real-time 3D imaging.

Read more

5/14/2024

Autoregressive Image Diffusion: Generating Image Sequence and Application in MRI

Autoregressive Image Diffusion: Generating Image Sequence and Application in MRI

Guanxiong Luo, Shoujin Huang, Martin Uecker

YC

0

Reddit

0

Magnetic resonance imaging (MRI) is a widely used non-invasive imaging modality. However, a persistent challenge lies in balancing image quality with imaging speed. This trade-off is primarily constrained by k-space measurements, which traverse specific trajectories in the spatial Fourier domain (k-space). These measurements are often undersampled to shorten acquisition times, resulting in image artifacts and compromised quality. Generative models learn image distributions and can be used to reconstruct high-quality images from undersampled k-space data. In this work, we present the autoregressive image diffusion (AID) model for image sequences and use it to sample the posterior for accelerated MRI reconstruction. The algorithm incorporates both undersampled k-space and pre-existing information. Models trained with fastMRI dataset are evaluated comprehensively. The results show that the AID model can robustly generate sequentially coherent image sequences. In 3D and dynamic MRI, the AID can outperform the standard diffusion model and reduce hallucinations, due to the learned inter-image dependencies.

Read more

5/27/2024

Phy-Diff: Physics-guided Hourglass Diffusion Model for Diffusion MRI Synthesis

Phy-Diff: Physics-guided Hourglass Diffusion Model for Diffusion MRI Synthesis

Juanhua Zhang, Ruodan Yan, Alessandro Perelli, Xi Chen, Chao Li

YC

0

Reddit

0

Diffusion MRI (dMRI) is an important neuroimaging technique with high acquisition costs. Deep learning approaches have been used to enhance dMRI and predict diffusion biomarkers through undersampled dMRI. To generate more comprehensive raw dMRI, generative adversarial network based methods are proposed to include b-values and b-vectors as conditions, but they are limited by unstable training and less desirable diversity. The emerging diffusion model (DM) promises to improve generative performance. However, it remains challenging to include essential information in conditioning DM for more relevant generation, i.e., the physical principles of dMRI and white matter tract structures. In this study, we propose a physics-guided diffusion model to generate high-quality dMRI. Our model introduces the physical principles of dMRI in the noise evolution in the diffusion process and introduce a query-based conditional mapping within the difussion model. In addition, to enhance the anatomical fine detials of the generation, we introduce the XTRACT atlas as prior of white matter tracts by adopting an adapter technique. Our experiment results show that our method outperforms other state-of-the-art methods and has the potential to advance dMRI enhancement.

Read more

6/6/2024

Rethinking Diffusion Model for Multi-Contrast MRI Super-Resolution

Rethinking Diffusion Model for Multi-Contrast MRI Super-Resolution

Guangyuan Li, Chen Rao, Juncheng Mo, Zhanjie Zhang, Wei Xing, Lei Zhao

YC

0

Reddit

0

Recently, diffusion models (DM) have been applied in magnetic resonance imaging (MRI) super-resolution (SR) reconstruction, exhibiting impressive performance, especially with regard to detailed reconstruction. However, the current DM-based SR reconstruction methods still face the following issues: (1) They require a large number of iterations to reconstruct the final image, which is inefficient and consumes a significant amount of computational resources. (2) The results reconstructed by these methods are often misaligned with the real high-resolution images, leading to remarkable distortion in the reconstructed MR images. To address the aforementioned issues, we propose an efficient diffusion model for multi-contrast MRI SR, named as DiffMSR. Specifically, we apply DM in a highly compact low-dimensional latent space to generate prior knowledge with high-frequency detail information. The highly compact latent space ensures that DM requires only a few simple iterations to produce accurate prior knowledge. In addition, we design the Prior-Guide Large Window Transformer (PLWformer) as the decoder for DM, which can extend the receptive field while fully utilizing the prior knowledge generated by DM to ensure that the reconstructed MR image remains undistorted. Extensive experiments on public and clinical datasets demonstrate that our DiffMSR outperforms state-of-the-art methods.

Read more

4/9/2024