Abstract:Diffusion Posterior Sampling(DPS) methodology is a novel framework that permits nonlinear CT reconstruction by integrating a diffusion prior and an analytic physical system model, allowing for one-time training for different applications. However, baseline DPS can struggle with large variability, hallucinations, and slow reconstruction. This work introduces a number of strategies designed to enhance the stability and efficiency of DPS CT reconstruction. Specifically, jumpstart sampling allows one to skip many reverse time steps, significantly reducing the reconstruction time as well as the sampling variability. Additionally, the likelihood update is modified to simplify the Jacobian computation and improve data consistency more efficiently. Finally, a hyperparameter sweep is conducted to investigate the effects of parameter tuning and to optimize the overall reconstruction performance. Simulation studies demonstrated that the proposed DPS technique achieves up to 46.72% PSNR and 51.50% SSIM enhancement in a low-mAs setting, and an over 31.43% variability reduction in a sparse-view setting. Moreover, reconstruction time is sped up from >23.5 s/slice to <1.5 s/slice. In a physical data study, the proposed DPS exhibits robustness on an anthropomorphic phantom reconstruction which does not strictly follow the prior distribution. Quantitative analysis demonstrates that the proposed DPS can accommodate various dose levels and number of views. With 10% dose, only a 5.60% and 4.84% reduction of PSNR and SSIM was observed for the proposed approach. Both simulation and phantom studies demonstrate that the proposed method can significantly improve reconstruction accuracy and reduce computational costs, greatly enhancing the practicality of DPS CT reconstruction.

Diffusion posterior sampling for magnetic resonance imaging

Diffusion Model Based Posterior Sampling for Noisy Linear Inverse Problems

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

Strategies for CT Reconstruction using Diffusion Posterior Sampling with a Nonlinear Model

Diffusion Posterior Sampling for Nonlinear CT Reconstruction

Joint Non-Linear MRI Inversion with Diffusion Priors

Ambient Diffusion Posterior Sampling: Solving Inverse Problems with Diffusion Models trained on Corrupted Data

CT Reconstruction using Diffusion Posterior Sampling conditioned on a Nonlinear Measurement Model

CT Reconstruction using Nonlinear Diffusion Posterior Sampling with Detector Blur Modeling

High-Frequency Space Diffusion Models for Accelerated MRI

Sub-DM:Subspace Diffusion Model with Orthogonal Decomposition for MRI Reconstruction

High-Frequency Space Diffusion Model for Accelerated MRI

Diffusion Modeling with Domain-conditioned Prior Guidance for Accelerated MRI and qMRI Reconstruction

Variational Diffusion Posterior Sampling with Midpoint Guidance

Bayesian MRI reconstruction with joint uncertainty estimation using diffusion models

Think Twice Before You Act: Improving Inverse Problem Solving With MCMC

ODE-DPS: ODE-based Diffusion Posterior Sampling for Inverse Problems in Partial Differential Equation

Diffusion State-Guided Projected Gradient for Inverse Problems

Towards performant and reliable undersampled MR reconstruction via diffusion model sampling

LDPM: Towards undersampled MRI reconstruction with MR-VAE and Latent Diffusion Prior

MRI Reconstruction with Side Information using Diffusion Models