M ANY P ATHS TO E QUILIBRIUM : GAN S D O N OT N EED TO D ECREASE A D IVERGENCE A T E VERY S TEP
Anonymous authors
Paper under double-blind review
A BSTRACT
Generative adversarial networks (GANs) are a family of generative models that do not minimize a single training criterion.
Unlike other generative models, the data distribution is learned via a game between a generator (the generative model) and a discriminator (a teacher providing training signal) that each minimize their own cost.
GANs are designed to reach a Nash equilibrium at which each player cannot reduce their cost without changing the other players’ parameters.
One useful approach for the theory of GANs is to show that a divergence between the training distribution and the model distribution obtains its minimum value at equilibrium.
Several recent research directions have been motivated by the idea that this divergence is the primary guide for the learning process and that every step of learning should decrease the divergence.
We show that this view is overly restrictive.
During GAN training, the discriminator provides learning signal in situations where the gradients of the divergences between distributions would not be useful.
We provide empirical counterexamples to the view of GAN training as divergence minimization.
Speciﬁcally, we demonstrate that GANs are able to learn distributions in situations where the divergence minimization point of view predicts they would fail.
We also show that gradient penalties motivated from the divergence minimization perspective are equally helpful when applied in other contexts in which the divergence minimization perspective does not predict they would be helpful.
This contributes to a growing body of evidence that GAN training may be more usefully viewed as approaching Nash equilibria via trajectories that do not necessarily minimize a speciﬁc divergence at each step.
I NTRODUCTION
Generative adversarial networks (GANs) (Goodfellow et al., 2014) are generative models based on a competition between a generator network G and a discriminator network D .
The generator network
G represents a probability distribution p model ( x ) .
To obtain a sample from this distribution, we apply the generator network to a noise vector z sampled from p z , that is x = G ( z ) .
Typically, z is drawn from a Gaussian or uniform distribution, but any distribution with sufﬁcient diversity is possible.
The discriminator D ( x ) attempts to distinguish whether an input value x is real (came from the training data) or fake (came from the generator).
The goal of the training process is to recover the true distribution p data that generated the data.
Several variants of the GAN training process have been proposed.
Different variants of GANs have been interpreted as approximately minimizing different divergences or distances between p data and p model .
However, it has been difﬁcult to understand whether the improvements are caused by a change in the underlying divergence or the learning dynamics.
We conduct several experiments to assess whether the improvements associated with new GAN methods are due to the reasons cited in their design motivation.
We perform a comprehensive study of GANs on simpliﬁed, synthetic tasks for which the true p data is known and the relevant distances are straightforward to calculate, to assess the performance of proposed models against baseline methods.
We also evaluate GANs using several independent evaluation measures on real data to better understand new approaches.
Our contributions are:
• We aim to clarify terminology used in recent papers, where the terms “standard GAN,” “regular GAN,” or “traditional GAN” are used without deﬁnition (e.g., (Arjovsky et al., 2017;
Denton et al., 2015; Salimans et al., 2016; Donahue et al., 2016)).
The original GAN paper described two different losses: the “minimax” loss and the “non-saturating” loss, equations (10) and (13) of Goodfellow (2016), respectively.
Recently, it has become important to clarify this terminology, because many of the criticisms of “standard GANs”, e.g. Arjovsky et al.
(2017), are applicable only to the minimax GAN, while the non-saturating GAN is the standard for GAN implementations.
The non-saturating GAN was recommended for use in practice and implemented in the original paper of Goodfellow et al.
(2014), and is the default in subsequent papers (
Radford et al., 2015; Salimans et al., 2016;
Donahue et al., 2016; Nowozin et al., 2016) 1 .
To avoid confusion we will always indicate whether we mean minimax GAN (M-GAN) or non-saturating GAN (NS-GAN).
• We demonstrate that gradient penalties designed in the divergence minimization framework— to improve Wasserstein GANs (Gulrajani et al., 2017) or justiﬁed from a game theory perspective to improve minimax GANs (Kodali et al., 2017)—also improve the non-saturating GAN on both synthetic and real data.
We observe improved sample quality and diversity.
We ﬁnd that non-saturating GANs are able to ﬁt problems that cannot be ﬁt by JensenShannon divergence minimization.
Speciﬁcally, Figure 1 shows a GAN using the loss from the original non-saturating GAN succeeding on a task where the Jensen-Shannon divergence provides no useful gradient.
Figure 2 shows that the non-saturating GAN does not suffer from vanishing gradients when applied to two widely separated Gaussian distributions.
V ARIANTS OF G ENERATIVE A DVERSARIAL N ETWORKS 2. N ON -S ATURATING AND M INIMAX GAN S
In the original GAN formulation (Goodfellow et al., 2014) the output of the discriminator is a probability and the cost function for the discriminator is given by the negative log-likelihood of the binary discrimination task of classifying samples as real or fake:
[Equation]
The theoretical analysis in (Goodfellow et al., 2014) is based on a zero-sum game in which the generator maximizes J ( D ) , a situation that we refer to here as “minimax GANs”.
In minimax GANs the generator attempts to generate samples that have low probability of being fake, by minimizing the objective (2).
However, in practice, Goodfellow et al.
(2014) recommend implementing an alternative cost function that instead ensures that generated samples have high probability of being real, and the generator instead minimizes an alternative objective (3).
[Equation]
We refer to the alternative objective as non-saturating, due to the non-saturating behavior of the gradient (see ﬁgure 2), and was the implementation used in the code of the original paper.
We use the non-saturating objective (3) in all our experiments
As shown in (Goodfellow et al., 2014), whenever D successfully minimizes J ( D ) optimally, maximizing J ( D ) with respect to the generator is equivalent to minimizing the Jensen-Shannon divergence.
Goodfellow et al.
(2014) use this observation to establish that there is a unique Nash equilibrium in function space corresponding to p data = p model .
2. W ASSERSTEIN GAN
Wasserstein GANs (Arjovsky et al., 2017) modify the discriminator to emit an unconstrained real number rather than a probability (analogous to emitting the logits rather than the probabilities used in the original GAN paper).
The cost function for the WGAN then omits the log-sigmoid functions used in the original GAN paper.
The cost function for the discriminator is now:
[Equation]
The cost function for the generator is simply W ( G ) =
− W ( D )
( D, G ) .
When the discriminator is
Lipschitz smooth, this approach approximately minimizes the earth mover’s distance between p data and p model .
To enforce Lipschitz smoothness, the weights of D are clipped to lie within ( − c, c ) where c is some small real number.
2. G RADIENT P ENALTIES FOR G ENERATIVE A DVERSARIAL N ETWORKS
Multiple formulations of gradient penalties have been proposed for GANs.
As introduced in WGANGP (Gulrajani et al., 2017), the gradient penalty is justiﬁed from the perspective of the Wasserstein distance, by imposing properties which hold for an optimal critic as an additional training criterion.
In this approach, the gradient penalty is typically a penalty on the gradient norm, and is applied on a linear interpolation between data points and samples, thus smoothing out the space between the two distributions.
Kodali et al.
(2017) introduce DRAGAN with a gradient penalty from the perspective of regret minimization, by setting the regularization function to be a gradient penalty on points around the data manifold, as in Follow The Regularized Leader (Cesa-Bianchi & Lugosi, 2006), a standard no-regret algorithm.
This encourages the discriminator to be close to linear around the data manifold, thus bringing the set of possible discriminators closer to a convex set, the set of linear functions.
We also note that they used the minimax version of the game to deﬁne the loss, in which the generator maximizes J ( D ) rather than minimizing J ( G ) .
To formalize the above, both proposed gradient penalties of the form:
[Equation]
where p ˆ x is deﬁned as the distribution deﬁned by the sampling process:
[Equation]
As we will note in our experimental section, Kodali et al.
(2017) also reported that mode-collapse is reduced using their version of the gradient penalty.
2.3. N ON - SATURATING GAN WITH G RADIENT P ENALTY
We consider the non-saturating GAN objective (3) supplemented by two gradient penalties: the penalty proposed by Gulrajani et al.
(2017), which we refer to as “GAN-GP”; the gradient penalty proposed by DRAGAN (Kodali et al., 2017), which we refer to as DRAGAN-NS, to emphasize that we use the non-saturating generator loss function.
In both cases, the gradient penalty applies only to the discriminator, with the generator loss remaining unchanged (as deﬁned in Equation 3).
In this setting, the loss of the discriminator becomes:
[Equation]
We consider these GAN variants because:
[Figure]
• We want to assess whether gradient penalties are effective outside their original deﬁning scope.
Namely, we perform experiments to determine whether the beneﬁt obtained by applying the gradient penalty for Wasserstein GANs is obtained from properties of the earth mover’s distance, or from the penalty itself.
Similarly, we evaluate whether the DRAGAN gradient penalty is beneﬁcial outside the minimax GAN setting.
• We want to assess whether the exact form of the gradient penalty matters.
• We compare three models, to control over different aspects of training: same gradient penalty but different underlying adversarial losses (GAN-GP versus WGAN-GP), as well as the same underlying adversarial loss, but different gradient penalties (GAN-GP versus DRAGAN-NS).
We note that we do not compare with the original DRAGAN formulation, which uses the minimax
GAN formulation, since in this work we focus on non-saturating GAN variants.
M ANY P ATHS TO E QUILIBRIUM
The original GAN paper (Goodfellow et al., 2014) used the correspondence between J ( D )
( D ∗ , G ) and the Jensen-Shannon divergence to characterize the Nash equilibrium of minimax GANs.
It is important to keep in mind that there are many ways for the learning process to approach this equilibrium point, and the majority of them do not correspond to gradually reducing the JensenShannon divergence at each step.
Divergence minimization is useful for understanding the outcome of training, but GAN training is not the same thing as running gradient descent on a divergence and
GAN training may not encounter the same problems as gradient descent applied to a divergence.
Arjovsky et al.
(2017) describe the learning process of GANs from the perspective of divergence minimization and show that the Jensen-Shannon divergence is unable to provide a gradient that will bring p data and p model together if both are sharp manifolds that do not overlap early in the learning process.
Following this line of reasoning, they suggest that when applied to probability distributions that are supported only on low dimensional manifolds, the Kullback Leibler (KL), Jensen Shannon (JS) and Total Variation (TV) divergences do not provide a useful gradient for learning algorithms based on gradient descent, the “traditional GANs” is inappropriate for ﬁtting such low dimensional manifolds (“traditional GAN” seems to refer to the minimax version of GANs used for theoretical analysis in the original paper, and there is no explicit statement about whether the argument is intended to apply to the non-saturating GAN implemented in the code accompanying the original
GAN paper).
In Section 4 we show that non-saturating GANs are able to learn on tasks where the data distribution lies on the low dimensional manifold.
− 5 x 0 .0 .0 .0 .0 .
.p
data ( x ) p model ( x )
D ( x ) − 5 x −−−−     12 log(1 − D ( x )) − 12 log D ( x )
We show that non-saturating GANs do not suffer from vanishing gradients for two widely separated
Gaussians in Figure 2.
The fact that the gradient of the recommended loss does not actually vanish explains why GANs with the non-saturating objective (3), are able to bring together two widely separated Gaussian distributions.
Note that the gradient for this loss does not vanish even when the discriminator is optimal .
The discriminator has vanishing gradients but the generator loss ampliﬁes small differences in discriminator outputs to recover strong gradients.
This means it is possible to train the GAN by changing the loss rather than the discriminator.
For the parallel lines thought experiment (Arjovsky et al., 2017)
(see Figure 1), the main problem with the Jensen-Shannon divergence is that it is parameterized in terms of the density function, and the two density functions have no support in common.
Most GANs, and many other models, can solve this problem by parameterizing their loss functions in terms of samples from the two distributions rather than in terms of their density functions.
S YNTHETIC E XPERIMENTS
To assess the learning process of GANs we empirically examine GAN training on pathological tasks where the data is constructed to lie on a low dimensional manifold, and show the model is able to learn the data distribution in cases where using the underlying divergence obtained at optimality would not provide useful gradients.
We then evaluate convergence properties of common GAN variants on this task where the parameters generating the distribution are known.
4. E XPERIMENT I: 1-D D ATA M ANIFOLD AND 1-D GENERATOR
In our ﬁrst experiment, we generate synthetic training data that lies along a one-dimensional line and design a one-dimensional generative model, however, we embed the problem in a higher ddimensional space where d (cid:29) 1 .
This experiment is essentially an implementation of a thought experiment from Arjovsky et al.
Speciﬁcally, in a d -dimensional space, we deﬁne p data by randomly generating parameters deﬁning the distribution once at the beginning of the experiment.
We generate a random b r ∈ R d and random W r ∈ R 1 × d .
Our latent z r ∈ R ∼ N (0 , σ ) where σ is the standard deviation of the normal distribution.
The synthetic training data of m examples is then given by
[Equation]
The real synthetic data is therefore Gaussian distributed on a 1-D surface within the space, where the position is determined by b r and the orientation is determined by W r .
The generator also assumes the same functional form, that is, it is also intrinsically one dimensional,
[Equation]
where b θ ∈ R d and W θ ∈ R 1 × d .
The discriminator is a single hidden layer ReLU network, which is of higher complexity than the generator so that it may learn non-linear boundaries in the space.
This experiment captures the idea of sharp, non-overlapping manifolds that motivate alternative
GAN losses.
Further, because we know the true generating parameters of the training data, we may explicitly test convergence properties of the various methodologies.
4. E XPERIMENT II: 1-D D ATA M ANIFOLD AND OVERCOMPLETE GENERATOR
In our second experiment, the synthetic training data is still the same (lying on a 1-D line) and given by Eq.
but now the generator is overcomplete for this task, and has a higher latent dimension g , where 1 < g ≤ d .
[Equation]
(13) where matrix W θ ∈ R g × d and vector b θ ∈ R d , so that the generator is able to represent a manifold with too high of a dimensionality.
The generator parameterizes a multivariate Gaussian N ( x ; µ, Σ ) with µ = b .
The covariance matrix elements Σ ij = E [ σ 2 ( X i − µ i )
( X j − µ j )]
= σ 2 E
[( X i − µ i )
( X j − µ j )] .
In vector notation, Σ = σ 2 W T W .
4. R ESULTS
To evaluate the convergence of an experimental trial, we report the square Fréchet distance (Fréchet (1957)) between the true data Gaussian distribution and the ﬁtted Gaussian parameters.
In our notation, where the r subscript denotes real data, the θ subscript denotes the generator Gaussian parameters and (cid:107) x (cid:107) 2 is the squared l 2 norm of x , the Fréchet distance is deﬁned as (Dowson &
Landau (1982)):
[Equation]
Every GAN variant was trained for 200000 steps.
For each step, the generator is updated once and the discriminator is updated 5 times.
Throughout the paper, the number of steps will correspond to the number of generator updates.
The main conclusions from our synthetic data experiments are:
• Gradient penalties (both applied near the data manifold, DRAGAN-NS, and at an interpolation between data and samples, GAN-GP) stabilize training and improve convergence (Figures 3, 9, 10).
• Despite the inability of Jensen-Shannon divergence minimization to solve this problem, we ﬁnd that the non-saturating GAN succeeds in converging to the 1D data manifold (Figure 3).
However, in higher dimensions the resulting ﬁt is not as strong as the other methods: Figure 10 shows that increasing the number of dimensions while keeping the learning rate ﬁxed can decrease the performance of the non-saturating GAN model.
• Non-saturating GANs are able to learn data distributions which are disjoint from the training sample distribution at initialization (or another point in training), as demonstrated in Figure 1.
• Updating the discriminator 5 times per generator update does not result in vanishing gradients, when using the non-saturating cost.
However, when scaling the number of discriminator updates to 100 per generator update, non-saturating GANs perform worse than when using a smaller number of updates (1,5, 10). Gradient penalties help here too: GAN-GP scales better with the number of discriminator updates. The results are detailed in Appendix Section A.2.
• An over-capacity generator with the ability to have more directions of high variance than the underlying data is able to capture the data distribution using non-saturating GAN training (Figure 4).
[Figure] [Figure]
4. H YPERPARAMETER SENSITIVITY
We assess the robustness of the considered models by looking at results across hyperparameters for both experiment 1 and experiment 2.
In one setting, we keep the input dimension ﬁxed while varying the learning rate (Figure 9); in another setting, we keep the learning rate ﬁxed, while varying the input dimension (Figure 10).
In both cases, the results are averaged out over 1000 runs per setting, each starting from a different random seed.
We notice that:
• The non-saturating GAN model (with no gradient penalty) is most sensitive to hyperparameters.
• Gradient penalties make the non-saturating GAN model more robust.
• Both Wasserstein GAN formulations are quite robust to hyperparameter changes.
• For certain hyperparameter settings, there is no performance difference between the two gradient penalties for the non-saturating GAN, when averaging across random seeds.
This is especially visible in Experiment 1, when the number of latent variables is 1.
This could be due to the fact that the data sits on a low dimensional manifold, and because the discriminator is a small, shallow network.
[Figure]
R EAL DATA EXPERIMENTS
To assess the effectiveness of the gradient penalty on standard datasets for the non-saturating GAN formulation, we train a non-saturating GAN, a non-saturating GAN with the gradient penalty introduced by (Gulrajani et al., 2017) (denoted by GAN-GP), a non-saturating GAN with the gradient penalty introduced by (Kodali et al., 2017) (denoted by DRAGAN-NS), and a Wasserstein GAN with gradient penalty (WGAN-GP) on three datasets: Color MNIST (Metz et al., 2016) - data dimensionality (28 , 28 , 3) , CelebA (Liu et al., 2015) - data dimensionality (64 , 64 , 3) and CIFAR-10 (Krizhevsky, 2009) - data dimensionality (32 , 32 , 3) , as seen in Figure 6.
For all our experiments we used λ = 10 as the gradient penalty coefﬁcient and used batch normalization (Ioffe & Szegedy, 2015); Kodali et al.
(2017) suggests that batch normalization is not neeeded for DRAGAN, but we found that it also improved our DRAGAN-NS results.
We used the Adam optimizer (Kingma & Ba, 2014) with β 1 = 0 .
5 and β 2 = 0 .
9 and a batch size of 64.
The input data was scaled to be between -1 and 1.
We did not add any noise to the discriminator inputs or activations, as that regularization technique can be interpreted as having the same goal as gradient penalties, and we wanted to avoid a confounding factor.
We trained all Color MNIST models for 100000 steps, and
CelebA and CIFAR-10 models for 200000 steps.
We note that the experimental results on real data for the non-saturating GAN and for the Improved Wasserstein GAN (WGAN-GP) are quoted with permission from an earlier publication by Rosca et al.
We note that the WGAN-GP model was the only model for which we did 5 discriminator updates in real data experiments.
All other models (DCGAN, DRAGAN-NS, GAN-GP) used one discriminator update for generator update.
For all reported results, we sweep over two hyperparameters:
• Learning rates for the discriminator and generator.
Following Radford et al.
, we tried learning rates of 0.0001, 0.0002, 0.0003 for both the discriminator and the generator.
We note that this is consistent with WGAN-GP, where the authors use 0.0002 for CIFARexperiments.
• Number of latents.
For CelebA and CIFAR-10 we try latent sizes 100, 128 and 150, while for Color MNIST we try 10, 50, 75.
5. E VALUATION
Unlike the synthetic case, here we are unable to evaluate the performance of our models relative to the true solution, since that is unknown.
Moreover, there is no single metric that can evaluate the performance of GANs.
We thus complement visual inspection with three metrics, each measuring a different criteria related to model performance.
We use the Inception Score (Salimans et al., 2016) to measure how visually appealing CIFAR-10 samples are, the MS-SSIM metric (Wang et al., 2003;
Odena et al., 2016) to check sample diversity, and an Improved Wasserstein independent critic to assess overﬁtting, as well as sample quality (Danihelka et al., 2017).
For a more detailed explanation of these metrics, we refer to Rosca et al.
In all our experiments, we control over discriminator and generator architectures, using the ones used by DCGAN (Radford et al., 2015) and the original
WGAN paper (Arjovsky et al., 2017) 2 .
We note that the WGAN-GP paper used a different architecture when reporting the Inception Score on CIFAR10, and thus their results are not directly comparable.
For all the metrics, we report both the hyperparameter sensitivity of the model (by showing quartile statistics), as well as the 10 best results according to the metric.
The sample diversity measure needs to be seen in context with the value reported on the test set: too high diversity can mean failure to capture the data distribution.
For all other metrics, higher is better.
5. V ISUAL SAMPLE INSPECTION
By visually inspecting the results of our models, we noticed that applying gradient penalties to the non-saturating GAN results in more stable training across the board.
When training the non-saturating
GAN with no gradient penalty, we did observe cases of severe mode collapse (see Figure 17).
Gradient penalties improves upon that, but we can still observe mode collapse.
Each non-saturating GAN variant with gradient penalty (DRAGAN-NS and GAN-GP) only produced mode collapse on one dataset, see Figure 21).
We also noticed that for certain learning rates, WGAN-GPs fail to learn the data distribution (Figure 22).
For the GAN-GP and DRAGAN-NS models, most hyperparameters produced samples of equal quality - the models are quite robust.
We show samples from the GAN-GP,
DRAGAN-NS and WGAN-GP models in Figures 18, 19 and 20.
5. M ETRICS
We show that gradient penalties make non-saturating GANs more robust to hyperparameter changes.
For this, we report not only the best obtained results, but rather a box plot of the obtained results showing the quartiles obtained by each sweep, along with the top 10 best results explicitly shown in the graph (note that for each model we tried 27 different hyperparameter settings, corresponding to 3 discriminator learning rates, 3 generator learning rates and 3 generator input sizes).
We report two Inception Score metrics for CIFAR-10, one using the standard Inception network used when the metric was introduced (Salimans et al., 2016), trained on the Imagenet dataset, as well as a
VGG style network trained on CIFAR-10 (for details on the architecture
, we refer the reader to
Rosca et al.
We report the former to be compatible with existing literature, and the latter to obtain a more meaningful metric, since the network doing the evaluation was trained on the same at dataset as the one we evaluate, hence the learned features will be more relevant for the task at hand.
When reporting sample diversity, we subtract the average pairwise image similarity (as reported by
MS-SSIM) computed as the mean of the similarity of every pair of images from 5 batches from the test set.
Note that we can only apply this measure to CelebA, since for datasets such as CIFARdifferent classes are represented by very different images, making this metric meaningless across class borders.
Since our models are completely unsupervised, we do not compute the similarity across samples of the same class as in (Odena et al., 2016).
The Inception Score and sample diversity metric results can be seen in Figure 8.
The results obtained using the Independent Wasserstein critic on all datasets can be found in Figure 7.
5. K EY TAKEAWAYS FROM REAL DATA EXPERIMENTS
When analyzing the results obtained by training non-saturating GANs using gradient penalties (GAN-GP and DRAGAN-NS), we notice that:
• Both gradient penalties help when training non-saturating GANs, by making the models more robust to hyperparameters.
• On CelebA, for various hyperparameter settings WGAN-GP fails to learn the data distribution and produces samples that do not look like faces (Figure 22).
This results in a higher sample diversity than the reference diversity obtained on the test set, as reported by our diversity metric - see Figure 8(a) which compares sample diversity for the considered models across hyperparameters.
The same ﬁgure shows that for most hyperparameter values, the WGAN-GP model produces higher diversity than the one obtained on the test set (indicating failure to capture the data distribution), while for most hyperparameters non-saturating GAN variants produce samples with lower diversity than that of the test set (indicating mode collapse).
However, WGAN-GP is closer to the reference value for more hyperparameters, compared to the non-saturating GAN variants.
• Even if we are only interested in the best results (without looking across the hyperparameter sweep), we see that the gradient penalties tend to improve results for non-saturating GANs.
• The non-saturating GAN trained with gradient penalties produces better samples which give better Inception Scores, both when looking at the results obtained from the best set of hyperparameters and when looking at the entire sweep.
• While the non-saturating GAN variants are much faster to train than the WGAN-GP model (since we do only one discriminator update per generator update), they perform similarly to the WGAN-GP model.
Thus, non-saturating GANs with penalties offer a better computation versus performance tradeoff.
When we trained WGAN-GP models in which we update the discriminator only once per generator update, we noticed a decrease in sample quality for all datasets, reﬂected by our reported metrics, as seen in Figure 15.
• When looking at the independent Wasserstein critic results, we see that the WGAN-GP models perform best on Color MNIST and CIFAR-10.
However, on CelebA the Independent Wasserstein Critic can distinguish between validation data examples and samples from the model (see Figure 7(b)).
This is consistent with what we have seen by examining samples: the hyperparameters which result in samples of reduced quality are the same with a reduced negative Wasserstein distance.
• The sample diversity metric and the Independent Wasserstein critic detect mode collapse.
When DRAGAN-NS collapses for two hyperparameter settings, the negative Wasserstein distance reported by the critic for these jobs is low, showing that the critic captures the difference in distributions, and the sample diversity reported for those settings is greatly reduced (Figure 16).
D ISCUSSION
We have shown that viewing the training dynamics of GANs through the lens of the underlying divergence at optimality can be misleading.
On low-dimensional synthetic problems, we showed that non-saturating GANs are able to learn the true data distribution where Jensen-Shannon divergence minimization would fail.
We also showed that gradient penalty regularizers help improve the
[Figure] [Figure]
training dynamics and robustness of non-saturating GANs.
It is worth noting that one of the gradient penalty regularizers was originally proposed for Wasserstein GANs, motivated by properties of the
Wasserstein distance; evaluating non-saturating GANs with similar gradient penalty regularizers helps disentangle the improvements arising from optimizing a different divergence (or distance) and the improvements from better training dynamics.
Comparison between explored gradient penalties: As described in Section 2.3, we have evaluated two gradient penalties on non-saturating GANs.
We now turn our attention to the distinction between the two gradient penalties.
We have already noted that for a few hyperparameter settings,
DRAGAN-NS produced samples with mode collapse, while the GAN-GP model did not.
By looking at the resulting metrics, we note that there is no clear winner between the two types of gradient penalties.
To assess whether the two penalties have a different regularization effect, we also tried applying both (with a gradient penalty coefﬁcient of 10 for both, or of 5 for both), but that did not result in better models.
This could be because the two penalties have a very similar effect, or due to optimization considerations (they might conﬂict with each other).
Other gradient penalties: Besides the gradient penalties explored in this work, several other regularizers have been proposed for stabilizing GAN training.
Roth et al.
(2017) proposed a gradient penalty aiming to smooth the discriminator of f -GANs (including the minimax GAN), which we refer to as f -GAN-GP, inspired by Sønderby et al.
and Arjovsky & Bottou (2017).
Their gradient penalty is different from the ones explored here; speciﬁcally, their gradient penalty is weighted by the square of the discriminator’s probability of real for each data instance and the penalty is applied to data and samples (no noise is added).
In Fisher-GAN (Mroueh & Sercu, 2017), an equality constraint that is added on the magnitude of the output of the discriminator on data as well as samples is directly penalized, as opposed to the magnitude of the discriminator gradients, as in WGAN-GP.
Similar to WGAN-GP, the penalty was introduced in the framework of integral probability metrics, but it can be directly applied to other approaches to GAN training.
Unlike
WGAN-GP, Fisher GAN uses augmented Lagrangians to impose the equality constraint, instead of a penalty method.
To the best of our knowledge, this has not been tried yet and we leave it for future work.
The regularizers assessed in this work (the penalties proposed by DRAGAN and WGAN-GP), as well as others (such as f -GAN-GP and Fisher-GAN) are similar in spirit, but have been proposed from distinct theoretical considerations.
Future study of GAN regularizers will determine how these regularizers interact, and help us understand the mechanism by which they stabilize GAN training and motivate new approaches.
A CKNOWLEDGEMENTS
We thank Ivo Danihelka and Jascha Sohl-Dickstein for helpful feedback and discussions.
A R ESULTS A. S YNTHETIC E XPERIMENTS
We present here more detailed results for our synthetic experiments.
A. T HE EFFECT OF THE NUMBER OF DISCRIMINATOR UPDATES ON GAN AND GAN-GP
In this section we assess the affects of varying the discriminator update count per generator update
We notice that using 100 discriminator updates per generator update results in a bad distribution ﬁt for the non saturating GAN.
GAN-GP scales better with the number of discriminator updates but increasing the number of discriminator updates does not always result in better ﬁts for this model either.
A. R EAL D ATA E XPERIMENTS
We present here generated samples and other evaluation metrics on real data.
[Figure] [Figure]