We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Robust model training and generalisation with Studentising flows

Abstract: Normalising flows are tractable probabilistic models that leverage the power of deep learning to describe a wide parametric family of distributions, all while remaining trainable using maximum likelihood. We discuss how these methods can be further improved based on insights from robust (in particular, resistant) statistics. Specifically, we propose to endow flow-based models with fat-tailed latent distributions such as multivariate Student's $t$, as a simple drop-in replacement for the Gaussian distribution used by conventional normalising flows. While robustness brings many advantages, this paper explores two of them: 1) We describe how using fatter-tailed base distributions can give benefits similar to gradient clipping, but without compromising the asymptotic consistency of the method. 2) We also discuss how robust ideas lead to models with reduced generalisation gap and improved held-out data likelihood. Experiments on several different datasets confirm the efficacy of the proposed approach in both regards.
Comments: 9 pages, 8 figures, accepted for publication at INNF+ 2020 (Second ICML Workshop on Invertible Neural Networks, Normalizing Flows, and Explicit Likelihood Models)
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
MSC classes: 68T07 (Primary), 62F35 (Secondary)
ACM classes: I.2.6; G.3
Cite as: arXiv:2006.06599 [cs.LG]
  (or arXiv:2006.06599v2 [cs.LG] for this version)

Submission history

From: Gustav Eje Henter [view email]
[v1] Thu, 11 Jun 2020 16:47:01 GMT (450kb,D)
[v2] Sat, 11 Jul 2020 12:50:13 GMT (528kb,D)

Link back to: arXiv, form interface, contact.