Posterior Sampling by Combining Diffusion Models with Annealed Langevin Dynamics

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xun, Zhiyang, Gupta, Shivam, Price, Eric
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908660115963904
author Xun, Zhiyang
Gupta, Shivam
Price, Eric
author_facet Xun, Zhiyang
Gupta, Shivam
Price, Eric
contents Given a noisy linear measurement $y = Ax + ξ$ of a distribution $p(x)$, and a good approximation to the prior $p(x)$, when can we sample from the posterior $p(x \mid y)$? Posterior sampling provides an accurate and fair framework for tasks such as inpainting, deblurring, and MRI reconstruction, and several heuristics attempt to approximate it. Unfortunately, approximate posterior sampling is computationally intractable in general. To sidestep this hardness, we focus on (local or global) log-concave distributions $p(x)$. In this regime, Langevin dynamics yields posterior samples when the exact scores of $p(x)$ are available, but it is brittle to score--estimation error, requiring an MGF bound (sub-exponential error). By contrast, in the unconditional setting, diffusion models succeed with only an $L^2$ bound on the score error. We prove that combining diffusion models with an annealed variant of Langevin dynamics achieves conditional sampling in polynomial time using merely an $L^4$ bound on the score error.
format Preprint
id arxiv_https___arxiv_org_abs_2510_26324
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Posterior Sampling by Combining Diffusion Models with Annealed Langevin Dynamics
Xun, Zhiyang
Gupta, Shivam
Price, Eric
Machine Learning
Artificial Intelligence
Data Structures and Algorithms
Statistics Theory
Given a noisy linear measurement $y = Ax + ξ$ of a distribution $p(x)$, and a good approximation to the prior $p(x)$, when can we sample from the posterior $p(x \mid y)$? Posterior sampling provides an accurate and fair framework for tasks such as inpainting, deblurring, and MRI reconstruction, and several heuristics attempt to approximate it. Unfortunately, approximate posterior sampling is computationally intractable in general. To sidestep this hardness, we focus on (local or global) log-concave distributions $p(x)$. In this regime, Langevin dynamics yields posterior samples when the exact scores of $p(x)$ are available, but it is brittle to score--estimation error, requiring an MGF bound (sub-exponential error). By contrast, in the unconditional setting, diffusion models succeed with only an $L^2$ bound on the score error. We prove that combining diffusion models with an annealed variant of Langevin dynamics achieves conditional sampling in polynomial time using merely an $L^4$ bound on the score error.
title Posterior Sampling by Combining Diffusion Models with Annealed Langevin Dynamics
topic Machine Learning
Artificial Intelligence
Data Structures and Algorithms
Statistics Theory
url https://arxiv.org/abs/2510.26324