Posterior Sampling with Denoising Oracles via Tilted Transport
On the map: Diffusion models → Inverse problems and image restoration
Abstract (arXiv)
Score-based diffusion models have significantly advanced high-dimensional data generation across various domains, by learning a denoising oracle (or score) from datasets. From a Bayesian perspective, they offer a realistic modeling of data priors and facilitate solving inverse problems through posterior sampling. Although many heuristic methods have been developed recently for this purpose, they lack the quantitative guarantees needed in many scientific applications. In this work, we introduce the textittilted transport technique, which leverages the quadratic structure of the log-likelihood in linear inverse problems in combination with the prior denoising oracle to transform the original posterior sampling problem into a new `boosted' posterior that is provably easier to sample from. We quantify the conditions under which this boosted posterior is strongly log-concave, highlighting the dependencies on the condition number of the measurement matrix and the signal-to-noise ratio. The resulting posterior sampling scheme is shown to reach the computational threshold predicted for sampling Ising models [Kunisky'23] with a direct analysis, and is further validated on high-dimensional Gaussian mixture models and scalar field varphi4 models.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Placed in Inverse problems and image restoration, one of 14 directions in Diffusion models. Builds on 35 of its 72 references that are in the map. First posted on arXiv 30 Jun 2024.
Builds on
35 of 72 references.
Closest ancestors · 9
Same direction, newest first
- 2024Divide-and-Conquer Posterior Sampling for Denoising Diffusion Priors18 Mar
- 2024Diffusion Posterior Sampling is Computationally Intractable20 Feb
- 2023A Variational Perspective on Solving Inverse Problems with Diffusion Models7 May
- 2023Score-Based Diffusion Models as Principled Priors for Inverse Imaging23 Apr
- 2022Diffusion Model Based Posterior Sampling for Noisy Linear Inverse Problems20 Nov
- 2022DOLPH: Diffusion Models for Phase Retrieval1 Nov
- 2022Diffusion Posterior Sampling for General Noisy Inverse Problems29 Sep
- 2022Denoising Diffusion Restoration Models27 Jan
- 2021Come-Closer-Diffuse-Faster: Accelerating Conditional Diffusion Models for Inverse Problems through Stochastic Contraction9 Dec
Other references · 26
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2021Palette: Image-to-Image Diffusion Modelscited by 1,296
- 2021Maximum Likelihood Training of Score-Based Diffusion Modelscited by 670
- 2021ILVR: Conditioning Method for Denoising Diffusion Probabilistic Modelscited by 634
- 2021Solving Inverse Problems in Medical Imaging with Score-Based Generative Modelscited by 514
- 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptionscited by 440
- 2021Robust Compressed Sensing MRI with Deep Generative Priorscited by 284
- 2022Diffusion models as plug-and-play priorscited by 230
- 2023Nearly d-Linear Convergence Bounds for Diffusion Models via Stochastic Localizationcited by 230
14 more
- 2023Practical and Asymptotically Exact Conditional Sampling in Diffusion Modelscited by 218
- 2022Improved Analysis of Score-based Generative Modeling: User-Friendly Bounds under Minimal Smoothness Assumptionscited by 215
- 2021SNIPS: Solving Noisy Inverse Problems Stochasticallycited by 191
- 2022Convergence of score-based generative modeling for general data distributionscited by 179
- 2022Convergence for score-based generative modeling with polynomial complexitycited by 176
- 2015A Complete Recipe for Stochastic Gradient MCMCcited by 146
- 2023The probability flow ODE is provably fastcited by 145
- 2019Notes on Computational Hardness of Hypothesis Testing: Predictions using the Low-Degree Likelihood Ratiocited by 82
- 2018Sampling Can Be Faster Than Optimizationcited by 68
- 2022Sampling from the Sherrington-Kirkpatrick Gibbs measure via algorithmic stochastic localizationcited by 48
- 2024Learning general Gaussian mixtures with efficient score matchingcited by 43
- 2022Sudakov-Fernique post-AMP, and a new proof of the local convexity of the TAP free energycited by 9
- 2024Listening to the Noise: Blind Denoising with Gibbs Diffusioncited by 5
- 2023Optimality of Glauber dynamics for general-purpose Ising model sampling and free energy approximationcited by 2
Built on by
Citing papers per year: 2024: 6; 2025: 4; 2026: 2.
12 papers.
- 2024Adjoint Matching: Fine-tuning Flow and Diffusion Generative Models with Memoryless Stochastic Optimal Controlcited by 222
- 2024NETS: A Non-Equilibrium Transport Samplercited by 67
- 2025DriftLite: Lightweight Drift Control for Inference-Time Scaling of Diffusion Modelscited by 21
- 2024Taming Diffusion Models for Image Restoration: A Reviewcited by 12
- 2024Annealing Flow Generative Models Towards Sampling High-Dimensional and Multi-Modal Distributionscited by 11
- 2024Posterior sampling via Langevin dynamics based on generative priorscited by 5
- 2025Efficient Approximate Posterior Sampling with Annealed Langevin Monte Carlocited by 3
- 2024HJ-sampler: A Bayesian sampler for inverse problems of a stochastic process by leveraging Hamilton-Jacobi PDEs and score-based generative modelscited by 2
- 2026SURGE: Approximation and Training Free Particle Filter for Diffusion Surrogatecited by 2
- 2025Exploiting the Exact Denoising Posterior Score in Training-Free Guidance of Diffusion Modelscited by 1
- 2026Diffusion-Based Posterior Sampling: A Feynman-Kac Analysis of Bias and Stabilitycited by 1
- 2025Briding Diffusion Posterior Sampling and Monte Carlo methods: a surveycited by 0
- 2026SURGE: Approximation and Training Free Particle Filter for Diffusion Surrogate18 May 2026
- 2026Diffusion-Based Posterior Sampling: A Feynman-Kac Analysis of Bias and Stability7 May 2026
- 2025Briding Diffusion Posterior Sampling and Monte Carlo methods: a survey15 Oct 2025
- 2025DriftLite: Lightweight Drift Control for Inference-Time Scaling of Diffusion Models25 Sep 2025
- 2025Efficient Approximate Posterior Sampling with Annealed Langevin Monte Carlo11 Aug 2025
- 2025Exploiting the Exact Denoising Posterior Score in Training-Free Guidance of Diffusion Models16 Jun 2025
- 2024NETS: A Non-Equilibrium Transport Sampler3 Oct 2024
- 2024Posterior sampling via Langevin dynamics based on generative priors2 Oct 2024
- 2024Annealing Flow Generative Models Towards Sampling High-Dimensional and Multi-Modal Distributions30 Sep 2024
- 2024Taming Diffusion Models for Image Restoration: A Review16 Sep 2024
- 2024HJ-sampler: A Bayesian sampler for inverse problems of a stochastic process by leveraging Hamilton-Jacobi PDEs and score-based generative models15 Sep 2024
- 2024Adjoint Matching: Fine-tuning Flow and Diffusion Generative Models with Memoryless Stochastic Optimal Control13 Sep 2024
Similar, not cited
Close in wording, no citation either way
- 2025Solving Diffusion Inverse Problems with Restart Posterior Sampling
- 2025ReGuidance: A Simple Diffusion Wrapper for Boosting Sample Quality on Hard Inverse Problems
- 2025A Mixture-Based Framework for Guiding Diffusion Models
- 2025Exact Evaluation of the Accuracy of Diffusion Models for Inverse Problems with Gaussian Data Distributions
- 2023Conditional score-based diffusion models for Bayesian inference in infinite dimensions
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problems
- 2023Removing Structured Noise with Diffusion Models
- 2024Variational Diffusion Posterior Sampling with Midpoint Guidance