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 General Noisy Inverse Problem via Posterior Sampling: A Policy Gradient Viewpoint

Diffusion Model Based Posterior Sampling for Noisy Linear Inverse Problems

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

Diffusion Posterior Sampling for General Noisy Inverse Problems

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

Diffusion State-Guided Projected Gradient for Inverse Problems

Enhancing Diffusion Posterior Sampling for Inverse Problems by Integrating Crafted Measurements

A Variational Perspective on Solving Inverse Problems with Diffusion Models

Improving Decoupled Posterior Sampling for Inverse Problems using Data Consistency Constraint

Posterior sampling via Langevin dynamics based on generative priors

G2D2: Gradient-guided Discrete Diffusion for image inverse problem solving

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

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

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

Improving Diffusion Inverse Problem Solving with Decoupled Noise Annealing

Tweedie Moment Projected Diffusions For Inverse Problems

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

Adaptive Sampling and Reconstruction for Gradient-Domain Rendering

Solving Inverse Problems via Diffusion Optimal Control

Unleashing the Denoising Capability of Diffusion Prior for Solving Inverse Problems