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

MaCow: Masked Convolutional Generative Flow

About

Flow-based generative models, conceptually attractive due to tractability of both the exact log-likelihood computation and latent-variable inference, and efficiency of both training and sampling, has led to a number of impressive empirical successes and spawned many advanced variants and theoretical investigations. Despite their computational efficiency, the density estimation performance of flow-based generative models significantly falls behind those of state-of-the-art autoregressive models. In this work, we introduce masked convolutional generative flow (MaCow), a simple yet effective architecture of generative flow using masked convolution. By restricting the local connectivity in a small kernel, MaCow enjoys the properties of fast and stable training, and efficient sampling, while achieving significant improvements over Glow for density estimation on standard image benchmarks, considerably narrowing the gap to autoregressive models.

Xuezhe Ma, Xiang Kong, Shanghang Zhang, Eduard Hovy• 2019

Related benchmarks

TaskDatasetResultRank
Density EstimationCIFAR-10 (test)
Bits/dim3.16
134
Density EstimationImageNet 64x64 (test)
Bits Per Sub-Pixel3.69
62
Generative ModelingCIFAR-10
BPD3.16
46
Density EstimationCIFAR-10
bpd3.16
40
Unconditional Image GenerationCIFAR10
BPD3.16
33
Unconditional Image GenerationImageNet 64
BPD3.69
22
Density EstimationImageNet 64
Bits-per-dimension3.69
16
Generative ModelingImageNet 64x64 downsampled
Bits Per Dimension3.69
13
Density EstimationImageNet 32 x 32
NLL (bits/dim)3.69
12
Density EstimationCelebAHQ 256 x 256 5-bits
NLL (bits/dim)0.67
8
Showing 10 of 16 rows

Other info

Code

Follow for update