ARM-gradient / ARSMLinks
Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder and reinforcement learning. ICML 2019
☆17Updated 5 years ago
Alternatives and similar repositories for ARSM
Users that are interested in ARSM are comparing it to the libraries listed below
Sorting:
- Low-variance, efficient and unbiased gradient estimation for optimizing models with binary latent variables. (ICLR 2019)☆27Updated 6 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- Reparameterize your PyTorch modules☆71Updated 5 years ago
- Tensorflow implementation of "Meta Dropout: Learning to Perturb Latent Features for Generalization" (ICLR 2020)☆27Updated 5 years ago
- Pytorch Implemetation for our NAACL2019 Paper "Riemannian Normalizing Flow on Variational Wasserstein Autoencoder for Text Modeling" http…☆63Updated 5 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆57Updated 2 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆39Updated 5 years ago
- ☆65Updated last year
- Toy datasets to evaluate algorithms for domain generalization and invariance learning.☆43Updated 4 years ago
- Code for "Training Deep Energy-Based Models with f-Divergence Minimization" ICML 2020☆37Updated 2 years ago
- Learning To Stop While Learning To Predict☆34Updated 3 years ago
- Hybrid Discriminative-Generative Training via Contrastive Learning☆75Updated 2 years ago
- ☆30Updated 5 years ago
- ☆21Updated 5 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆41Updated 2 years ago
- implements optimal transport algorithms in pytorch☆104Updated 3 years ago
- Gradients as Features for Deep Representation Learning☆43Updated 5 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆69Updated 3 years ago
- ☆12Updated 6 years ago
- Official PyTorch (Lightning) implementation of the NeurIPS 2020 paper "Efficient Marginalization of Discrete and Structured Latent Variab…☆27Updated 4 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆47Updated 2 years ago
- ☆50Updated 5 years ago
- Official PyTorch code release for Implicit Gradient Transport, NeurIPS'19☆21Updated 6 years ago
- ☆14Updated 6 years ago
- Official code for ICLR 2020 paper "A Neural Dirichlet Process Mixture Model for Task-Free Continual Learning."☆101Updated 5 years ago
- Code to implement the AND-mask and geometric mean to do gradient based optimization, from the paper "Learning explanations that are hard …☆41Updated 5 years ago
- ☆34Updated 4 years ago
- Explaining a black-box using Deep Variational Information Bottleneck Approach☆46Updated 3 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 5 years ago
- Official code for "In Search of Robust Measures of Generalization" (NeurIPS 2020)☆28Updated 5 years ago