Scalable Stochastic Gradient Riemannian Langevin Dynamics in Non-Diagonal Metrics

Hanlin Yu, Marcelo Hartmann, Bernardo Williams +1

Stochastic-gradient sampling methods are often used to perform Bayesian inference on neural networks. It has been observed that the methods in which notions of differential geometr…