Stochastic Gradient MCMC with Repulsive Forces
Victor Gallego Affiliation: Institute of Mathematical Sciences (ICMAT)、Affiliation: SAMSI, Duke University Email: David Rios Insua Affiliation: Institute of Mathematical Sciences (ICMAT)、Affiliation: SAMSI, Duke University Email:
Abstract
We propose a unifying view of two different Bayesian inference algorithms, Stochastic Gradient Markov Chain Monte Carlo (SG-MCMC) and Stein Variational Gradient Descent (SVGD), leading to improved and efficient novel sampling schemes. We show that SVGD combined with a noise term can be framed as a multiple chain SG-MCMC method. Instead of treating each parallel chain independently from others, our proposed algorithm implements a repulsive force between particles, avoiding collapse and facilitating a better exploration of the parameter space. We also show how the addition of this noise term is necessary to obtain a valid SG-MCMC sampler, a significant difference with SVGD. Experiments with both synthetic distributions and real datasets illustrate the benefits of the proposed scheme.
原文 arXiv:1812.00071;中英对照 + 大白话阅读 https://aha.fim.ai/paper/1812.00071v2