paper-with-me

홈 › Papers

Lost Relatives of the Gumbel Trick

2017-06-13 · ICML 2017 8 · Matej Balog, Nilesh Tripuraneni, Zoubin Ghahramani, Adrian Weller

The Gumbel trick is a method to sample from a discrete probability distribution, or to estimate its normalizing partition function. The method relies on repeatedly applying a random perturbation to the distribution in a particular way, each time solving for the most likely configuration. We derive an entire family of related methods, of which the Gumbel trick is one member, and show that the new methods have superior properties in several settings with minimal additional computational cost. In particular, for the Gumbel trick to yield computational benefits for discrete graphical models, Gumbel perturbations on all configurations are typically replaced with so-called low-rank perturbations. We show how a subfamily of our new methods adapts to this setting, proving new upper and lower bounds on the log partition function and deriving a family of sequential samplers for the Gibbs distribution. Finally, we balance the discussion by showing how the simpler analytical form of the Gumbel trick enables additional theoretical results.

📄 PDF Abstract BibTeX arXiv:1706.04161

Code (1)

matejbalog/gumbel-relatives 공식 구현

Similar Papers 제목 키워드 기반

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

2021-10-04 · Iris A. M. Huijben, Wouter Kool, Max B. Paulus, Ruud J. G. van Sloun

The Gumbel-max trick is a method to draw a sample from a categorical distribution, given by its unnormalized (log-)probabilities. Over the past years, the machine learning community has proposed several extensions of thi…

BIG-bench Machine Learning

Fast Gumbel-Max Sketch and its Applications

2023-02-10 · Yuanming Zhang, Pinghui Wang, Yiyan Qi, Kuankuan Cheng 외

The well-known Gumbel-Max Trick for sampling elements from a categorical distribution (or more generally a non-negative vector) and its variants have been widely used in areas such as machine learning and information ret…

Information RetrievalRetrieval

Fast Generating A Large Number of Gumbel-Max Variables

2020-02-02 · Yiyan Qi, Pinghui Wang, Yuanming Zhang, Junzhou Zhao 외

The well-known Gumbel-Max Trick for sampling elements from a categorical distribution (or more generally a nonnegative vector) and its variants have been widely used in areas such as machine learning and information retr…

Graph EmbeddingInformation RetrievalRetrieval

Gradient Estimation with Stochastic Softmax Tricks

2020-06-15 · NeurIPS 2020 12 · Max B. Paulus, Dami Choi, Daniel Tarlow, Andreas Krause 외

The Gumbel-Max trick is the basis of many relaxed gradient estimators. These estimators are easy to implement and low variance, but the goal of scaling them comprehensively to large combinatorial distributions is still o…

GumBolt: Extending Gumbel trick to Boltzmann priors

2018-05-18 · NeurIPS 2018 12 · Amir H. Khoshaman, Mohammad H. Amin

Boltzmann machines (BMs) are appealing candidates for powerful priors in variational autoencoders (VAEs), as they are capable of capturing nontrivial and multi-modal distributions over discrete variables. However, non-di…