Augmented Message Passing Stein Variational Gradient Descent

Jiankui Zhou, Yue Qiu

Stein Variational Gradient Descent (SVGD) is a popular particle-based method for Bayesian inference. However, its convergence suffers from the variance collapse, which reduces the…