English

A New Distribution on the Simplex with Auto-Encoding Applications

Machine Learning 2019-12-17 v3 Machine Learning

Abstract

We construct a new distribution for the simplex using the Kumaraswamy distribution and an ordered stick-breaking process. We explore and develop the theoretical properties of this new distribution and prove that it exhibits symmetry under the same conditions as the well-known Dirichlet. Like the Dirichlet, the new distribution is adept at capturing sparsity but, unlike the Dirichlet, has an exact and closed form reparameterization--making it well suited for deep variational Bayesian modeling. We demonstrate the distribution's utility in a variety of semi-supervised auto-encoding tasks. In all cases, the resulting models achieve competitive performance commensurate with their simplicity, use of explicit probability models, and abstinence from adversarial training.

Keywords

Cite

@article{arxiv.1905.12052,
  title  = {A New Distribution on the Simplex with Auto-Encoding Applications},
  author = {Andrew Stirn and Tony Jebara and David A Knowles},
  journal= {arXiv preprint arXiv:1905.12052},
  year   = {2019}
}

Comments

15 pages, 6 figures, 1 tables

R2 v1 2026-06-23T09:30:02.090Z