Improved Finite-Particle Convergence Rates for Stein Variational Gradient Descent
Sayan Banerjee, Krishna Balasubramanian, Promit Ghosal
Abstract
We provide finite-particle convergence rates for the Stein Variational Gradient Descent (SVGD) algorithm in the Kernelized Stein Discrepancy () and Wasserstein-2 metrics. Our key insight is that the time derivative of the relative entropy between the joint density of particle locations and the -fold product target measure, starting from a regular initial distribution, splits into a dominant negative part' proportional to $N$ times the expected $\mathsf{KSD}^2$ and a smaller positive part'. This observation leads to rates of order , in both continuous and discrete time, providing a near optimal (in the sense of matching the corresponding i.i.d. rates) double exponential improvement over the recent result by Shi and Mackey (2024). Under mild assumptions on the kernel and potential, these bounds also grow polynomially in the dimension . By adding a bilinear component to the kernel, the above approach is used to further obtain Wasserstein-2 convergence in continuous time. For the case of `bilinear + Matérn' kernels, we derive Wasserstein-2 rates that exhibit a curse-of-dimensionality similar to the i.i.d. setting. We also obtain marginal convergence and long-time propagation of chaos results for the time-averaged particle laws.
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.
Cited by top-tier papers2
- Stationary MMD PointsZonghao Chen, Toni Karvonen, Heishiro Kanagawa, Francois-Xavier Briol et al.ICML 2026
- Long-time asymptotics of noisy SVGD outside the population limitVictor Priser, Pascal Bianchi, Adil SalimICLR 2025
Builds on10
- A Non-Asymptotic Analysis for Stein Variational Gradient DescentAnna Korba, Adil Salim, Michael Arbel, Giulia Luise et al.NeurIPS 2020 · 102 citations
- SVGD as a kernelized Wasserstein gradient flow of the chi-squared divergenceSinho Chewi, Thibaut Le Gouic, Chen Lu, Tyler Maunu et al.NeurIPS 2020 · 92 citations
- Stochastic Stein DiscrepanciesJackson Gorham, Anant Raj, Lester MackeyNeurIPS 2020 · 40 citations
- A Finite-Particle Convergence Rate for Stein Variational Gradient DescentJiaxin Shi, Lester MackeyNeurIPS 2023 · 34 citations
- A Convergence Theory for SVGD in the Population Limit under Talagrand's Inequality T1Adil Salim, Lukang Sun, Peter RichtárikICML 2022 · 28 citations
Related papers
- Provably Fast Finite Particle Variants of SVGD via Virtual Particle Stochastic ApproximationAniket Das, Dheeraj NagarajNeurIPS 2023 · 10 citations
- Accurate Quantization of Measures via Interacting Particle-based OptimizationLantian Xu, Anna Korba, Dejan SlepcevICML 2022 · 18 citations
- Towards Understanding the Dynamics of Gaussian-Stein Variational Gradient DescentTianle Liu, Promit Ghosal, Krishnakumar Balasubramanian, Natesh S. PillaiNeurIPS 2023 · 19 citations
- Understanding the Variance Collapse of SVGD in High DimensionsJimmy Ba, Murat A. Erdogdu, Marzyeh Ghassemi, Shengyang Sun et al.ICLR 2022 · 35 citations
- Particle-based Variational Inference with Generalized Wasserstein Gradient FlowZiheng Cheng, Shiyue Zhang, Longlin Yu, Cheng ZhangNeurIPS 2023 · 14 citations
