Abstract:In a great number of tasks in science and engineering, the goal is to infer an unknown image from a small number of measurements collected from a known forward model describing certain sensing or imaging modality. Due to resource constraints, this task is often extremely ill-posed, which necessitates the adoption of expressive prior information to regularize the solution space. Score-based diffusion models, due to its impressive empirical success, have emerged as an appealing candidate of an expressive prior in image reconstruction. In order to accommodate diverse tasks at once, it is of great interest to develop efficient, consistent and robust algorithms that incorporate unconditional score functions of an image prior distribution in conjunction with flexible choices of forward models. This work develops an algorithmic framework for employing score-based diffusion models as an expressive data prior in general nonlinear inverse problems. Motivated by the plug-and-play framework in the imaging community, we introduce a diffusion plug-and-play method (DPnP) that alternatively calls two samplers, a proximal consistency sampler based solely on the likelihood function of the forward model, and a denoising diffusion sampler based solely on the score functions of the image prior. The key insight is that denoising under white Gaussian noise can be solved rigorously via both stochastic (i.e., DDPM-type) and deterministic (i.e., DDIM-type) samplers using the unconditional score functions. We establish both asymptotic and non-asymptotic performance guarantees of DPnP, and provide numerical experiments to illustrate its promise in solving both linear and nonlinear image reconstruction tasks. To the best of our knowledge, DPnP is the first provably-robust posterior sampling method for nonlinear inverse problems using unconditional diffusion priors.

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

Diffusion Model Based Posterior Sampling for Noisy Linear Inverse Problems

Particle-Filtering-based Latent Diffusion for Inverse Problems

Solving Inverse Problems via Diffusion Optimal Control

A Variational Perspective on Solving Inverse Problems with Diffusion Models

Solving General Noisy Inverse Problem via Posterior Sampling: A Policy Gradient Viewpoint

Diffusion Posterior Sampling for General Noisy Inverse Problems

Beyond First-Order Tweedie: Solving Inverse Problems using Latent Diffusion

Enhancing Diffusion Posterior Sampling for Inverse Problems by Integrating Crafted Measurements

Blind Inversion using Latent Diffusion Priors

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

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

Removing Structured Noise with Diffusion Models

Accelerating Diffusion Models for Inverse Problems through Shortcut Sampling

Provably Robust Score-Based Diffusion Posterior Sampling for Plug-and-Play Image Reconstruction

Gaussian is All You Need: A Unified Framework for Solving Inverse Problems via Diffusion Posterior Sampling

Diffusion State-Guided Projected Gradient for Inverse Problems

Denoising Diffusion Restoration Tackles Forward and Inverse Problems for the Laplace Operator

Improving Diffusion Models for Inverse Problems using Manifold Constraints

Diffusion with Forward Models: Solving Stochastic Inverse Problems Without Direct Supervision