Groupe d’études et de recherche en analyse des décisions

Sampling in combinatorial spaces with SurVAE flow augmented MCMC

Priyank Jaini Huawei Noah’s Ark, Canada

Priyank Jaini

Lien pour le webinaire
Nº du webinaire : 935 2181 2801
Code secret : 007207

Hybrid Monte Carlo is a powerful Markov Chain Monte Carlo method for sampling from complex continuous distributions. However, a major limitation of HMC is its inability to be applied to discrete domains due to the lack of gradient signal. In this work, we introduce a new approach based on augmenting Monte Carlo methods with SurVAE Flows to sample from discrete distributions using a combination of neural transport methods like normalizing flows and variational dequantization, and the Metropolis-Hastings rule. Our method first learns a continuous embedding of the discrete space using a surjective map and subsequently learns a bijective transformation from the continuous space to an approximately Gaussian distributed latent variable. Sampling proceeds by simulating MCMC chains in the latent space and mapping these samples to the target discrete space via the learned transformations. We demonstrate the efficacy of our algorithm on a range of examples from statistics, computational physics and machine learning, and observe improvements compared to alternative algorithms.


Inscrivez-vous à la notification par e-mail des séminaires d'apprentissage automatique efficace du GERAD.