ARM-gradient / ARSM
Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder and reinforcement learning. ICML 2019
☆17Updated 4 years ago
Alternatives and similar repositories for ARSM:
Users that are interested in ARSM are comparing it to the libraries listed below
- Low-variance, efficient and unbiased gradient estimation for optimizing models with binary latent variables. (ICLR 2019)☆28Updated 6 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆37Updated 4 years ago
- Pytorch Implemetation for our NAACL2019 Paper "Riemannian Normalizing Flow on Variational Wasserstein Autoencoder for Text Modeling" http…☆62Updated 4 years ago
- Toy datasets to evaluate algorithms for domain generalization and invariance learning.☆42Updated 3 years ago
- Code for "Training Deep Energy-Based Models with f-Divergence Minimization" ICML 2020☆36Updated last year
- ☆20Updated 4 years ago
- ☆12Updated 5 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- Code for Augment & Reduce, a scalable stochastic algorithm for large categorical distributions☆10Updated 6 years ago
- A variational inference method with accurate uncertainty estimation. It uses a new semi-implicit variational family built on neural netwo…☆53Updated 5 months ago
- MAE: Mutual Posterior-Divergence Regularization for Variational AutoEncoders☆23Updated 5 years ago
- Scalable Training of Inference Networks for Gaussian-Process Models, ICML 2019☆41Updated 2 years ago
- ☆31Updated 4 years ago
- Official PyTorch (Lightning) implementation of the NeurIPS 2020 paper "Efficient Marginalization of Discrete and Structured Latent Variab…☆28Updated 3 years ago
- Explaining a black-box using Deep Variational Information Bottleneck Approach☆46Updated 2 years ago
- A tensorflow implementation of the NIPS 2018 paper "Variational Inference with Tail-adaptive f-Divergence"☆20Updated 6 years ago
- ☆65Updated 7 months ago
- Tensorflow implementation of "Meta Dropout: Learning to Perturb Latent Features for Generalization" (ICLR 2020)☆27Updated 4 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆54Updated last year
- Distributional and Outlier Robust Optimization (ICML 2021)☆26Updated 3 years ago
- Official PyTorch code release for Implicit Gradient Transport, NeurIPS'19☆21Updated 5 years ago
- Implementation of Information Dropout☆39Updated 7 years ago
- ☆49Updated 4 years ago
- Code for "A Spectral Approach to Gradient Estimation for Implicit Distributions" (ICML'18)☆33Updated last year
- Implementation of iterative inference in deep latent variable models☆43Updated 5 years ago
- Humans understand novel sentences by composing meanings and roles of core language components. In contrast, neural network models for nat…☆27Updated 4 years ago
- An implementation of DIP-VAE from the paper "Variational Inference of Disentangled Latent Concepts from Unlabelled Observations" by Kumar…☆24Updated 6 years ago
- Learning To Stop While Learning To Predict☆34Updated 2 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 3 years ago