Stein Variational Gradient Descent With Matrix-Valued Kernels

Dilin Wang, Ziyang Tang, Chandrajit Bajaj +1

Stein variational gradient descent (SVGD) is a particle-based inference algorithm that leverages gradient information for efficient approximate inference. In this work, we enhance…