A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learning

4 October 2021

Papers citing "A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learning"

50 / 103 papers shown

Title
Advancing Sequential Numerical Prediction in Autoregressive Models Xiang Fei Jinghui Lu Qi Sun Hao Feng Yanjie Wang Wei Shi An-Lan Wang Jingqun Tang Can Huang AI4TS 167 3 0 19 May 2025
Multi-Fidelity Policy Gradient Algorithms Xinjie Liu Cyrus Neary Kushagra Gupta Christian Ellis Ufuk Topcu David Fridovich-Keil OffRL 435 0 0 07 Mar 2025
FedGrAINS: Personalized SubGraph Federated Learning with Adaptive Neighbor Sampling Emir Ceyani Han Xie Baturalp Buyukates Carl Yang Salman Avestimehr FedML 263 0 0 22 Jan 2025
A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentation M. Valiuddin R. V. Sloun C.G.A. Viviers Peter H. N. de With Fons van der Sommen UQCV 258 1 0 25 Nov 2024
Coupling without Communication and Drafter-Invariant Speculative Decoding Majid Daliri Christopher Musco A. Suresh 61 2 0 15 Aug 2024
Causal Dynamic Variational Autoencoder for Counterfactual Regression in Longitudinal Data Mouad El Bouchattaoui Myriam Tami Benoit Lepetit P. Cournède CML OOD 160 2 0 16 Oct 2023
Dynamic Probabilistic Pruning: A general framework for hardware-constrained pruning at different granularities L. Gonzalez-Carabarin Iris A. M. Huijben Bastian Veeling A. Schmid Ruud J. G. van Sloun 45 11 0 26 May 2021
Zero-Shot Text-to-Image Generation Aditya A. Ramesh Mikhail Pavlov Gabriel Goh Scott Gray Chelsea Voss Alec Radford Mark Chen Ilya Sutskever VLM 395 4,941 0 24 Feb 2021
Rao-Blackwellizing the Straight-Through Gumbel-Softmax Gradient Estimator Max B. Paulus Chris J. Maddison Andreas Krause BDL 64 42 0 09 Oct 2020
Event Representation with Sequential, Semi-Supervised Discrete Variables Mehdi Rezaee Francis Ferraro BDL DRL 57 14 0 09 Oct 2020
Interpreting Graph Neural Networks for NLP With Differentiable Edge Masking Michael Schlichtkrull Nicola De Cao Ivan Titov AI4CE 67 218 0 01 Oct 2020
Deep Learning Based Antenna Selection for Channel Extrapolation in FDD Massive MIMO Yindi Yang Shun Zhang Fei Gao Chao Xu Jianpeng Ma O. Dobre 32 21 0 03 Sep 2020
RetrieveGAN: Image Synthesis via Differentiable Patch Retrieval Hung-Yu Tseng Hsin-Ying Lee Lu Jiang Ming-Hsuan Yang Weilong Yang DiffM 3DV 136 54 0 16 Jul 2020
Operation-Aware Soft Channel Pruning using Differentiable Masks Minsoo Kang Bohyung Han AAML 64 140 0 08 Jul 2020
Gradient Estimation with Stochastic Softmax Tricks Max B. Paulus Dami Choi Daniel Tarlow Andreas Krause Chris J. Maddison BDL 71 88 0 15 Jun 2020
Improving Inference for Neural Image Compression Yibo Yang Robert Bamler Stephan Mandt 72 123 0 07 Jun 2020
Learning to Branch for Multi-Task Learning Pengsheng Guo Chen-Yu Lee Daniel Ulbricht 53 179 0 02 Jun 2020
Community Detection Clustering via Gumbel Softmax D. Acharya Huaming Zhang 35 17 0 05 May 2020
Learning Sampling and Model-Based Signal Recovery for Compressed Sensing MRI Iris A. M. Huijben Bastiaan S. Veeling Ruud J. G. van Sloun 75 33 0 22 Apr 2020
Gumbel-softmax-based Optimization: A Simple General Framework for Optimization Problems on Graphs Yaoxin Li Jing Liu Guozheng Lin Y. Hou Muyun Mou Jiang Zhang 6 7 0 14 Apr 2020
FBNetV2: Differentiable Neural Architecture Search for Spatial and Channel Dimensions Alvin Wan Xiaoliang Dai Peizhao Zhang Zijian He Yuandong Tian ... Matthew Yu Tao Xu Kan Chen Peter Vajda Joseph E. Gonzalez 67 291 0 12 Apr 2020
Predictive Sampling with Forecasting Autoregressive Models Auke Wiggers Emiel Hoogeboom BDL 53 16 0 23 Feb 2020
Incremental Sampling Without Replacement for Sequence Models Kensen Shi David Bieber Charles Sutton VLM SyDa BDL 38 25 0 21 Feb 2020
Learning with Differentiable Perturbed Optimizers Quentin Berthet Mathieu Blondel O. Teboul Marco Cuturi Jean-Philippe Vert Francis R. Bach 65 109 0 20 Feb 2020
Decision-Making with Auto-Encoding Variational Bayes Romain Lopez Pierre Boyeau Nir Yosef Michael I. Jordan Jeffrey Regier BDL 393 10,591 0 17 Feb 2020
Estimating Gradients for Discrete Random Variables by Sampling without Replacement W. Kool H. V. Hoof Max Welling BDL 105 50 0 14 Feb 2020
Fast Generating A Large Number of Gumbel-Max Variables Yiyan Qi Peijie Wang Yuanming Zhang Junzhou Zhao Guangjian Tian X. Guan 23 4 0 02 Feb 2020
Paraphrase Generation with Latent Bag of Words Yao Fu Yansong Feng John P. Cunningham BDL 52 92 0 07 Jan 2020
Invertible Gaussian Reparameterization: Revisiting the Gumbel-Softmax Andres Potapczynski Gabriel Loaiza-Ganem John P. Cunningham 55 30 0 19 Dec 2019
Dynamic Convolutions: Exploiting Spatial Sparsity for Faster Inference Thomas Verelst Tinne Tuytelaars 54 152 0 06 Dec 2019
PyTorch: An Imperative Style, High-Performance Deep Learning Library Adam Paszke Sam Gross Francisco Massa Adam Lerer James Bradbury ... Sasank Chilamkurthy Benoit Steiner Lu Fang Junjie Bai Soumith Chintala ODL 493 42,407 0 03 Dec 2019
SAG-VAE: End-to-end Joint Inference of Data Representations and Feature Relations Chen Wang Chengyuan Deng Vladimir A. Ivanov GNN DRL 32 6 0 27 Nov 2019
Feature Selection and Extraction for Graph Neural Networks D. Acharya Huaming Zhang 33 28 0 23 Oct 2019
Deep clustering with concrete k-means Boyan Gao Yongxin Yang Henry Gouk Timothy M. Hospedales 32 17 0 17 Oct 2019
vq-wav2vec: Self-Supervised Learning of Discrete Speech Representations Alexei Baevski Steffen Schneider Michael Auli SSL 153 666 0 12 Oct 2019
Deep Declarative Networks: A New Hope Stephen Gould Leonid Sigal Dylan Campbell AI4CE 66 105 0 11 Sep 2019
Stochastic Filter Groups for Multi-Task CNNs: Learning Specialist and Generalist Convolution Kernels Felix J. S. Bragman Ryutaro Tanno Sebastien Ourselin Daniel C. Alexander M. Jorge Cardoso 60 85 0 26 Aug 2019
Learning Sub-Sampling and Signal Recovery with Applications in Ultrasound Imaging Iris A. M. Huijben Bastiaan S. Veeling K. Janse M. Mischi Ruud J. G. van Sloun 29 52 0 15 Aug 2019
Batch-Shaping for Learning Conditional Channel Gated Networks B. Bejnordi Tijmen Blankevoort Max Welling AI4CE 59 77 0 15 Jul 2019
Latent Multi-Criteria Ratings for Recommendations Pan Li Alexander Tuzhilin 39 23 0 26 Jun 2019
Monte Carlo Gradient Estimation in Machine Learning S. Mohamed Mihaela Rosca Michael Figurnov A. Mnih 67 411 0 25 Jun 2019
Network Pruning via Transformable Architecture Search Xuanyi Dong Yi Yang 3DPC 72 242 0 23 May 2019
Seeker: Real-Time Interactive Search Ari Biswas T. T. Pham Michael Vogelsong Benjamin Snyder Houssam Nassif 13 13 0 17 May 2019
Counterfactual Off-Policy Evaluation with Gumbel-Max Structural Causal Models Michael Oberst David Sontag CML OffRL 71 173 0 14 May 2019
ARSM: Augment-REINFORCE-Swap-Merge Estimator for Gradient Backpropagation Through Categorical Variables Mingzhang Yin Yuguang Yue Mingyuan Zhou 35 23 0 04 May 2019
The Curious Case of Neural Text Degeneration Ari Holtzman Jan Buys Li Du Maxwell Forbes Yejin Choi 184 3,175 0 22 Apr 2019
Modeling Point Clouds with Self-Attention and Gumbel Subset Sampling Jiancheng Yang Qiang Zhang Bingbing Ni Linguo Li Jinxian Liu Mengdie Zhou Qi Tian 3DPC 80 381 0 06 Apr 2019
Stochastic Optimization of Sorting Networks via Continuous Relaxations Aditya Grover Eric Wang Aaron Zweig Stefano Ermon 55 174 0 21 Mar 2019
Stochastic Beams and Where to Find Them: The Gumbel-Top-k Trick for Sampling Sequences Without Replacement W. Kool H. V. Hoof Max Welling 118 219 0 14 Mar 2019
Reparameterizable Subset Sampling via Continuous Relaxations Sang Michael Xie Stefano Ermon BDL 50 99 0 29 Jan 2019