https://github.com/priyankjaini/discFlowMH

Pytorch code for Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC
https://github.com/priyankjaini/discFlowMH

Last synced: 3 months ago
JSON representation

Pytorch code for Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC

• Host: GitHub
• URL: https://github.com/priyankjaini/discFlowMH
• Owner: priyankjaini
• License: mit
• Created: 2021-02-23T18:08:31.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2021-03-01T15:58:16.000Z (over 3 years ago)
• Last Synced: 2024-07-04T01:01:52.945Z (4 months ago)
• Language: Python
• Size: 35.2 KB
• Stars: 12
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# SurVAE Augmented MCMC for Combinatorial Spaces

> Official code for [Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC](https://arxiv.org/abs/2102.02374)  

by Priyank Jaini, Didrik Nielsen, Max Welling.

This paper presents a general purpose sampler for discrete distributions using recent advances in neural transport methods like normalizing flows and the Metropolis-Hastings rule. The algorithm 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.

![alt text](https://pbs.twimg.com/media/EtfbV80XIAEXmXU?format=jpg&name=medium)

For more details, see [the paper](https://arxiv.org/abs/2102.02374).

## Contents

* `/ising/`: Code for experiments for the Ising model.

* `/quantized/`: Code for experiments for Quantized regression.

* `/bayesian_regression/`: Code for experiments for Bayesian variable selection

## Reference

    @inproceedings{jaini2021sampling,

        title={Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC},

        author={Jaini, Priyank and Nielsen, Didrik and Welling, Max},

        booktitle={The 24th International Conference on Artificial Intelligence and Statistics, AISTATS},

        year={2021}

    }

    

For any questions, please contact Priyank Jaini ([email protected]) or Didrik Nielsen ([email protected])