English

Convergence of Stein Variational Gradient Descent under a Weaker Smoothness Condition

Statistics Theory 2022-06-02 v1 Machine Learning Probability Statistics Theory

Abstract

Stein Variational Gradient Descent (SVGD) is an important alternative to the Langevin-type algorithms for sampling from probability distributions of the form π(x)exp(V(x))\pi(x) \propto \exp(-V(x)). In the existing theory of Langevin-type algorithms and SVGD, the potential function VV is often assumed to be LL-smooth. However, this restrictive condition excludes a large class of potential functions such as polynomials of degree greater than 22. Our paper studies the convergence of the SVGD algorithm for distributions with (L0,L1)(L_0,L_1)-smooth potentials. This relaxed smoothness assumption was introduced by Zhang et al. [2019a] for the analysis of gradient clipping algorithms. With the help of trajectory-independent auxiliary conditions, we provide a descent lemma establishing that the algorithm decreases the KL\mathrm{KL} divergence at each iteration and prove a complexity bound for SVGD in the population limit in terms of the Stein Fisher information.

Keywords

Cite

@article{arxiv.2206.00508,
  title  = {Convergence of Stein Variational Gradient Descent under a Weaker Smoothness Condition},
  author = {Lukang Sun and Avetik Karagulyan and Peter Richtarik},
  journal= {arXiv preprint arXiv:2206.00508},
  year   = {2022}
}
R2 v1 2026-06-24T11:36:00.208Z