HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

MADE: Masked Autoencoder for Distribution Estimation

Mathieu Germain; Karol Gregor; Iain Murray; Hugo Larochelle

MADE: Masked Autoencoder for Distribution Estimation

Abstract

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.

Code Repositories

eyalbetzalel/pytorch-generative-v2
pytorch
Mentioned in GitHub
mgermain/MADE
Official
pytorch
Mentioned in GitHub
chayanchatterjee/cbc-skynet
tf
Mentioned in GitHub
eyalbetzalel/pytorch-generative
pytorch
Mentioned in GitHub
ritviksrivastava/MADE
tf
Mentioned in GitHub
hardbyte/sorting-gym
Mentioned in GitHub
bthorne93/DustVAEder
tf
Mentioned in GitHub
khangt1k25/MADE
pytorch
Mentioned in GitHub
princetonlips/mam
pytorch
Mentioned in GitHub
naru-project/naru
pytorch
Mentioned in GitHub
ANLGBOY/MADE-with-PyTorch
pytorch
Mentioned in GitHub
chayanchatterjee/gw-skylocator
tf
Mentioned in GitHub
eyalbetzalel/pytorch-generative-v6
pytorch
Mentioned in GitHub
e-hulten/made
pytorch
Mentioned in GitHub
tbrx/compiled-inference
pytorch
Mentioned in GitHub
karpathy/pytorch-made
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
density-estimation-on-uci-gasMADE MoG
Log-likelihood: 8.47
image-generation-on-binarized-mnistMADE 2hl (32 orders)
nats: 86.64

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
MADE: Masked Autoencoder for Distribution Estimation | Papers | HyperAI