42 citations · 56 across the 3 of their papers we have counts for
3 papers · 1 filter
Sinkhorn EM: An Expectation-Maximization algorithm based on entropic optimal transport
Gonzalo Mena, Amin Nejatbakhsh, Erdem Varol +1
We study Sinkhorn EM (sEM), a variant of the expectation maximization (EM) algorithm for mixtures based on entropic optimal transport. sEM differs from the classic EM algorithm in…
Learning Latent Permutations with Gumbel-Sinkhorn Networks
Gonzalo Mena, David Belanger, Scott Linderman +1
Permutations and matchings are core building blocks in a variety of latent variable models, as they allow us to align, canonicalize, and sort data. Learning in such models is diffi…
Reparameterizing the Birkhoff Polytope for Variational Permutation Inference
Scott W. Linderman, Gonzalo E. Mena, Hal Cooper +2
Many matching, tracking, sorting, and ranking problems require probabilistic reasoning about possible permutations, a set that grows factorially with dimension. Combinatorial optim…