English

Variance-Reduced Diffusion Sampling via Target Score Identity

Machine Learning 2026-01-26 v3 Machine Learning

Abstract

We study variance reduction for score estimation and diffusion-based sampling in settings where the clean (target) score is available or can be approximated. Starting from the Target Score Identity (TSI), which expresses the noisy marginal score as a conditional expectation of the target score under the forward diffusion, we develop: (i) a plug-and-play nonparametric self-normalized importance sampling estimator compatible with standard reverse-time solvers, (ii) a variance-minimizing \emph{state- and time-dependent} blending rule between Tweedie-type and TSI estimators together with an anti-correlation analysis, (iii) a data-only extension based on locally fitted proxy scores, and (iv) a likelihood-tilting extension to Bayesian inverse problems. We also propose a \emph{Critic--Gate} distillation scheme that amortizes the state-dependent blending coefficient into a neural gate. Experiments on synthetic targets and PDE-governed inverse problems demonstrate improved sample quality for a fixed simulation budget.

Keywords

Cite

@article{arxiv.2601.01594,
  title  = {Variance-Reduced Diffusion Sampling via Target Score Identity},
  author = {Alois Duston and Tan Bui-Thanh},
  journal= {arXiv preprint arXiv:2601.01594},
  year   = {2026}
}

Comments

Updated to match journal submission and add ACM & MSC class info

R2 v1 2026-07-01T08:50:00.729Z