Augmented Normalizing Flows: Bridging the Gap Between Generative Flows and Latent Variable Models

17 Feb 2020  ·  Chin-wei Huang, Laurent Dinh, Aaron Courville ·

In this work, we propose a new family of generative flows on an augmented data space, with an aim to improve expressivity without drastically increasing the computational cost of sampling and evaluation of a lower bound on the likelihood. Theoretically, we prove the proposed flow can approximate a Hamiltonian ODE as a universal transport map. Empirically, we demonstrate state-of-the-art performance on standard benchmarks of flow-based generative modeling.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Image Generation CelebA 256x256 ANF Huang et al. (2020) bpd 0.72 # 6
Image Generation CIFAR-10 ANF Huang et al. (2020) bits/dimension 3.05 # 36
Image Generation ImageNet 32x32 ANF Huang et al. (2020) bpd 3.92 # 16

Methods


No methods listed for this paper. Add relevant methods here