Source-linked AI summary

Towards Principled Methods for Training Generative Adversarial Networks

Martin Arjovsky, Léon Bottou

arXiv:1701.04862v1stat.MLcs.LG

TL;DR

GAN training remains difficult to explain and stabilize, particularly when divergences and discriminator behavior produce unreliable generator updates. The paper studies these dynamics theoretically, proves conditions behind instability, and develops tools and a direction toward softer metrics. Its analysis establishes perfect discrimination with vanishing discriminator gradients under disjoint-support assumptions and motivates alternatives to divergences that saturate.

  • Problem

    GANs remain difficult to train, while unstable behavior lacks sufficient theory and existing remedies rely on modification-sensitive heuristics.

  • Method

    The paper uses rigorous theoretical analysis and proofs of discriminator behavior, then develops bounds and tools for studying softer metrics and distributions.

  • Results

    Under disjoint compact supports, a smooth optimal discriminator can achieve accuracy 1 with zero input gradients on both supports, while the analysis motivates softer metrics when divergences saturate.

  • Takeaways & Limitations

    The paper provides a principled framework for understanding GAN instability and directions for replacing unusably saturated divergence-based analysis with distance-sensitive alternatives.

  • Takeaways & Limitations

    The proposed gradient analysis must backpropagate through noisy generator samples because exact-support gradients can disregard errors on g(Z) and yield meaningless low-cost samples.

Abstract

from arXiv · show

The goal of this paper is not to introduce a single algorithm or method, but to make theoretical steps towards fully understanding the training dynamics of generative adversarial networks. In order to substantiate our theoretical analysis, we perform targeted experiments to verify our assumptions, illustrate our claims, and quantify the phenomena. This paper is divided into three sections. The first section introduces the problem at hand. The second section is dedicated to studying and proving rigorously the problems including instability and saturation that arize when training generative adversarial networks. The third section examines a practical and theoretically grounded direction towards solving these problems, while introducing new tools to study them.

1 INTRODUCTION

GANs produce realistic images and are being applied broadly, but their training remains difficult to explain and stabilize. This paper develops a theoretical account of instability and principled directions for addressing it.

  • GANs have achieved success generating realistic, sharp-looking images and are being applied to semisupervised learning, sequence learning, and 3D modelling.
  • Despite this success, GANs remain remarkably difficult to train, and much existing work searches heuristically for stable architectures.
  • The paper identifies limited theory of unstable GAN behavior and the sensitivity of heuristic approaches as barriers to experimenting with variants and applying them to new domains.
  • GAN generators sample z from a simple prior and transform it through a neural network gθ, with training distinguishing them from related generative models.
  • Traditional likelihood-based modeling minimizes KL divergence, whose asymmetric penalties can strongly discourage missing data regions while weakly penalizing implausible generated samples.
  • The original GAN formulation optimizes the Jensen-Shannon divergence, while the paper asks why generator updates worsen, why training is unstable, and whether these issues can be avoided.
  • The paper answers these questions while introducing tools for analyzing GAN training, proposing a direction designed to avoid instability, and identifying open research problems.

2 SOURCES OF INSTABILITY

GAN training becomes unreliable when real and generated distributions are disjoint or concentrated on differently aligned low-dimensional manifolds: discriminators can become perfect and nearly constant on the supports. As the discriminator improves, the generator’s original-cost gradients either vanish or become unstable under noisy approximation, while the resulting updates can provide no useful feedback or reduce sample quality.

  • 2.1 THE PERFECT DISCRIMINATION THEOREMS: Low-dimensional generator supports make discontinuity plausible because Pg is contained in the image of a lower-dimensional latent space.When the latent dimension is smaller than the data-space dimension, the image typically lies in a union of low-dimensional manifolds with measure 0 in the ambient space.
  • 2.1 THE PERFECT DISCRIMINATION THEOREMS: Perfect discriminators exist when supports are disjoint or lie on differently aligned low-dimensional manifolds, achieving accuracy 1 with zero input gradients almost everywhere.The manifold result assumes continuous distributions on closed, non-full-dimensional manifolds that do not perfectly align.
  • 2.1 THE PERFECT DISCRIMINATION THEOREMS: The discriminator’s constancy on the supports prevents backpropagation from reliably conveying information to the generator.The relevant nonzero-gradient or nonsmooth region is confined to the manifold intersection, which has measure 0 under the stated assumptions.
  • 2.1 THE PERFECT DISCRIMINATION THEOREMS: Divergences can be maximized even when the manifolds lie arbitrarily close, so they may be poor direct tests of similarity for these distributions.The paper notes that samples can look impressively good despite both KL divergences being infinite.
  • 2.2.1 THE ORIGINAL COST FUNCTION: The original generator gradient vanishes as the discriminator approaches optimality, forcing a trade-off between inaccurate discriminator updates and vanishing generator updates.The paper experimentally verifies rapid gradient decay and identifies this approximation-quality trade-off as fundamental to the original cost function.
  • 2.2.1 THE ORIGINAL COST FUNCTION: The derived update cost combines an inverted KL(Pg∥Pr) term with oppositely signed JSD terms, explaining pressure toward realistic samples alongside extensive mode dropping.KL(Pg∥Pr) heavily penalizes fake-looking samples but assigns low cost to missing modes, while the JSD terms push distributions apart in the derived expression.
  • 2.2.1 THE ORIGINAL COST FUNCTION: Under white-noise errors in discriminator values and gradients, each generator-gradient coordinate follows a centered Cauchy distribution with infinite expectation and variance.Even when updates are bounded, their centered distribution gives expected update 0 and therefore no gradient feedback.

3 TOWARDS SOFTER METRICS AND DISTRIBUTIONS

The paper proposes smoothing GAN distributions with input noise and using Wasserstein distance to address perfect discrimination, unreliable generator gradients, and training instability. Noise makes distributions continuous and provides gradients that can be annealed while still targeting the original noiseless distributions.

  • Noise and smooth gradients: Adding continuous noise to discriminator inputs smooths probability mass and breaks the assumptions that produce perfect discrimination and vanishing gradients.The paper also argues that generator backpropagation should pass through noisy samples, so the discriminator responds on a set of positive measure.
  • Smoothing distributions: Theorem 3.1 states that adding an absolutely continuous noise variable makes a manifold-supported distribution absolutely continuous with a density.The density is described as depending on an average distance to points in the original support, weighted by their probability.
  • Noise and smooth gradients: For manifold-supported distributions, Gaussian noise yields generator gradients that move samples toward probable data-manifold points while pushing them away from probable generated samples.The relative weights are determined by whether the noisy data density exceeds the noisy generator density.
  • Smoothing distributions: When nearby manifolds are perturbed by noise, the resulting noisy distributions nearly overlap and their Jensen-Shannon divergence becomes small, unlike the maximized noiseless divergence.The paper notes that this noisy JSD depends on the noise level and therefore is not an intrinsic measure of the original distributions.
  • Softer metrics: The Wasserstein metric decreases smoothly as noise variance decreases, whereas Jensen-Shannon divergence remains maximized regardless of the noise amount in the stated setting.Lemma 4 relates the Wasserstein bound to the variance of the mean-zero noise.
  • Softer metrics: Theorem 3.3 supports annealing noise to control one distance term while training on noisy inputs controls another, removing the need for delicate discriminator–generator training schedules.The proposed procedure aims to obtain smooth gradients while minimizing distance between the original noiseless distributions.

A PROOFS OF THINGS

The appendix proves geometric facts underlying GAN behavior when generator and data distributions lie on lower-dimensional manifolds. Its arguments use manifold intersections, perturbations, measure-zero sets, and absolute continuity.

  • Generator images: For piecewise-linear generator nonlinearities, the generator image is contained in a finite union of linear manifolds.The representation follows by enumerating the diagonal activation-pattern matrices induced by rectifiers or leaky rectifiers.
  • Generator images: Smooth strictly increasing nonlinearities preserve the relevant manifold structure, while affine maps are analyzed through singular-value decomposition, embeddings, and projections.These steps support the claim that composing the transformations preserves a countable union of manifolds of the appropriate dimension.
  • Perturbed manifolds: Under small random perturbations, two manifolds intersect transversally or fail to intersect with probability 1, depending on their combined dimensions.The proof uses the General Position Lemma when dimensions are sufficient and a measure-zero argument when their sum is below the ambient dimension.
  • Manifold intersections: When two non-full-dimensional manifolds do not perfectly align, their intersection has strictly lower dimension and measure zero on both manifolds.With boundaries, the intersection is decomposed into at most four lower-dimensional manifold pieces.
  • Noise and densities: Adding independent absolutely continuous noise to a manifold-supported random variable is shown by measure-theoretic arguments to produce an absolutely continuous distribution with a density.The proof establishes null-set preservation and then identifies the density using independence and the Radon–Nikodym theorem.

B FURTHER CLARIFICATIONS

The clarifications appendix informally explains notation and technical assumptions used throughout the paper. It distinguishes continuous from absolutely continuous variables and clarifies discriminator behavior outside both distribution supports.

  • Notation and probability: A continuous random variable assigns zero probability to every single point, whereas an absolutely continuous variable also assigns zero probability to every Lebesgue-measure-zero set.The paper notes that it often uses “continuous” informally when it means “absolutely continuous” and therefore having a density.
  • Notation and probability: The notation Pr[D(x) = 1] = 1 abbreviates that the discriminator equals 1 on a set having full probability under Pr.Equivalently, the inverse image D^-1(1) has Pr-measure 1.
  • Geometric notation: The distance d(A, B) is the usual distance between sets in the metric space, based on the Euclidean distance d(x, y) between points.This definition is used in the proof of Theorem 2.1.
  • Discriminator interpretation: Points outside both supports need not be generated images and do not affect discriminator performance, because the discriminator distinguishes Pr from Pg rather than Pr from every non-data point.The intended accuracy conditions are D(x)=1 almost surely for x sampled from Pr and D(x)=0 almost surely for x sampled from Pg.
  • Discriminator interpretation: An optimal discriminator is defined as one whose objective is at least as good as that of every measurable discriminator D: X → [0,1].The clarification ties this definition to the paper’s objective L from equation (1).
Loading 1701.04862v1…