Repository logo
 

Lagrangian Manifold Monte Carlo on Monge Patches

Accepted version
Peer-reviewed

Type

Article

Change log

Authors

Hartmann, M 
Klami, A 

Abstract

The efficiency of Markov Chain Monte Carlo (MCMC) depends on how the underlying geometry of the problem is taken into account. For distributions with strongly varying curvature, Riemannian metrics help in efficient exploration of the target distribution. Unfortunately, they have significant computational overhead due to e.g. repeated inversion of the metric tensor, and current geometric MCMC methods using the Fisher information matrix to induce the manifold are in practice slow. We propose a new alternative Riemannian metric for MCMC, by embedding the target distribution into a higher-dimensional Euclidean space as a Monge patch and using the induced metric determined by direct geometric reasoning. Our metric only requires first-order gradient information and has fast inverse and determinants, and allows reducing the computational complexity of individual iterations from cubic to quadratic in the problem dimensionality. We demonstrate how Lagrangian Monte Carlo in this metric efficiently explores the target distributions.

Description

Keywords

cs.AI, cs.LG, stat.ME, stat.ME

Journal Title

Proceedings of Machine Learning Research

Conference Name

International Conference on Artificial Intelligence and Statistics (AISTATS '22)

Journal ISSN

2640-3498
2640-3498

Volume Title

Publisher

PMLR

Publisher DOI

Sponsorship
Engineering and Physical Sciences Research Council (EP/R034710/1)
Royal Academy of Engineering (RAEng) (RCSRF\1718\6\34)
EPSRC (via University of Warwick) (EP/R034710/1)
EPSRC (EP/V056441/1)
Engineering and Physical Sciences Research Council (EP/V056522/1)