English

Gradient penalty from a maximum margin perspective

Machine Learning 2020-11-25 v2 Machine Learning

Abstract

A popular heuristic for improved performance in Generative adversarial networks (GANs) is to use some form of gradient penalty on the discriminator. This gradient penalty was originally motivated by a Wasserstein distance formulation. However, the use of gradient penalty in other GAN formulations is not well motivated. We present a unifying framework of expected margin maximization and show that a wide range of gradient-penalized GANs (e.g., Wasserstein, Standard, Least-Squares, and Hinge GANs) can be derived from this framework. Our results imply that employing gradient penalties induces a large-margin classifier (thus, a large-margin discriminator in GANs). We describe how expected margin maximization helps reduce vanishing gradients at fake (generated) samples, a known problem in GANs. From this framework, we derive a new LL^\infty gradient norm penalty with Hinge loss which generally produces equally good (or better) generated output in GANs than L2L^2-norm penalties (based on the Fr\'echet Inception Distance).

Keywords

Cite

@article{arxiv.1910.06922,
  title  = {Gradient penalty from a maximum margin perspective},
  author = {Alexia Jolicoeur-Martineau and Ioannis Mitliagkas},
  journal= {arXiv preprint arXiv:1910.06922},
  year   = {2020}
}

Comments

Code at https://github.com/AlexiaJM/MaximumMarginGANs

R2 v1 2026-06-23T11:44:32.768Z