Adaptive Diffusion Priors for Accelerated MRI Reconstruction

Abstract

Deep MRI reconstruction is commonly performed with conditional models that de-alias undersampled acquisitions to recover images consistent with fully-sampled data. Since conditional models are trained with knowledge of the imaging operator, they can show poor generalization across variable operators. Unconditional models instead learn generative image priors decoupled from the operator to improve reliability against domain shifts related to the imaging operator. Recent diffusion models are particularly promising given their high sample fidelity. Nevertheless, inference with a static image prior can perform suboptimally. Here we propose the first adaptive diffusion prior for MRI reconstruction, AdaDiff, to improve performance and reliability against domain shifts. AdaDiff leverages an efficient diffusion prior trained via adversarial mapping over large reverse diffusion steps. A two-phase reconstruction is executed following training: a rapid-diffusion phase that produces an initial reconstruction with the trained prior, and an adaptation phase that further refines the result by updating the prior to minimize data-consistency loss. Demonstrations on multi-contrast brain MRI clearly indicate that AdaDiff outperforms competing conditional and unconditional methods under domain shifts, and achieves superior or on par within-domain performance.

References in corpus (24)

• Image reconstruction by domain transform manifold learning
• Score-Based Generative Modeling through Stochastic Differential Equations
• MoDL: Model Based Deep Learning Architecture for Inverse Problems
• Which Training Methods for GANs do actually Converge?
• Learned Primal-dual Reconstruction
• Compressed Sensing MRI Reconstruction using a Generative Adversarial Network with a Cyclic Loss
• ResViT: Residual vision transformers for multi-modal medical image synthesis
• Deep learning for undersampled MRI reconstruction
• Self-Supervised Learning of Physics-Guided Reconstruction Neural Networks without Fully-Sampled Reference Data
• Plug and play methods for magnetic resonance imaging (long version)
• Inertial Proximal Alternating Linearized Minimization (iPALM) for Nonconvex and Nonsmooth Problems
• MR image reconstruction using deep density priors
• RARE: Image Reconstruction using Deep Priors Learned without Ground Truth
• Tackling the Generative Learning Trilemma with Denoising Diffusion GANs
• Solving Inverse Problems in Medical Imaging with Score-Based Generative Models
• TranSMS: Transformers for Super-Resolution Calibration in Magnetic Particle Imaging
• Dense Recurrent Neural Networks for Accelerated MRI: History-Cognizant Unrolling of Optimization Algorithms
• Wasserstein GANs for MR Imaging: from Paired to Unpaired Training
• Multi-Modal MRI Reconstruction Assisted with Spatial Alignment Network
• MRI Reconstruction Using Deep Bayesian Estimation
• Unsupervised MRI Reconstruction with Generative Adversarial Networks
• An Unsupervised Deep Learning Method for Multi-coil Cine MRI
• One Model to Unite Them All: Personalized Federated Learning of Multi-Contrast MRI Synthesis
• A Few-Shot Learning Approach for Accelerated MRI via Fusion of Data-Driven and Subject-Driven Priors

Cited by in corpus (13)

• High-Frequency Space Diffusion Models for Accelerated MRI
• Data and Physics driven Deep Learning Models for Fast MRI Reconstruction: Fundamentals and Methodologies
• CT Reconstruction using Diffusion Posterior Sampling conditioned on a Nonlinear Measurement Model
• A Tutorial on MRI Reconstruction: From Modern Methods to Clinical Implications
• Self-Supervised Adversarial Diffusion Models for Fast MRI Reconstruction
• Noise Level Adaptive Diffusion Model for Robust Reconstruction of Accelerated MRI
• Fast Inference in Denoising Diffusion Models via MMD Finetuning
• Enhancing quality and speed in database-free neural network reconstructions of undersampled MRI with SCAMPI
• Advancing MRI Reconstruction: A Systematic Review of Deep Learning and Compressed Sensing Integration
• UNISELF: A Unified Network with Instance Normalization and Self-Ensembled Lesion Fusion for Multiple Sclerosis Lesion Segmentation
• Fast MRI Reconstruction Using Deep Learning-based Compressed Sensing: A Systematic Review
• Robust Simultaneous Multislice MRI Reconstruction Using Slice-Wise Learned Generative Diffusion Priors
• MCMC-Correction of Score-Based Diffusion Models for Model Composition