mlsanityData to 7 Oct 2026

A Unified Framework for Data-Free One-Step Sampling via Wasserstein Gradient Flows

arXiv 2605.17808 ↗First posted 18 May 2026cs.LG stat.MLcited by 0

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

  1. 2015Variational Inference with Normalizing Flowscited by 2,161
  2. 2023Consistency Modelscited by 1,778
  3. 2025Mean Flows for One-step Generative Modelingcited by 524
  4. 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 498
  5. 2021Path Integral Sampler: a stochastic control approach for samplingcited by 175
  6. 2020Stochastic Normalizing Flowscited by 156
  7. 2022An optimal control perspective on diffusion-based generative modelingcited by 145
  8. 2023Denoising Diffusion Samplerscited by 144
  9. 2022Flow Annealed Importance Sampling Bootstrapcited by 124
  10. 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 113
  11. 2023Improved sampling via learned diffusionscited by 103
  12. 2026Generative Modeling via Driftingcited by 98
10 more
  1. 2019Theoretical guarantees for sampling and inference in generative models with latent diffusionscited by 97
  2. 2020Solving high-dimensional Hamilton-Jacobi-Bellman PDEs using neural networks: perspectives from the theory of controlled diffusions and measures on path spacecited by 80
  3. 2020VarGrad: A Low-Variance Gradient Estimator for Variational Inferencecited by 73
  4. 2024NETS: A Non-Equilibrium Transport Samplercited by 67
  5. 2024Improved off-policy training of diffusion samplerscited by 47
  6. 2024Training Neural Samplers with Reverse Diffusive KL Divergencecited by 21
  7. 2022Robust SDE-Based Variational Formulations for Solving Linear PDEs via Deep Learningcited by 13
  8. 2024Neural Sampling from Boltzmann Densities: Fisher-Rao Curves in the Wasserstein Geometrycited by 13
  9. 2024Efficient and Unbiased Sampling of Boltzmann Distributions via Consistency Modelscited by 2
  10. 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flowcited by 1

Built on by

Not cited yet.