MADE: Masked Autoencoder for Distribution Estimation

by   Mathieu Germain, et al.

There has been a lot of recent interest in designing neural network models to estimate a distribution from a set of examples. We introduce a simple modification for autoencoder neural networks that yields powerful generative models. Our method masks the autoencoder's parameters to respect autoregressive constraints: each input is reconstructed only from previous inputs in a given ordering. Constrained this way, the autoencoder outputs can be interpreted as a set of conditional probabilities, and their product, the full joint probability. We can also train a single network that can decompose the joint probability in multiple different orderings. Our simple framework can be applied to multiple architectures, including deep ones. Vectorized implementations, such as on GPUs, are simple and fast. Experiments demonstrate that this approach is competitive with state-of-the-art tractable distribution estimators. At test time, the method is significantly faster and scales better than other autoregressive estimators.


Locally Masked Convolution for Autoregressive Models

High-dimensional generative models have many applications including imag...

Neural Autoregressive Distribution Estimation

We present Neural Autoregressive Distribution Estimation (NADE) models, ...

HCNAF: Hyper-Conditioned Neural Autoregressive Flow and its Application for Probabilistic Occupancy Map Forecasting

We introduce Hyper-Conditioned Neural Autoregressive Flow (HCNAF); a pow...

An Improved Training Procedure for Neural Autoregressive Data Completion

Neural autoregressive models are explicit density estimators that achiev...

Hierarchical Multinomial-Dirichlet model for the estimation of conditional probability tables

We present a novel approach for estimating conditional probability table...

Generating equilibrium molecules with deep neural networks

Discovery of atomistic systems with desirable properties is a major chal...

The DEformer: An Order-Agnostic Distribution Estimating Transformer

Order-agnostic autoregressive distribution estimation (OADE), i.e., auto...

Code Repositories


Implementation of Inverse Autoregressive Flows with Keras

view repo