Annealed Stein Variational Gradient Descent

Francesco D'Angelo, Vincent Fortuin

Keywords: stein variational gradient descent, mode-collapse, annealing

TL;DR: We propose an annealing scheme for Stein Variational Gradient Descent that alleviates the mode-collapse problem.

Abstract: Particle based optimization algorithms have recently been developed as sampling methods that iteratively update a set of particles to approximate a target distribution. In particular Stein variational gradient descent has gained attention in the approximate inference literature for its flexibility and accuracy. We empirically explore the ability of this method to sample from multi-modal distributions and focus on two important issues: (i) the inability of the particles to escape from local modes and (ii) the inefficacy in reproducing the density of the different regions. We propose an annealing schedule to solve these issues and show, through various experiments, how this simple solution leads to significant improvements in mode coverage, without invalidating any theoretical properties of the original algorithm.