arrow
Return

Accelerated Stein Variational Gradient Flow

delete2026-01-01
delete0
PRE
AI
V
Viktor Stein *
W
Wuchen Li
DOI:10.1007/978-3-032-03921-7_9delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Stein variational gradient descent (SVGD) is a kernel-based particle method for sampling from a target distribution, e.g., in Bayesian inference. SVGD does not require estimating the gradient of the log-density, which is called score estimation. In practice, SVGD can be slow compared to score-estimation-based sampling algorithms. To design fast and efficient high-dimensional sampling algorithms, we introduce ASVGD, an accelerated SVGD, based on an accelerated gradient flow in a metric space of probability densities following Nesterov's method. We then derive a momentum-based discrete-time sampling algorithm, which evolves a set of particles deterministically. To stabilize the particles position update, we also study a Wasserstein metric regularization. For the generalized bilinear kernel and the Gaussian kernel, toy numerical examples with varied target distributions demonstrate the effectiveness of ASVGD compared to SVGD and other popular sampling methods.
Keywords:
Nesterov's accelerated gradient method
Information geometry
Kernel methods
Score functions
Interacting particle system

Journal

G
GEOMETRIC SCIENCE OF INFORMATION, GSI 2025, PT II
IF:
0
Papers:
41
Citations:
0

Organization

T
Technical University of Berlin
Scholars:
1.3W
Papers: 1.1W
Citations: 18
U
University of South Carolina System
Scholars:
1.5W
Papers: 1.4W
Citations: 27