Geometric GAN
Jae Hyun Lim, Jong Chul Ye
Introduction
Recently, inspired by the success of the deep discriminative models, Goodfellow et al proposed a novel generative model training method called generative adversarial nets (GAN). GAN is formulated as a minimax game between a generative network (generator) that maps a random vector into the data space and a discriminative network (discriminator) trying to distinguish the generated samples from real samples. Unlike the classical generative models such as Variational Auto-Encoders (VAEs) , the minimax formation of GAN can transfer the success of deep discriminative models to generative models, resulting in significant improvement in generative model performance .
Specifically, the original form of the GAN solves the following minmax game:
where is the sample distribution; is the discriminator that takes as input and outputs a scalar between $G(z)zP_{Z}{\mathcal{X}}f$-divergences. Moreover, Maximum Mean Discrepancy objective (MMD) for GAN training was also proposed in .
It is well-known that the training GAN is difficult. In particular, the authors in have identified the following sources of the difficulties: 1) when the discriminator becomes accurate, the gradient for generator vanishes, 2) a popular fixation using a generator gradient updating with is unstable because of the singularity at the denominator when the discriminator is accurate. The main motivation of Wasserstein GAN (W-GAN) was, therefore, to introduce the weight clipping to address the above-described limitations. In fact, Wasserstein GAN is a special instance of minimizing the integral probability metric (IPM) , and Mroueh et al recently generalized the W-GAN for wider function classes and proposed the mean feature matching and/or covariance feature matching GAN (McGAN) using the IPM minimization framework .
Inspired by McGAN, here we propose a novel geometric generalization called geometric GAN. Specifically, geometric GAN is inspired by our novel observation that McGAN is composed of three geometric operations in feature space:
Separating hyperplane search: finding the separating hyperplane for a linear classifier
Discriminator update away from the hyperplane: discriminator parameter update away from the separating hyperplane using stochastic gradient direction (SGD).
Generator update toward the hyperplane: generator parameter update along the normal vector direction of the separating hyperplane using stochastic gradient direction (SGD).
This geometric interpretation proves to be very general, so it can be applied to most of the existing GAN and its variants. Indeed, the main differences between the algorithms come from the construction of the separating hyperplanes for a linear classifier on feature space and the geometric scaling factors for the feature vectors. Based on this observation, we provide new geometric interpretations of GAN , -GAN , EB-GAN , and W-GAN in terms of separating hyperplanes and geometric scaling factors, and discuss their limitations. Furthermore, we propose a novel geometric GAN using the support vector machine (SVM) separating hyperplane that has maximal margin between two classes of separable data . Our numerical experiments clearly showed that the proposed geometric GAN outperforms the existing GANs in all data set.
Related Approaches
In order to introduce the geometric interpretation of GAN and its variants, we begin with the review of the mean feature matching GAN (McGAN) .
Let be a set of bounded real valued functions on the sample space . Suppose that and are two probability distributions on . Then, the integral probability metric (IPM) between and on the function space is defined as follows :
where denotes the expectation with respect to the probability distribution . We can easily show that is non-negative, symmetric and satisfies the triangle inequality. So can be used as a distance measure in the probability space. For example, when is defined as a collection of functions with a finite Lipschitz constant, the IPM is the Wasserstein distance or the earth mover’s distance that forms the basis of the Wasserstein GAN .
In McGAN, the generator network that maps a random input to a target is shown as a block in Fig. 1(a), and the function space under study is defined as follows :
where is a bounded map from to a (often higher-dimensional) feature space . Note that is a symmetric function spaces because if , then . Then, the IPM between and is given by
under the constraints that is bounded and .
Now, given a finite sequence of mini-batch training data set , an empirical estimate of the minmax game that minimizes the is given by :
Then, the discriminator update can be done using a stochastic gradient descent (SGD) :
where is a learning rate. The authors in also used the projection onto the unit ball and weight clipping for and updates, respectively, to meet the constraints. Using the updated , the generator update is then given by :
2 Geometric interpretation of the mean feature matching GAN
The primal form of the update in (6) is geometrically less informative, so here we use the Cauchy-Swartz inequality to obtain a closed-form update for :
where the constant is given by . Given , the corresponding discriminator and generator updates are represented by
Note that the update equations (8) to (10) are equivalent to the dual form of the McGAN , where the authors derived the following minmax problem by using (8):
However, we notice that the explicit representation by Eqs. (8) to (10) gives clearer geometric intuition that plays the key role in designing a geometric GAN, ass will become clear soon.
More specifically, in designing a linear classifier for two class classification problems, (8) is known as the normal vector for the separating hyperplane for the mean difference (MD) classifier . As shown in Fig. 2, once the separating hyperplane is defined, the SGD udpate (9) is to update the discriminator parameters such that the true and fake samples are maximally separately away from the separating hyperplane parallel to the normal vector. On the other hand, the SGD udpate using (10) is to update the generator parameters to make the fake samples approach the separating hyperplane along the normal vector direction (see Fig. 2).
Recall that a linear classifier is defined via the normal vector to the separating hyperplane and an offset . Thus, comparing the direction between two classifiers means comparing their normal vector directions. As shown in Appendix, aside from the geometric scaling factors, the existing GAN and its variants mainly differ in their definition of the normal vector for the separating hyperplane. Based on this observation, in the next section, we discuss an optimal separating hyperplane for generative model training that has the maximal margin.
Geometric GAN
In adversarial training in feature space, a discriminator is interested in discriminating true samples and the fake samples . In practice, the minibatch size is much smaller than the dimension of the feature space , and this type of classification problem is often called the high-dimension low-sample size (HDLSS) problem .
In fact, the mean difference (MD) classifier is one of the popular methods for HDLSS. Specifically, the MD classifier selects the hyperplane that lies half way between the two class means. In particular the normal vector for the seperating hyperplane is given by the difference of the class means:
Note that if the variables are first mean centered then scaled by the standard deviation, then the mean difference is equivalent to the naive Bayes classifier . Furthermore, in HDLSS, there always exists a maximal data piling direction (MDP) , where mulitple points in each class have the identical projection on the line spanned by the normal vector.
On the other hand, the Support Vector Machine (SVM) and its many variants is one of the most widely used and well studied classification algorithms and its robustness has been also proven for HDLSS setup . Although the aforementioned classification algorithms are motivated by fitting a statistical distribution to the data, SVM is motivated by a geometric heuristic that leads directly to an optimization problem: maximize the margin between two classes of separable data. In addition, soft-margin SVM balances two competing objectives to maximize the margin while penalizing points on the wrong side of the margin.
Recently, Carmichael et al investigated the Karush-Kuhn-Tucker conditions to provide rigorous mathematical proof for new insights into the behaviour of soft-margin SVM in the large and small tuning parameter regimes in HDLSS. They revealed that for small tuning parameter, if the number of data in two classes are the same (which is the case in our problem), then the SVM direction becomes exactly the MD direction. In addition, for sufficiently large tuning parameter, the authors showed that soft margin SVM is equivalent to hard margin SVM if the data are separable, and the hard-margin SVM has data piling. Due to this generality of the soft-margin SVM, the proposed geometric GAN is designed based on soft-margin SVM linear classifier.
2 Geometric GAN with SVM hyperplane
Note that soft-margin SVM is designed by adding a tunning parameter and slack variables which allows points to be on the wrong side of the margin . In our problem classifying the true samples versus fake samples, the primal form of soft-margin SVM can be formulated by
Equivalently, the primal form of the soft-margin SVM can be represented using loss + penalty form :
The goal of the SVM optimization (13) is to maximize the margin between the two classes. This implies that the discriminator update can be also easily incorporated with SVM update, because the goal of the discriminator update can be also regarded to maximize the margin between the two classes. More specifically, our optimization problem is given by
for a given generator parameter .
Specifically, in SVM, the normal vector for the optimal separating hyperplane from (13) is given by :
where will be nonzero only for the support vectors, where the set of support vectors now includes all data points on the margin boundary as well as those on the wrong side of the margin boundary (see Fig. 3). More specifically, we define the region between the margin boudaries as shown in Fig. 3(a):
Then, for given and , the cost function (14) then becomes
where are geometric scaling factors defined by
This is because the SVM cost function value is not dependent on the feature vectors outside of the margin boundaries and is now fully determined by the supporting vectors in .
Accordingly, the discriminator update is given by following SGD updates:
In another word, to update the discriminator parameters, we only need to push out the supporting vectors toward the margin boundaries.
On the other hand, generator update requires more geometric intuition. As shown in Fig. 3, the generator update tries to move the fake feature vectors toward the normal vector direction of the separating hyperplane so that they can be classified as the true feature vectors. This means that the generator update should be given by the following minimization problem:
This is because gets smaller as moves toward the upper-left side in Fig. 3 along the normal vector direction . This results in the following SGD updates:
Note that the SGD updates (20) and (23) are strikingly similar to (9) and (10) of McGAN. Aside from the different choice of separating hyperplane by (16), the discriminator update (20) has additional geometric scaling factors . As will be shown in Appendix A, the appearance of the geometric scaling factors is a recurrent theme in geometric interpretation of GAN and its variants, which we believe is fundamental to account for the geometry of the classifiers.
3 Convergence of Geometric GAN
In order to show the convergence of the geometric GAN to a Nash equilibrium, we investigate the behaviour at large sample limit. Specifically, as for a fixed , the soft margin SVM cost in (14) becomes
where and is a linear discriminator in (22) parameterized by . Here, and denote the probability density functions (pdf) for the distribution and , respectively. Similarly, the generator cost function in (21) becomes
Then, the adversarial training between discriminator and generator can be achieved by the following alternating minimization:
Suppose that the optimal solution of the aforementioned adversarial training is a pair . Then, we can prove the following key convergence result.
Suppose that is a minimizer of the alternating minimization of (24) and (25). Then, almost everywhere, and
In the following example, we provide a specific example where the discriminator and generator cost function has close form expressions, which also has intuitive meaning of the minimum value 2 in Theorem 3.1. In particular, we consider an example of learning parallel lines as in the original Wasserstein GAN paper .
where if and if . Thus, the generator cost function becomes
which achieves its minimum at . Then, the corresponding discriminator cost value at is given by
which coincides the results by Theorem 3.1.
In this example, because all the true and fake samples lies on the separating hyperplane. This informs that at the Nash equilibrium of this problem, all the true samples and the fake samples are not separable, which is the desired property of GAN training. However, Theorem 3.1 is only a necessary condition to make the true and fakes sample non-separable. The proof for the sufficiency condition would be very interesting, which is beyond the scope of current paper.
Experimental Results
In order to evaluate the proposed geometric GAN, we perform comparative studies with three representative types of GANs; 1) Jenson-Shannon (GAN) , 2) mean difference in (Wasserstein GAN) , and 3) mean difference in . Here, the behavior of the maximum margin separating hyperplane of the geometric GAN is empirically analyzed against those of the aforementioned approaches.
In addition, to evaluate the dependency of each variants on Lipschitz continuity constraints, the Lipschitz constraints suggested in was also applied to each adversarial training approach. More specifically, the parameters of the final linear layer in discriminator is determined to represent the aforementioned hyperplane properties, whereas the Lipschitz constraints are applied for other network parameters, such as in and in . In this paper, we only consider Lipschitz density constraints in , so we follow to use weight decay on generators and feature space mapping in discriminators.
We test the four hyperplane searching approaches for discriminators, as well as their complementary generator losses, on two dimensional synthetic data. The synthetic data consists of 100K data points generated from a mixture of 25 Gaussians, akin to the data that have been used for describing mode collapsing behaviors of GANs . Specifically, the means of the Gaussians are evenly spaced as a 5 by 5 grid along and axis from -21 to 21. The standard deviation of each normal distribution is 0.316 (so that the variance would be 0.1). The sampled data from the true distribution can be seen in Figure 4 and 5.
For discriminator and generator, a multi-layered fully-connected neural network architecture is used, as described below. RMSprop is used to train these networks, except vanilla GAN (without any Lipschitz constraints). For vanilla GAN, Adam with momentum is used. Base learning rate is set to 0.001. When weight clipping is applied, parameters in feature mapping is clipped within the range of . When weight projection on unit norm is applied, the following rule, described in is used to update any parameter for every iteration. For weight decay, weight decaying parameter is set to 0.001. Batch size is set to 500 for all experiment. For the number of discriminator update and the one of generator update , we set them as 1, i.e. .
Discriminator: ------
Generator: ---------
The results of the experiment with the mixture of 25 Gaussians are illustrated in Figure 4 and 5. Amongst all GAN variants in this experiment, geometric GAN demonstrated the least mode collapsing behavior independently with Lipschitz continuity regularization constraints.
As shown in Fig. 5, under the same Lipschitz density constraints, linear hyperplane approaches demonstrated less mode collapsing behaviors by virtue of consistent gradients unlike nonlinear separating hyperplane of original GAN. However, mean difference-driven hyperplanes in Wasserstein GAN or McGAN led generators to the mean of arbitrary number of modes in true distributions since the characteristics of mean difference. One the other hand, geometric GAN generally showed robust and consistent convergence behavior towards true distributions.
2 Image Datasets
In order to analyze the proposed method on large-scale dataset, the geometric GAN is empirically analyzed on well-studied datasets in the context of adversarial training; MNIST, CelebA, and LSUN datasets. Since consistent quantitative measures are still under debate, we only perform qualitative comparisons of generated samples from the learned generators of the propsed method against the results of previous literatures. In favor of fair comparisons with other adversarial training methods, we adopt the settings from the previous literatures except the hyperparameters of stochastic optimizations and the tuning parameter of the proposed method.
The DCGAN neural network architectures was used, including batch normalization for generator. Note that the currently known adversarial training methods that demonstrated stable learning without batch normalization have resorted to Lipschitz constraints; therefore, it can also be applied to other adversarial training criterions, including geometric GAN, in order to train batch normalization-free generators.
Each pixel value in input image was rescaled to $(K_{d}=1,K_{g}=10)C$ for discriminator is set to 1.
Specifically for MNIST dataset, input images were resized to 64 by 64 pixels in order to use the same DCGAN network architecture, and the number of epochs for training was set to 20. For CelebA dataset, input images were resized to 96 by 96 pixels and center-cropped with 64 by 64 pixels, and the number of epochs for training is set to 50. For LSUN dataset, only bedroom dataset is used, and an input image is resized to 64 by 64 pixels. The number of epochs for training is set to 2 for LSUN dataset.
The results in Figure 6, 7, and 8 clearly show that the geometric GAN generates very realistic images without mode collapsing or divergent behaviours.
Conclusion
This paper proposed a novel geometric GAN using SVM separating hyperplane, based on geometric intuitions revealed from previous adversarial training approaches. The geometric GAN was based on SVM separating hyperplanes that has the maximal margins between the two classes. Compared to the most of the existing approaches that are based on statistical design criterion, the geometric GAN is derived based on geometric intuition similar to the derivation of SVM. Extensive numerical experiments showed that the proposed method has demonstrated less mode collapsing and more stable training behavior. Moreover, our theoretical results showed that the proposed algorithm converges to the Nash equilibrium between the discriminator and generator, which has also geometric meaning.
Acknowledgement
This work is supported by Korea Science and Engineering Foundation, Grant number NRF2016R1A2B3008104. The first author would like to thank Yunhun Jang for helpful discussions.
References
Appendix A Geometric interpretation of GAN and its variants
This appendix provides geometric interpretation of GAN and its variants. In particular, we consider a specific form of the discriminator given by
where is an output activation function, is the output layer composed of linear layer and the convolutional neural network below corresponding to . Under this choice of the discriminator, we will show that the differences between existing approaches come from the choice of the separating hyperplanes and geometric scaling factors.
Recall that the empirical estimate of the GAN cost in (2) is given by :
We now define geometric scaling factors for true and synthetic (or fake) feature vectors:
In particular, if the activation function is the sigmoid, i.e. , then we can easily see that
Then, the separating hyperplane update is given by:
Using another application of chain rules,
Aside from different choice of separating hyperplane, the only difference is that the features vectors needs to be scaled appropriated using geometric scaling parameters. In fact, the scale parameter is directly related to the geometry of the underlying curved feature spaces due to the and nonlinear activations.
From (29), we can easily see that as discriminator becomes accurate, we have , so the update of the generator becomes more difficult. This is the main technical limitation of the GAN training.
A.2 f𝑓f-GAN
The -GAN formulation is given by the minmax game of the following empirical cost:
where is the convex conjugate of the divergence function . We again define a geometric scale factors for true and fake feature vectors:
The explicit forms of the geometric scaling factors for different -divergences are‘ summarized in Table 1,
Then, the separating hyperplane update is given by:
From the chain rules, we have discriminator and generator update rules:
Note that -GAN is only different from each other in their construction of the weight coefficient (see Table 1) that reflects the underlying geometry of the curved feature space. Other than the total variation-based divergence, the scaling factors are asymmetric. Thus, controlling the balance between discriminator and generator updates are one of the important technical issues of -GAN training.
A.3 Wasserstein GAN
Wasserstein GAN minimizes the following IPM:
where is called the Lipschitz seminorm of a real-valued function on . Using the discriminator model (26), the Wasserstein GAN update can be written by:
Therefore, other than the mean difference on ball for the hyperplane normal vector update, the W-GAN update is same as the mean matching GAN update with geometric scaling factor .
A.4 Energy-based GAN
For a given a positive margin , the energy-based GAN (EBGAN) is given by the alternating minimization of the discriminator and generator cost functions :
where . For a function with , its subgradient is given by:
Due to the margin, geometric scale factors for true and fake feature vectors should be defined accordingly. More specifically, we have
Then, the separating hyperplane update is given by:
It is worthy to note that the introduction of margin appears similar to our geometric GAN with SVM hyperplane. In particular, when a linear activation function is used, we have , the update equations (35) and (36) appears very similar to (20) and (23), respectively. However, there exists fundamental differences. First, in EB-GAN, only the fake samples outside the margins are excluded for the hyperplane and discriminator updates. On the other hand, in geometric GAN, both the true and fake samples outside the margins are excluded for the hyperplane and discriminator updates. The symmetric exclusion in geometric GAN is observed to make the algorithm more robust to outliers. Second, in EBGAN, the margin is defined for the discriminator values. On the other hand, in geometric GAN, the margin is determined by the geometric distance between the feature vectors. Therefore, it is much easier to rely on geometric intuition in designing the geometric GAN.
A.5 Empirical risk minimization
The empirical risk minimization (ERM) with cost is one of the standard method for regression problems. Although the empirical risk minimization (ERM) is rarely used for generator model, our analysis also provides the geometric intuition of ERM update.
Specifically, for a given mini-batch training data set , recall that the empirical risk minimization (ERM) in the feature space is given by
Then, the stochastic gradient for (37) can be represented in the identical form to (7):
which is dependent on the sample index . Therefore, aside from the geometric scaling factors, the main difference comes from the separating hyperplane for linear classifiers. More specifically, the hyperplane for geometric GAN is obtained for samples within each mini-batch, while the classifier for regression is optimally designed for each pair of samples.
Appendix B Proof for Theorem 3.1
The proof technique is inspired from that of EB-GAN . We first need the following two lemmas as the extensions of Lemma 1 in .
Let . The minimum of is and is reached at all .
If , then . If , the , whose minimum is achieved at . ∎
For given , The minimum of exist if . More specifically, the minimum of is at if , or at if .
If , then . Thus, at . Similarly, if , then at . For , . If , at since is a decreasing function on . Similarly, if , at since is increasing at . ∎
Since and are lower semi-continuous functions, has a finite value for the optimal solution . Moreover, due to the alternating minimization, the pair satisfies:
From Lemma B.2, we know that 1) when , the term within the integral achieves its minimum value of at , or 2) when , the term within the integral achieves its minimum value of at . Therefore,
where the last inequality comes from for all .
Second, we will show that . Because (42) holds for arbitrary pdf , we have
where the last inequality comes from . Now, by adding \int{p_{g_{\theta}^{*}}(x)\big{[}1+D^{*}(x)\big{]}_{+}}dx on both sides, we have:
From Lemma B.1, we know that \big{(}1-D^{*}(x)\big{)}+\big{[}1+D^{*}(x)\big{]}_{+}\geq 2. Thus, we have
Finally, the equality in (43) holds if and only if
The above equalities hold if and only if almost everywhere . This concludes the proof.