Improving Generalization and Stability of Generative Adversarial Networks

Abstract

Generative Adversarial Networks (GANs) are one of the most popular tools for learning complex high dimensional distributions. However, generalization properties of GANs have not been well understood. In this paper, we analyze the generalization of GANs in practical settings. We show that discriminators trained on discrete datasets with the original GAN loss have poor generalization capability and do not approximate the theoretically optimal discriminator. We propose a zero-centered gradient penalty for improving the generalization of the discriminator by pushing it toward the optimal discriminator. The penalty guarantees the generalization and convergence of GANs. Experiments on synthetic and large scale datasets verify our theoretical analysis.

ICLR 2019

Cited by in corpus (5)

• Unsupervised anomaly localization using VAE and beta-VAE
• From Shadow Segmentation to Shadow Removal
• Reliable Estimation of Individual Treatment Effect with Causal Information Bottleneck
• Semantic Preserving Generative Adversarial Models
• Improving Generative Adversarial Networks with Local Coordinate Coding