Patch-Based Diffusion for Data-Efficient, Radiologist-Preferred MRI Reconstruction

๐Ÿ“… 2025-09-25
๐Ÿ“ˆ Citations: 0
โœจ Influential: 0
๐Ÿ“„ PDF
๐Ÿค– AI Summary
MRI reconstruction faces challenges including prolonged acquisition times, motion artifacts, and scarcity of high-quality training data. To address these, we propose PaDIS-MRIโ€”the first patch-based diffusion model explicitly designed for complex-valued, multi-coil MRI reconstruction. It instantiates the Patch-based Diffusion Inverse Solver (PaDIS) framework for severely undersampled k-space recoveryโ€”a novel application of patch-based diffusion to inverse MRI problems. Remarkably, PaDIS-MRI achieves competitive performance with only 25 k-space samples for training, substantially outperforming the full-image diffusion baseline FastMRI-EDM: it attains superior PSNR, SSIM, and NRMSE on the FastMRI brain dataset. In blinded clinical evaluation, radiologists rated 91.7% of reconstructions as diagnostically superior. Moreover, PaDIS-MRI provides robust uncertainty quantification and demonstrates strong cross-contrast generalization. This work enables high-fidelity, clinically trustworthy MRI reconstruction at minimal data cost.

Technology Category

Computer Vision: Diffusion Models for VisionMachine Learning: Multimodal LearningKnowledge Representation and Reasoning: Diagnosis and Abductive Reasoning

Application Category

User Modeling, Personalization and Recommendation: Fairness-aware retrieval and rankingSearch and Retrieval-Augmented AI: Web learning to rank, online learning, and counterfactual learning for rankingGraph Algorithms and Modeling for the Web: Representation, reconstruction, and subgraph or motif discovery in Web-related graphs
๐Ÿ“ Abstract
Magnetic resonance imaging (MRI) requires long acquisition times, raising costs, reducing accessibility, and making scans more susceptible to motion artifacts. Diffusion probabilistic models that learn data-driven priors can potentially assist in reducing acquisition time. However, they typically require large training datasets that can be prohibitively expensive to collect. Patch-based diffusion models have shown promise in learning effective data-driven priors over small real-valued datasets, but have not yet demonstrated clinical value in MRI. We extend the Patch-based Diffusion Inverse Solver (PaDIS) to complex-valued, multi-coil MRI reconstruction, and compare it against a state-of-the-art whole-image diffusion baseline (FastMRI-EDM) for 7x undersampled MRI reconstruction on the FastMRI brain dataset. We show that PaDIS-MRI models trained on small datasets of as few as 25 k-space images outperform FastMRI-EDM on image quality metrics (PSNR, SSIM, NRMSE), pixel-level uncertainty, cross-contrast generalization, and robustness to severe k-space undersampling. In a blinded study with three radiologists, PaDIS-MRI reconstructions were chosen as diagnostically superior in 91.7% of cases, compared to baselines (i) FastMRI-EDM and (ii) classical convex reconstruction with wavelet sparsity. These findings highlight the potential of patch-based diffusion priors for high-fidelity MRI reconstruction in data-scarce clinical settings where diagnostic confidence matters.
Problem

Research questions and friction points this paper is trying to address.

Reducing MRI acquisition time through data-efficient reconstruction
Overcoming small dataset limitations with patch-based diffusion models
Achieving radiologist-preferred diagnostic quality in undersampled MRI
Innovation

Methods, ideas, or system contributions that make the work stand out.

Patch-based diffusion models for MRI reconstruction
Training on small datasets with 25 images
Outperforming baselines in radiologist diagnostic preference
๐Ÿ”Ž Similar Papers
No similar papers found.
๐Ÿ’ผ Related Jobs
No related jobs found.
R
Rohan Sanda
Department of Computer Science, Stanford University
A
Asad Aali
Department of Radiology, Stanford University School of Medicine
A
Andrew Johnston
Department of Radiology, Stanford University School of Medicine
E
Eduardo Reis
Stanford Center for Artificial Intelligence for Medicine and Imaging
J
Jonathan Singh
Department of Radiology, Kaiser Permanente
Gordon Wetzstein
Gordon Wetzstein
Associate Professor of Electrical Engineering and Computer Science, Stanford University
Computational ImagingComputational DisplaysComputational OpticsNeural Rendering
Sara Fridovich-Keil
Sara Fridovich-Keil
Assistant Professor in ECE, Georgia Tech
computational imagingmachine learningsignal processinginverse problems