Variance-Reduced Diffusion Sampling via Target Score Identity
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