Highly Accelerated MRI via Implicit Neural Representation Guided Posterior Sampling of Diffusion Models

Jiayue Chu,Chenhe Du,Xiyue Lin,Yuyao Zhang,Hongjiang Wei

2024-07-03

Abstract:Reconstructing high-fidelity magnetic resonance (MR) images from under-sampled k-space is a commonly used strategy to reduce scan time. The posterior sampling of diffusion models based on the real measurement data holds significant promise of improved reconstruction accuracy. However, traditional posterior sampling methods often lack effective data consistency guidance, leading to inaccurate and unstable reconstructions. Implicit neural representation (INR) has emerged as a powerful paradigm for solving inverse problems by modeling a signal's attributes as a continuous function of spatial coordinates. In this study, we present a novel posterior sampler for diffusion models using INR, named DiffINR. The INR-based component incorporates both the diffusion prior distribution and the MRI physical model to ensure high data fidelity. DiffINR demonstrates superior performance on experimental datasets with remarkable accuracy, even under high acceleration factors (up to R=12 in single-channel reconstruction). Notably, our proposed framework can be a generalizable framework to solve inverse problems in other medical imaging tasks.

Image and Video Processing,Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

The paper aims to address the problem of image reconstruction in accelerated Magnetic Resonance Imaging (MRI). Specifically, the researchers propose a novel method, which utilizes Implicit Neural Representation (INR) to guide the posterior sampling process of diffusion models (DiffINR) to achieve high-quality MRI image reconstruction at high acceleration rates.

In existing accelerated MRI techniques, undersampling k-space data to reduce scan time is a common strategy. However, due to the inherent issues of sparse data sampling, artifact-free reconstruction becomes very challenging. Traditional posterior sampling methods often lack effective data consistency guidance, leading to inaccurate and unstable reconstruction results. Additionally, although deep learning-based methods perform well, they require a large amount of high-quality paired training data and do not perform well under out-of-distribution conditions.

To address these issues, the paper proposes the DiffINR framework. This method combines the prior information generated by pre-trained diffusion models with the MRI physical model, ensuring data consistency during the reconstruction process. Experimental results show that DiffINR can generate more accurate and stable reconstruction results under different sampling patterns and high acceleration factors. Moreover, this method can be applied to different tasks without retraining, demonstrating its flexibility and generality.