Abstract:Diffusion models have been demonstrated as powerful deep learning tools for image generation in CT reconstruction and restoration. Recently, diffusion posterior sampling, where a score-based diffusion prior is combined with a likelihood model, has been used to produce high quality CT images given low-quality measurements. This technique is attractive since it permits a one-time, unsupervised training of a CT prior; which can then be incorporated with an arbitrary data model. However, current methods rely on a linear model of x-ray CT physics to reconstruct or restore images. While it is common to linearize the transmission tomography reconstruction problem, this is an approximation to the true and inherently nonlinear forward model. We propose a new method that solves the inverse problem of nonlinear CT image reconstruction via diffusion posterior sampling. We implement a traditional unconditional diffusion model by training a prior score function estimator, and apply Bayes rule to combine this prior with a measurement likelihood score function derived from the nonlinear physical model to arrive at a posterior score function that can be used to sample the reverse-time diffusion process. This plug-and-play method allows incorporation of a diffusion-based prior with generalized nonlinear CT image reconstruction into multiple CT system designs with different forward models, without the need for any additional training. We develop the algorithm that performs this reconstruction, including an ordered-subsets variant for accelerated processing and demonstrate the technique in both fully sampled low dose data and sparse-view geometries using a single unsupervised training of the prior.

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

Ambient Diffusion: Learning Clean Distributions from Corrupted Data

Diffusion posterior sampling for magnetic resonance imaging

Adapt and Diffuse: Sample-adaptive Reconstruction via Latent Diffusion Models

Learning Diffusion Model from Noisy Measurement using Principled Expectation-Maximization Method

Solving Linear Inverse Problems Provably via Posterior Sampling with Latent Diffusion Models

Consistent Diffusion Meets Tweedie: Training Exact Ambient Diffusion Models with Noisy Data

DiracDiffusion: Denoising and Incremental Reconstruction with Assured Data-Consistency

An Expectation-Maximization Algorithm for Training Clean Diffusion Models from Corrupted Observations

CT Reconstruction using Nonlinear Diffusion Posterior Sampling with Detector Blur Modeling

GibbsDDRM: A Partially Collapsed Gibbs Sampler for Solving Blind Inverse Problems with Denoising Diffusion Restoration

Diffusion Model Based Posterior Sampling for Noisy Linear Inverse Problems

Adaptive Diffusion Priors for Accelerated MRI Reconstruction

Diffusion Posterior Sampling for Nonlinear CT Reconstruction

Diffusion Prior-Based Amortized Variational Inference for Noisy Inverse Problems

Mask, Stitch, and Re-Sample: Enhancing Robustness and Generalizability in Anomaly Detection through Automatic Diffusion Models

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

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

Enhancing Diffusion Posterior Sampling for Inverse Problems by Integrating Crafted Measurements

Accelerated Motion Correction with Deep Generative Diffusion Models

Decoupled Data Consistency with Diffusion Purification for Image Restoration