2021/11/03 by Miguel Lazaro-Gredilla, Miguel Lázaro-Gredilla, Antoine Dedieu +4 · 1 citation
Computer Science · Mathematics · Physics and Astronomy · #Markov Chains and Monte Carlo Methods #Quantum many-body systems #Theoretical and Computational Physics #cs.LG #stat.ML
paper · pdf · doi:10.48550/arxiv.2111.02458
arxiv created 2021/11/05 · arxiv updated 2021/11/08
Perturb-and-MAP offers an elegant approach to approximately sample from a energy-based model (EBM) by computing the maximum-a-posteriori (MAP) configuration of a perturbed version of the model. Sampling in turn enables learning. However, this line of research has been hindered by the general intractability of the MAP computation. Very few works venture outside tractable models, and when they do, they use linear programming approaches, which as we will show, have several limitations. In this work we present perturb-and-max-product (PMP), a parallel and scalable mechanism for sampling and learning in discrete EBMs. Models can be arbitrary as long as they are built using tractable factors. We show that (a) for Ising models, PMP is orders of magnitude faster than Gibbs and Gibbs-with-Gradients (GWG) at learning and generating samples of similar or better quality; (b) PMP is able to learn and sample from RBMs; (c) in a large, entangled graphical model in which Gibbs and GWG fail to mix, PMP succeeds.