Highly Undersampled MRI Reconstruction via a Single Posterior Sampling of Diffusion Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Liu, Jin, Lin, Qing, Xiong, Zhuang, Shan, Shanshan, Liu, Chunyi, Li, Min, Liu, Feng, Pike, G. Bruce, Sun, Hongfu, Gao, Yang
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917208672698368
author Liu, Jin
Lin, Qing
Xiong, Zhuang
Shan, Shanshan
Liu, Chunyi
Li, Min
Liu, Feng
Pike, G. Bruce
Sun, Hongfu
Gao, Yang
author_facet Liu, Jin
Lin, Qing
Xiong, Zhuang
Shan, Shanshan
Liu, Chunyi
Li, Min
Liu, Feng
Pike, G. Bruce
Sun, Hongfu
Gao, Yang
contents Incoherent k-space undersampling and deep learning-based reconstruction methods have shown great success in accelerating MRI. However, the performance of most previous methods will degrade dramatically under high acceleration factors, e.g., 8$\times$ or higher. Recently, denoising diffusion models (DM) have demonstrated promising results in solving this issue; however, one major drawback of the DM methods is the long inference time due to a dramatic number of iterative reverse posterior sampling steps. In this work, a Single Step Diffusion Model-based reconstruction framework, namely SSDM-MRI, is proposed for restoring MRI images from highly undersampled k-space. The proposed method achieves one-step reconstruction by first training a conditional DM and then iteratively distilling this model four times using an iterative selective distillation algorithm, which works synergistically with a shortcut reverse sampling strategy for model inference. Comprehensive experiments were carried out on both publicly available fastMRI brain and knee images, as well as an in-house multi-echo GRE (QSM) subject. Overall, the results showed that SSDM-MRI outperformed other methods in terms of numerical metrics (e.g., PSNR and SSIM), error maps, image fine details, and latent susceptibility information hidden in MRI phase images. In addition, the reconstruction time for a 320$\times$320 brain slice of SSDM-MRI is only 0.45 second, which is only comparable to that of a simple U-net, making it a highly effective solution for MRI reconstruction tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2505_08142
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Highly Undersampled MRI Reconstruction via a Single Posterior Sampling of Diffusion Models
Liu, Jin
Lin, Qing
Xiong, Zhuang
Shan, Shanshan
Liu, Chunyi
Li, Min
Liu, Feng
Pike, G. Bruce
Sun, Hongfu
Gao, Yang
Image and Video Processing
Incoherent k-space undersampling and deep learning-based reconstruction methods have shown great success in accelerating MRI. However, the performance of most previous methods will degrade dramatically under high acceleration factors, e.g., 8$\times$ or higher. Recently, denoising diffusion models (DM) have demonstrated promising results in solving this issue; however, one major drawback of the DM methods is the long inference time due to a dramatic number of iterative reverse posterior sampling steps. In this work, a Single Step Diffusion Model-based reconstruction framework, namely SSDM-MRI, is proposed for restoring MRI images from highly undersampled k-space. The proposed method achieves one-step reconstruction by first training a conditional DM and then iteratively distilling this model four times using an iterative selective distillation algorithm, which works synergistically with a shortcut reverse sampling strategy for model inference. Comprehensive experiments were carried out on both publicly available fastMRI brain and knee images, as well as an in-house multi-echo GRE (QSM) subject. Overall, the results showed that SSDM-MRI outperformed other methods in terms of numerical metrics (e.g., PSNR and SSIM), error maps, image fine details, and latent susceptibility information hidden in MRI phase images. In addition, the reconstruction time for a 320$\times$320 brain slice of SSDM-MRI is only 0.45 second, which is only comparable to that of a simple U-net, making it a highly effective solution for MRI reconstruction tasks.
title Highly Undersampled MRI Reconstruction via a Single Posterior Sampling of Diffusion Models
topic Image and Video Processing
url https://arxiv.org/abs/2505.08142