A Finite-Particle Convergence Rate for Stein Variational Gradient Descent
Jiaxin Shi, Lester Mackey
Abstract
We provide the first finite-particle convergence rate for Stein variational gradient descent (SVGD), a popular algorithm for approximating a probability distribution with a collection of particles. Specifically, whenever the target distribution is sub-Gaussian with a Lipschitz score, SVGD with n particles and an appropriate step size sequence drives the kernel Stein discrepancy to zero at an order 1/sqrt(log log n) rate. We suspect that the dependence on n can be improved, and we hope that our explicit, non-asymptotic proof strategy will serve as a template for future refinements.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 6f5433bf-050b-48e2-baaa-ef82ac4d531eCited by top-tier papers13
- Forward-Backward Gaussian Variational Inference via JKO in the Bures-Wasserstein SpaceMichael Ziyang Diao, Krishna Balasubramanian, Sinho Chewi, Adil SalimICML 2023 · 47 citations
- Towards Understanding the Dynamics of Gaussian-Stein Variational Gradient DescentTianle Liu, Promit Ghosal, Krishnakumar Balasubramanian, Natesh S. PillaiNeurIPS 2023 · 19 citations
- Improving Adversarial Robustness Through the Contrastive-Guided Diffusion ProcessYidong Ouyang, Liyan Xie, Guang ChengICML 2023 · 11 citations
- Provably Fast Finite Particle Variants of SVGD via Virtual Particle Stochastic ApproximationAniket Das, Dheeraj NagarajNeurIPS 2023 · 10 citations
- Coin Sampling: Gradient-Based Bayesian Inference without Learning RatesLouis Sharrock, Christopher NemethICML 2023 · 10 citations
Builds on6
- A Non-Asymptotic Analysis for Stein Variational Gradient DescentAnna Korba, Adil Salim, Michael Arbel, Giulia Luise et al.NeurIPS 2020 · 102 citations
- Stochastic Stein DiscrepanciesJackson Gorham, Anant Raj, Lester MackeyNeurIPS 2020 · 40 citations
- A Convergence Theory for SVGD in the Population Limit under Talagrand's Inequality T1Adil Salim, Lukang Sun, Peter RichtárikICML 2022 · 28 citations
- Towards Understanding the Dynamics of Gaussian-Stein Variational Gradient DescentTianle Liu, Promit Ghosal, Krishnakumar Balasubramanian, Natesh S. PillaiNeurIPS 2023 · 19 citations
- Learning Equivariant Energy Based Models with Equivariant Stein Variational Gradient DescentPriyank Jaini, Lars Holdijk, Max WellingNeurIPS 2021 · 14 citations
Related papers
- Improved Finite-Particle Convergence Rates for Stein Variational Gradient DescentSayan Banerjee, Krishna Balasubramanian, Promit GhosalICLR 2025
- Accurate Quantization of Measures via Interacting Particle-based OptimizationLantian Xu, Anna Korba, Dejan SlepcevICML 2022 · 18 citations
- Understanding the Variance Collapse of SVGD in High DimensionsJimmy Ba, Murat A. Erdogdu, Marzyeh Ghassemi, Shengyang Sun et al.ICLR 2022 · 35 citations
- Kernel Stein Discrepancy DescentAnna Korba, Pierre-Cyril Aubin-Frankowski, Szymon Majewski, Pierre AblinICML 2021 · 64 citations
- Long-time asymptotics of noisy SVGD outside the population limitVictor Priser, Pascal Bianchi, Adil SalimICLR 2025
