A Unified Framework for Data-Free One-Step Sampling via Wasserstein Gradient Flows
Not on the map. Why?
Abstract (arXiv)
We develop a unified theoretical framework for data-free one-step sampling from unnormalized target distributions based on Wasserstein gradient flows. For a broad class of standard f-divergence objectives, we show that the induced velocity field admits the universal form V(x)=w(r(x)),β(x), where β(x)=nabla log (p(x)/q(x)) is shared across objectives and w is determined solely by the choice of divergence. This decomposition shows that standard f-divergence drifts share the same asymptotic target distribution p and differ primarily in how they redistribute transient repair effort across under-covered regions. To formalize this distinction, we derive a one-step regional-response theory for a soft under-coverage functional and obtain a compression--elasticity identity that links divergence choice to the geometry of mass transport into under-covered regions. We further extend the framework beyond the f-divergence family to the Log-Variance (LV) divergence, analyze how the reference distribution alters the resulting drift structure, and motivate a practical LV-inspired surrogate for data-free training. Based on this theory, we instantiate the framework with a KDE-based implementation and describe a complementary normalizing-flow route, enabling one-step inference after training. Experiments on multimodal Gaussian-mixture benchmarks are consistent with the theoretical predictions and demonstrate effective one-step sampling on these targets.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 42 references that are in the map. First posted on arXiv 18 May 2026.
Builds on
22 of 42 references.
References · 22
- 2015Variational Inference with Normalizing Flowscited by 2,161
- 2023Consistency Modelscited by 1,778
- 2025Mean Flows for One-step Generative Modelingcited by 524
- 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 498
- 2021Path Integral Sampler: a stochastic control approach for samplingcited by 175
- 2020Stochastic Normalizing Flowscited by 156
- 2022An optimal control perspective on diffusion-based generative modelingcited by 145
- 2023Denoising Diffusion Samplerscited by 144
- 2022Flow Annealed Importance Sampling Bootstrapcited by 124
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 113
- 2023Improved sampling via learned diffusionscited by 103
- 2026Generative Modeling via Driftingcited by 98
10 more
- 2019Theoretical guarantees for sampling and inference in generative models with latent diffusionscited by 97
- 2020Solving high-dimensional Hamilton-Jacobi-Bellman PDEs using neural networks: perspectives from the theory of controlled diffusions and measures on path spacecited by 80
- 2020VarGrad: A Low-Variance Gradient Estimator for Variational Inferencecited by 73
- 2024NETS: A Non-Equilibrium Transport Samplercited by 67
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024Training Neural Samplers with Reverse Diffusive KL Divergencecited by 21
- 2022Robust SDE-Based Variational Formulations for Solving Linear PDEs via Deep Learningcited by 13
- 2024Neural Sampling from Boltzmann Densities: Fisher-Rao Curves in the Wasserstein Geometrycited by 13
- 2024Efficient and Unbiased Sampling of Boltzmann Distributions via Consistency Modelscited by 2
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flowcited by 1
Built on by
Not cited yet.