Rethinking Diffusion Model for Multi-Contrast MRI Super-Resolution

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

2024-04-07

Abstract: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.

Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

The problems this paper attempts to address are:

1. **High number of iterations and low efficiency**: Existing MRI super-resolution reconstruction methods based on diffusion models (DM) require a large number of iterative steps to generate the final image, which is not only inefficient but also consumes a significant amount of computational resources.
2. **Poor alignment of reconstruction results with real high-resolution images**: The images generated by these methods often deviate from real high-resolution images, leading to noticeable distortions in the reconstructed MRI images.

To tackle these issues, the authors propose an efficient diffusion model for multi-contrast MRI super-resolution reconstruction, called DiffMSR. Specifically, the authors apply the diffusion model in a highly compact low-dimensional latent space to generate prior knowledge containing high-frequency detail information. Additionally, they design a Prior-Guide Large Window Transformer (PLWformer) as the decoder of the diffusion model to expand the receptive field and fully utilize the generated prior knowledge, ensuring that the reconstructed MRI images are free from distortion.

With these improvements, experimental results on public and clinical datasets show that DiffMSR outperforms existing state-of-the-art methods.