Improved Finite-Particle Convergence Rates for Stein Variational Gradient Descent
Sayan Banerjee, Krishna Balasubramanian, Promit Ghosal
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper2
- Stationary MMD PointsZonghao Chen, Toni Karvonen, Heishiro Kanagawa, Francois-Xavier Briol 等ICML 2026
- Long-time asymptotics of noisy SVGD outside the population limitVictor Priser, Pascal Bianchi, Adil SalimICLR 2025
它引用的顶会 Paper10
- A Non-Asymptotic Analysis for Stein Variational Gradient DescentAnna Korba, Adil Salim, Michael Arbel, Giulia Luise 等NeurIPS 2020 · 被引用 102 次
- SVGD as a kernelized Wasserstein gradient flow of the chi-squared divergenceSinho Chewi, Thibaut Le Gouic, Chen Lu, Tyler Maunu 等NeurIPS 2020 · 被引用 92 次
- Stochastic Stein DiscrepanciesJackson Gorham, Anant Raj, Lester MackeyNeurIPS 2020 · 被引用 40 次
- A Finite-Particle Convergence Rate for Stein Variational Gradient DescentJiaxin Shi, Lester MackeyNeurIPS 2023 · 被引用 34 次
- A Convergence Theory for SVGD in the Population Limit under Talagrand's Inequality T1Adil Salim, Lukang Sun, Peter RichtárikICML 2022 · 被引用 28 次
相关 Paper
- Provably Fast Finite Particle Variants of SVGD via Virtual Particle Stochastic ApproximationAniket Das, Dheeraj NagarajNeurIPS 2023 · 被引用 10 次
- Accurate Quantization of Measures via Interacting Particle-based OptimizationLantian Xu, Anna Korba, Dejan SlepcevICML 2022 · 被引用 18 次
- Towards Understanding the Dynamics of Gaussian-Stein Variational Gradient DescentTianle Liu, Promit Ghosal, Krishnakumar Balasubramanian, Natesh S. PillaiNeurIPS 2023 · 被引用 19 次
- Understanding the Variance Collapse of SVGD in High DimensionsJimmy Ba, Murat A. Erdogdu, Marzyeh Ghassemi, Shengyang Sun 等ICLR 2022 · 被引用 35 次
- Particle-based Variational Inference with Generalized Wasserstein Gradient FlowZiheng Cheng, Shiyue Zhang, Longlin Yu, Cheng ZhangNeurIPS 2023 · 被引用 14 次
