Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Transformation Autoregressive Networks

About

The fundamental task of general density estimation $p(x)$ has been of keen interest to machine learning. In this work, we attempt to systematically characterize methods for density estimation. Broadly speaking, most of the existing methods can be categorized into either using: \textit{a}) autoregressive models to estimate the conditional factors of the chain rule, $p(x_{i}\, |\, x_{i-1}, \ldots)$; or \textit{b}) non-linear transformations of variables of a simple base distribution. Based on the study of the characteristics of these categories, we propose multiple novel methods for each category. For example we proposed RNN based transformations to model non-Markovian dependencies. Further, through a comprehensive study over both real world and synthetic data, we show for that jointly leveraging transformations of variables and autoregressive conditional models, results in a considerable improvement in performance. We illustrate the use of our models in outlier detection and image modeling. Finally we introduce a novel data driven framework for learning a family of distributions.

Junier B. Oliva, Avinava Dubey, Manzil Zaheer, Barnab\'as P\'oczos, Ruslan Salakhutdinov, Eric P. Xing, Jeff Schneider• 2018

Related benchmarks

TaskDatasetResultRank
Unconditional Density EstimationPOWER (test)
Average Test Log Likelihood (nats)0.6
30
Density EstimationGAS d=8; N=1,052,065 (test)
Avg Test Log-Likelihood12.06
25
Density EstimationBSDS300 (test)
NLL (nats)-157
25
Unconditional Density EstimationHEPMASS (test)
NLL (nats)15.12
22
Unconditional Density EstimationMINIBOONE (test)
NLL (nats)11.01
22
Density EstimationHEPMASS UCI (test)
Log-likelihood-13.78
12
Density EstimationMINIBOONE d=43; N=36,488 (test)
Avg Test Log-Likelihood-11.01
11
Density EstimationGAS (test)
Average Log-Likelihood12.06
10
Density EstimationMINIBOON (test)
Avg Log-Likelihood-11.01
10
Density EstimationPOWER d=6; N=2,049,280 (test)
Avg Test Log-Likelihood0.6
8
Showing 10 of 12 rows

Other info

Code

Follow for update