Sampling as optimization in the space of measures: The Langevin dynamics as a composite optimization problem
Not on the map. Why?
Abstract (arXiv)
We study sampling as optimization in the space of measures. We focus on gradient flow-based optimization with the Langevin dynamics as a case study. We investigate the source of the bias of the unadjusted Langevin algorithm (ULA) in discrete time, and consider how to remove or reduce the bias. We point out the difficulty is that the heat flow is exactly solvable, but neither its forward nor backward method is implementable in general, except for Gaussian data. We propose the symmetrized Langevin algorithm (SLA), which should have a smaller bias than ULA, at the price of implementing a proximal gradient step in space. We show SLA is in fact consistent for Gaussian target measure, whereas ULA is not. We also illustrate various algorithms explicitly for Gaussian target measure, including gradient descent, proximal gradient, and Forward-Backward, and show they are all consistent.
Key facts. Cited by 123 papers in the map (data through 7 Oct 2026). First posted on arXiv 22 Feb 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 22; 2023: 21; 2024: 24; 2025: 25; 2026: 17.
123 papers.
Sort:
By area:
- 2022Variational inference via Wasserstein gradient flowscited by 84
- 2024Provably Robust Score-Based Diffusion Posterior Sampling for Plug-and-Play Image Reconstructioncited by 73
- 2022Improved analysis for a proximal algorithm for samplingcited by 72
- 2022Towards a Theory of Non-Log-Concave Sampling: First-Order Stationarity Guarantees for Langevin Monte Carlocited by 63
- 2021Kernel Stein Discrepancy Descentcited by 57
- 2021Optimizing Functionals on the Space of Probabilities with Input Convex Neural Networkscited by 53
- 2021Variational Wasserstein gradient flowcited by 53
- 2022Convex Analysis of the Mean Field Langevin Dynamicscited by 53
- 2025Inference-time Scaling of Diffusion Models through Classical Searchcited by 48
- 2023Faster high-accuracy log-concave sampling via algorithmic warm startscited by 44
- 2021The Mirror Langevin Algorithm Converges with Vanishing Biascited by 37
- 2023Forward-backward Gaussian variational inference via JKO in the Bures-Wasserstein Spacecited by 36
38 more
- 2022Mirror Descent with Relative Smoothness in Measure Spaces, with application to Sinkhorn and EMcited by 33
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraintscited by 33
- 2022Birth-death dynamics for sampling: Global convergence, approximations and their asymptoticscited by 30
- 2021Sqrt(d) Dimension Dependence of Langevin Monte Carlocited by 25
- 2021A Proximal Algorithm for Sampling from Non-smooth Potentialscited by 23
- 2024Implicit Diffusion: Efficient Optimization through Stochastic Samplingcited by 22
- 2021A Probabilistic Interpretation of Self-Paced Learning with Applications to Reinforcement Learningcited by 21
- 2021A Convergence Theory for SVGD in the Population Limit under Talagrand's Inequality T1cited by 21
- 2021Efficient Gradient Flows in Sliced-Wasserstein Spacecited by 21
- 2022A Proximal Algorithm for Samplingcited by 21
- 2022Resolving the Mixing Time of the Langevin Algorithm to its Stationary Distribution for Log-Concave Samplingcited by 21
- 2022Particle-based Variational Inference with Preconditioned Functional Gradient Flowcited by 20
- 2023Sampling via Gradient Flows in the Space of Probability Measurescited by 20
- 2023Algorithms for mean-field variational inference via polyhedral optimization in the Wasserstein spacecited by 20
- 2022Mean-field Variational Inference via Wasserstein Gradient Flowcited by 19
- 2024A Unified and General Framework for Continual Learningcited by 19
- 2021The query complexity of sampling from strongly log-concave distributions in one dimensioncited by 18
- 2022Regularized Stein Variational Gradient Flowcited by 18
- 2022Sampling with Mollified Interaction Energy Descentcited by 17
- 2022Improved Convergence Rate of Stochastic Gradient Langevin Dynamics with Variance Reduction and its Application to Optimizationcited by 16
- 2022Convergence of Stein Variational Gradient Descent under a Weaker Smoothness Conditioncited by 16
- 2023An Explicit Expansion of the Kullback-Leibler Divergence along its Fisher-Rao Gradient Flowcited by 16
- 2023A connection between Tempering and Entropic Mirror Descentcited by 16
- 2024Convergence of coordinate ascent variational inference for log-concave measures via optimal transportcited by 13
- 2024Fisher-Rao Gradient Flow: Geodesic Convexity and Functional Inequalitiescited by 13
- 2021When is the Convergence Time of Langevin Algorithms Dimension Independent? A Composite Optimization Viewpointcited by 12
- 2022Fisher information lower bounds for samplingcited by 12
- 2023Query lower bounds for log-concave samplingcited by 12
- 2023Fast sampling from constrained spaces using the Metropolis-adjusted Mirror Langevin algorithmcited by 11
- 2024Theoretical Guarantees for Variational Inference with Fixed-Variance Mixture of Gaussianscited by 11
- 2025Mixing Time of the Proximal Sampler in Relative Fisher Information via Strong Data Processing Inequalitycited by 11
- 2022Deterministic Langevin Monte Carlo with Normalizing Flows for Bayesian Inferencecited by 10
- 2024Provable Convergence and Limitations of Geometric Tempering for Langevin Dynamicscited by 10
- 2024Fast Convergence of Phi-Divergence Along the Unadjusted Langevin Algorithm and Proximal Samplercited by 10
- 2024Constrained Sampling with Primal-Dual Langevin Monte Carlocited by 10
- 2025Constrained Diffusers for Safe Planning and Controlcited by 9
- 2021Sampling from Log-Concave Distributions with Infinity-Distance Guaranteescited by 8
- 2023Coin Sampling: Gradient-Based Bayesian Inference without Learning Ratescited by 8
50 most cited of 123 All 123 (JSON)
- 2026Wasserstein Gradient Flows and Forward-Only Diffusion Are Not Enough for Multimodal Sampling1 Oct 2026
- 2026First-Order Stationarity of Reverse Diffusions25 Sep 2026
- 2026Preservation of Log-Concavity and Convergence of Wasserstein-Fisher-Rao Gradient Flows16 Sep 2026
- 2026Wasserstein mixing time of the unadjusted Langevin algorithm3 Aug 2026
- 2026Nesterov acceleration in optimizing over probability measures25 Jul 2026
- 2026Difference of Convex Programming in the Wasserstein Space with Applications to MMD Optimization26 Jun 2026
- 2026Recognize Your Orchestrator: An Entropy Dynamics Perspective for LLM Multi-Agent Systems31 May 2026
- 2026Accurate Large-sample Uncertainty Quantification using Stochastic Gradient Markov Chain Monte Carlo29 May 2026
- 2026Improved Guarantees for Langevin Monte Carlo with Average Smoothness29 May 2026
- 2026Sobolev Regularized MMD Gradient Flow12 May 2026
- 2026Consistency Regularised Gradient Flows for Inverse Problems8 May 2026
- 2026Generative models for decision-making under distributional shift6 Apr 2026
38 more
- 2026Deep unfolding of MCMC kernels: scalable, modular & explainable GANs for high-dimensional posterior sampling24 Feb 2026
- 2026Stochastic Gradient Variational Inference with Price's Gradient Estimator from Bures-Wasserstein to Parameter Space21 Feb 2026
- 2026Radon--Wasserstein Gradient Flows for Interacting-Particle Sampling in High Dimensions5 Feb 2026
- 2026Finite-Particle Rates for Regularized Stein Variational Gradient Descent5 Feb 2026
- 2026Functional Stochastic Localization3 Feb 2026
- 2025A Systems-Theoretic View on the Convergence of Algorithms under Disturbances19 Dec 2025
- 2025Worst-case generation via minimax optimization in Wasserstein space9 Dec 2025
- 2025An operator splitting analysis of Wasserstein--Fisher--Rao gradient flows22 Nov 2025
- 2025A Gradient Guided Diffusion Framework for Chance Constrained Programming14 Oct 2025
- 2025Generalization of Gibbs and Langevin Monte Carlo Algorithms in the Interpolation Regime7 Oct 2025
- 2025Dale meets Langevin: A Multiplicative Denoising Diffusion Model3 Oct 2025
- 2025Are We Really Learning the Score Function? Reinterpreting Diffusion Models Through Wasserstein Gradient Flow Matching30 Aug 2025
- 2025Efficient Approximate Posterior Sampling with Annealed Langevin Monte Carlo11 Aug 2025
- 2025From Kinetic Theory to AI: a Rediscovery of High-Dimensional Divergences and Their Properties15 Jul 2025
- 2025Constrained Diffusers for Safe Planning and Control14 Jun 2025
- 2025Assessing the Quality of Denoising Diffusion Models in Wasserstein Distance: Noisy Score and Optimal Bounds11 Jun 2025
- 2025Sequential Monte Carlo approximations of Wasserstein--Fisher--Rao gradient flows6 Jun 2025
- 2025Inference-time Scaling of Diffusion Models through Classical Search29 May 2025
- 2025Accelerated Markov Chain Monte Carlo Algorithms on Discrete States19 May 2025
- 2025Operator-Level Quantum Acceleration of Non-Logconcave Sampling8 May 2025
- 2025Mirror Mean-Field Langevin Dynamics5 May 2025
- 2025Evolution of Gaussians in the Hellinger-Kantorovich-Boltzmann gradient flow29 Apr 2025
- 2025Score-based deterministic density sampling25 Apr 2025
- 2025DDEQs: Distributional Deep Equilibrium Models through Wasserstein Gradient Flows3 Mar 2025
- 2025Langevin Multiplicative Weights Update with Applications in Polynomial Portfolio Management26 Feb 2025
- 2025Guiding Time-Varying Generative Models with Natural Gradients on Exponential Family Manifold11 Feb 2025
- 2025Towards Understanding Gradient Dynamics of the Sliced-Wasserstein Distance via Critical Point Analysis10 Feb 2025
- 2025Mixing Time of the Proximal Sampler in Relative Fisher Information via Strong Data Processing Inequality8 Feb 2025
- 2025Convergence Analysis of the Wasserstein Proximal Algorithm beyond Geodesic Convexity25 Jan 2025
- 2025Computational and Statistical Asymptotic Analysis of the JKO Scheme for Iterative Algorithms to update distributions11 Jan 2025
- 2024Subspace Langevin Monte Carlo18 Dec 2024
- 2024Parallel Simulation for Log-concave Sampling and Score-based Diffusion Models10 Dec 2024
- 2024Linear convergence of proximal descent schemes on the Wasserstein space22 Nov 2024
- 2024Constrained Sampling with Primal-Dual Langevin Monte Carlo1 Nov 2024
- 2024Fast Convergence of Phi-Divergence Along the Unadjusted Langevin Algorithm and Proximal Sampler14 Oct 2024
- 2024Provable Convergence and Limitations of Geometric Tempering for Langevin Dynamics13 Oct 2024
- 2024Provable Accuracy Bounds for Hybrid Dynamical Optimization and Sampling8 Oct 2024
- 2024Stochastic variance-reduced Gaussian variational inference on the Bures-Wasserstein manifold3 Oct 2024
50 newest of 123 All 123 (JSON)
- 2024Provably Robust Score-Based Diffusion Posterior Sampling for Plug-and-Play Image Reconstructioncited by 73
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraintscited by 33
- 2025Constrained Diffusers for Safe Planning and Controlcited by 9
- 2025Assessing the Quality of Denoising Diffusion Models in Wasserstein Distance: Noisy Score and Optimal Boundscited by 8
- 2023Geometry of Score Based Generative Modelscited by 7
- 2024Parallel Simulation for Log-concave Sampling and Score-based Diffusion Modelscited by 6
- 2025Are We Really Learning the Score Function? Reinterpreting Diffusion Models Through Wasserstein Gradient Flow Matchingcited by 5
- 2025Efficient Approximate Posterior Sampling with Annealed Langevin Monte Carlocited by 3
- 2025Dale meets Langevin: A Multiplicative Denoising Diffusion Modelcited by 2
- 2025A Gradient Guided Diffusion Framework for Chance Constrained Programmingcited by 1
- 2026Consistency Regularised Gradient Flows for Inverse Problemscited by 1
- 2026First-Order Stationarity of Reverse Diffusionscited by 0
- 2026First-Order Stationarity of Reverse Diffusions25 Sep 2026
- 2026Consistency Regularised Gradient Flows for Inverse Problems8 May 2026
- 2025A Gradient Guided Diffusion Framework for Chance Constrained Programming14 Oct 2025
- 2025Dale meets Langevin: A Multiplicative Denoising Diffusion Model3 Oct 2025
- 2025Are We Really Learning the Score Function? Reinterpreting Diffusion Models Through Wasserstein Gradient Flow Matching30 Aug 2025
- 2025Efficient Approximate Posterior Sampling with Annealed Langevin Monte Carlo11 Aug 2025
- 2025Constrained Diffusers for Safe Planning and Control14 Jun 2025
- 2025Assessing the Quality of Denoising Diffusion Models in Wasserstein Distance: Noisy Score and Optimal Bounds11 Jun 2025
- 2024Parallel Simulation for Log-concave Sampling and Score-based Diffusion Models10 Dec 2024
- 2024Provably Robust Score-Based Diffusion Posterior Sampling for Plug-and-Play Image Reconstruction25 Mar 2024
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraints28 Feb 2024
- 2023Geometry of Score Based Generative Models9 Feb 2023
- 2022Variational inference via Wasserstein gradient flowscited by 84
- 2022Improved analysis for a proximal algorithm for samplingcited by 72
- 2022Towards a Theory of Non-Log-Concave Sampling: First-Order Stationarity Guarantees for Langevin Monte Carlocited by 63
- 2021Kernel Stein Discrepancy Descentcited by 57
- 2021Optimizing Functionals on the Space of Probabilities with Input Convex Neural Networkscited by 53
- 2021Variational Wasserstein gradient flowcited by 53
- 2022Convex Analysis of the Mean Field Langevin Dynamicscited by 53
- 2023Faster high-accuracy log-concave sampling via algorithmic warm startscited by 44
- 2021The Mirror Langevin Algorithm Converges with Vanishing Biascited by 37
- 2023Forward-backward Gaussian variational inference via JKO in the Bures-Wasserstein Spacecited by 36
- 2022Mirror Descent with Relative Smoothness in Measure Spaces, with application to Sinkhorn and EMcited by 33
- 2022Birth-death dynamics for sampling: Global convergence, approximations and their asymptoticscited by 30
12 most cited of 109 All 109 (JSON)
- 2026Wasserstein Gradient Flows and Forward-Only Diffusion Are Not Enough for Multimodal Sampling1 Oct 2026
- 2026Preservation of Log-Concavity and Convergence of Wasserstein-Fisher-Rao Gradient Flows16 Sep 2026
- 2026Wasserstein mixing time of the unadjusted Langevin algorithm3 Aug 2026
- 2026Nesterov acceleration in optimizing over probability measures25 Jul 2026
- 2026Difference of Convex Programming in the Wasserstein Space with Applications to MMD Optimization26 Jun 2026
- 2026Accurate Large-sample Uncertainty Quantification using Stochastic Gradient Markov Chain Monte Carlo29 May 2026
- 2026Improved Guarantees for Langevin Monte Carlo with Average Smoothness29 May 2026
- 2026Sobolev Regularized MMD Gradient Flow12 May 2026
- 2026Generative models for decision-making under distributional shift6 Apr 2026
- 2026Deep unfolding of MCMC kernels: scalable, modular & explainable GANs for high-dimensional posterior sampling24 Feb 2026
- 2026Stochastic Gradient Variational Inference with Price's Gradient Estimator from Bures-Wasserstein to Parameter Space21 Feb 2026
- 2026Radon--Wasserstein Gradient Flows for Interacting-Particle Sampling in High Dimensions5 Feb 2026
12 newest of 109 All 109 (JSON)