53

Finite-Particle Rates for Regularized Stein Variational Gradient Descent

Ye He
Krishnakumar Balasubramanian
Sayan Banerjee
Promit Ghosal
Main:12 Pages
Bibliography:2 Pages
Appendix:26 Pages
Abstract

We derive finite-particle rates for the regularized Stein variational gradient descent (R-SVGD) algorithm introduced by He et al. (2024) that corrects the constant-order bias of the SVGD by applying a resolvent-type preconditioner to the kernelized Wasserstein gradient. For the resulting interacting NN-particle system, we establish explicit non-asymptotic bounds for time-averaged (annealed) empirical measures, illustrating convergence in the \emph{true} (non-kernelized) Fisher information and, under a W1I\mathrm{W}_1\mathrm{I} condition on the target, corresponding W1\mathrm{W}_1 convergence for a large class of smooth kernels. Our analysis covers both continuous- and discrete-time dynamics and yields principled tuning rules for the regularization parameter, step size, and averaging horizon that quantify the trade-off between approximating the Wasserstein gradient flow and controlling finite-particle estimation error.

View on arXiv
Comments on this paper