English

A Finite-Particle Convergence Rate for Stein Variational Gradient Descent

Machine Learning 2023-11-03 v5 Machine Learning

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.

Keywords

Cite

@article{arxiv.2211.09721,
  title  = {A Finite-Particle Convergence Rate for Stein Variational Gradient Descent},
  author = {Jiaxin Shi and Lester Mackey},
  journal= {arXiv preprint arXiv:2211.09721},
  year   = {2023}
}

Comments

NeurIPS 2023

R2 v1 2026-06-28T06:08:47.599Z