Global Convergence to the Equilibrium of GANs using Variational Inequalities
Ian Gemp, Sridhar Mahadevan
Introduction
When minimizing over , it is known that decreases fastest if moves in the direction . In addition, any direction orthogonal to will leave unchanged. In this work, we show that these orthogonal directions that are ignored by gradient descent can be critical in equilibrium problems, which are central to game theory. If each player in a game updates with , can follow a cyclical trajectory, similar to a person riding a merry-go-round (see Figure 1). This toy scenario actually perfectly reflects an aspect of training for a particular machine learning model mentioned below, and is depicted more technically later on in Figure 2. To arrive at the equilibrium point, a person riding the merry-go-round should walk perpendicularly to their direction of travel, taking them directly to the center.
Equilibrium problems have drawn heightened attention in machine learning due to the emergence of the Generative Adversarial Network (GAN) . GANs have served a variety of applications including generating novel images , simulating particle physics , and imitating expert policies in reinforcement learning . Despite this plethora of successes, GAN training remains heuristic.
Deep learning has benefited from an understanding of simpler, more fundamental techniques. For example, multinomial logistic regression formulates learning a multiclass classifier as minimizing the cross-entropy of a log-linear model where class probabilities are recovered via a softmax. The minimization problem is convex and is solved efficiently with guarantees using stochastic gradient descent (SGD). Unsurprisingly, the majority of deep classifiers incorporate a softmax at the final layer, minimize a cross-entropy loss, and train with a variant of SGD. This progression from logistic regression to classification with deep neural nets is not mirrored in GANs. In contrast, from their inception, GANs were architected with deep nets. Only recently has the Wasserstein Linear-Quadratic GAN (LQ-GAN) been proposed as a minimal model for understanding GANs.
In this work, we analyze the convergence of several GAN training algorithms in the LQ-GAN setting. We survey several candidate theories for understanding convergence in GANs, naturally leading us to select Variational Inequalities, an intuitive generalization of the widely relied-upon theories from Convex Optimization. According to our analyses, none of the current GAN training algorithms is globally convergent in this setting. We propose a new technique, Crossing-the-Curl, for training GANs that converges with high probability in the N-dimensional (N-d) LQ-GAN setting.
This work makes the following contributions (proofs can be found in the supplementary material):
The first global convergence analysis of several GAN training methods for the N-d LQ-GAN,
Crossing-the-Curl, the first technique with stochastic convergence for the N-d LQ-GAN,
An empirical demonstration of Crossing-the-Curl in the multivariate LQ-GAN setting as well as some common neural network driven settings in Appendix A.16.
Generative Adversarial Networks
The Generative Adversarial Network (GAN) formulates learning a generative model of data as finding a Nash equilibrium of a minimax game. The generator ( player) aims to synthesize realistic data samples by transforming vectors drawn from a fixed source distribution, e.g., . The discriminator ( player) attempts to learn a scoring function that assigns low scores to synthetic data and high scores to samples drawn from the true dataset. The generator’s transformation function, , and discriminator’s scoring function, , are typically chosen to be neural networks parameterized by weights and respectively. The minimax objective of the original GAN is
where is the source distribution, is the true data distribution, and .
In practice, finding the solution to (1) consists of local updates, e.g., SGD, to and . This continues until 1) has stabilized, 2) the generated data is judged qualitatively accurate, or 3) training has de-stabilized and appears irrecoverable, at which point, training is restarted. The difficulty of training GANs has spurred research that includes reformulating the minimax objective , devising training heuristics , proving the existence of equilibria , and conducting local stability analyses .
We acknowledge here that our algorithm, Crossing-the-Curl, was independently proposed in as Symplectic Gradient Adjustment (SGA). In contrast to that work, this paper specifies a non-trivial application of this algorithm to LQ-GAN which obtains global convergence with high probability.
Recent work has studied a simplified setting, the Wasserstein LQ-GAN, where is a linear function, is a quadratic function, , and is Gaussian . Follow-up research has shown that, in this setting, the optimal generator distribution is a rank- Gaussian containing the top- principal components of the data . Furthermore, it is shown that if the dimensionality of matches that of , LQ-GAN is equivalent to maximum likelihood estimation of the generator’s resulting Gaussian distribution. To our knowledge, no GAN training algorithm with guaranteed convergence is currently known for this setting. We revisit the LQ-GAN in more detail in Section 4.
Convergence of Equilibrium Dynamics
As in convex optimization, a hierarchy of monotonicity exists. For all and , is
If, in Equation (2), “” is replaced by “”, then is strictly-monotone; if “” is replaced by “”, then is -strongly-monotone. If is a gradient, then replace monotone with convex.
Table 1 cites algorithms with convergence rates for several settings. Whereas gradient descent achieves optimal convergence rates for various convex optimization settings, extragradient achieves optimal rates for VIs. Results have been extended to the online learning setting as well .
2 The ODE Method & Hurwitz Jacobians
Recently, Nagarajan and Kolter performed a local stability analysis of the gradient dynamics of Equation (1), proving that the Jacobian of evaluated at is HurwitzOur definition of Hurwitz is equivalent to the more standard: is Hurwitz if . , i.e., the real parts of its eigenvalues are strictly positive. This means that if simultaneous gradient descent using a “square-summable, not summable” step sequence enters an -ball with a low enough step size, it will converge to the equilibrium. This applies only in the deterministic setting because stochastic gradients can cause the iterates to exit this ball and diverge. Note that while the real parts of eigenvalues reveal exponential growth or decay of trajectories, the imaginary parts reflect any rotation in the systemLinearized Dynamical System: ; Euler’s formula: ..
The Hurwitz and monotonicity properties are complementary (see A.8). To summarize, Hurwitz encompasses dynamics with exponentially stable trajectories and with arbitrary rotation, while monotonicity includes cycles (Jacobians with zero eigenvalues) and is similar to convex optimization.
Given the preceding discussion, we believe VIs and monotone operator theory will serve as a strong foundation for deriving fundamental convergence results for GANs; this theory is
Similar to convexity suggesting its adoption by the GAN community should be smooth,
Mature with natural mechanisms for handling constraints, subdifferentials, and online scenarios,
Rich with algorithms with finite sample convergence for a hierarchy of monotone operators.
Finally, we suggest for a lucid comparison of convex optimization, game theory, and VIs.
The Wasserstein Linear Quadratic GAN
In the Wasserstein Linear-Quadratic GAN, the generator and discriminator are restricted to be linear and quadratic respectively: and . Equation (1) becomes
The map associated with this zero-sum game is constructed by concatenating the gradients of the two players’ losses ():
Crossing-the-Curl
In this section, we will derive our proposed technique, Crossing-the-Curl, motivated by an examination of the ()-subsystem of LQ-GAN, i.e., fixed at for any . The results discussed here hold for the N-dimensional case as well. The map associated with this subsystem is plotted in Figure 2 and formally stated in Equation (6).
The Jacobian of is not Hurwitz, and simultaneous gradient descent, defined in Equation (7), will diverge for this problem (see A.5). However, is monotone and Lipschitz in the sense that . Table 1 offers an extragradient method (see Figure 2) with convergence rate, which is optimal for worst case monotone maps.
Nevertheless, an algorithm that travels perpendicularly to the vector field will proceed directly to the equilibrium. The intuition is to travel in the direction that is perpendicular to both and the axis of rotation. For a 2-d system, the axis of rotation can be obtained by taking the curl of the vector field. To derive a direction perpendicular to both and the axis of rotation, we can take their cross product:
where is Feynman notation for the gradient with respect to only and means evaluate the expression at . The factor ensures the algorithm moves toward regions of “tighter cycles” and simplifies notation. It may be sensible to perform some linear combination of simultaneous gradient descent and Crossing-the-Curl, so we will refer to as .
Note that the fixed point of remains the same as the original field . Furthermore, the reader may recognize as the gradient of the function , which is strongly convex, allowing an convergence rate in the deterministic setting. is derived from intuition in 2-d, however, we discuss reasons in the next subsection for why this approach generalizes to higher dimensions.
For the ()-subsystem, Crossing-the-Curl is equivalent to two other methods: the consensus algorithm and a Taylor series approximation to extragradient .
These equivalences occur because the Jacobian is skew-symmetric () for the ()-subsystem. In the more general case, where is not necessarily skew-symmetric, Crossing-the-Curl represents a combination of the two techniques. Extragradient (EG) is key to solving VIs and the consensus algorithm has delivered impressive results for GANs, so this is promising for . To our knowledge, is novel and has not appeared in the Variational Inequality literature.
Crossing-the-Curl stands out in many ways though. Observe that in higher dimensions, the subspace orthogonal to is dimensional, which means is no longer the unique direction orthogonal to . However, every matrix can be decomposed into a symmetric part with real eigenvalues, , and a skew-symmetric part with purely imaginary eigenvalues, . Notice that for an optimization problem, where is the Hessian.Assuming the objective function has continuous second partial derivatives—see Schwarz’s theorem. It is the imaginary eigenvalues, i.e., rotation, that set equilibrium problems apart from optimization and necessitate the development of new algorithms like extragradient. It is reassuring that this matrix appears explicitly in . In addition, reduces to gradient descent when applied to an optimization problem making the map agnostic to the type of problem at hand: optimization or equilibration.
Consider the perspective of as preconditioning by a skew-symmetric matrix. Preconditioning with a positive definite matrix dates back to Newton’s method and has reappeared in machine learning with natural gradient . Dafermos considered asymmetric positive definite preconditioning matrices for VIs. Thomas extended the analysis of natural gradient to PSD matrices. We are not aware of any work using skew-symmetric matrices for preconditioning. The scalar for any skew-symmetric matrix , so calling a PSD matrix is not adequately descriptive.
Note that Crossing-the-Curl does not always improve convergence; this technique can transform a strongly-monotone field into a saddle and an unstable fixed point (non-monotone) into a strongly-monotone field (see A.9 for examples), so this technique should generally be used with caution.
Lastly, Crossing-the-Curl is inexpensive to compute. The Jacobian-vector product, , can be approximated accurately and efficiently with finite differences. Likewise, can be computed efficiently with double backprop by taking the gradient of . In total, three backprops are required, one for , one for , and one for .
In our analysis, we also consider the gradient regularization proposed in , , the Unrolled GAN proposed in , , alternating gradient descent, , as well as any linear combination of , , and , deemed , which forms a family of maps that includes , , and :
Keep in mind that we are proposing as a generalization of Crossing-the-Curl. We state our main results here for the -subsystem.
For any , with at least one of and positive and both non-negative is strongly monotone. Also, its Jacobian is Hurwitz. See Proposition 13.
, , , and with are strongly-monotone with Hurwitz Jacobians. See Proposition 1.
, , , and with any are monotone, but not strictly monotone. Of these maps, only ’s Jacobian is Hurwitz. See Propositions 12 and 13.
Analysis of the Full System
Here, we analyze the maps for each of the algorithms discussed above, testing for quasimonotonicity (the weakest monotone property) and whether the Jacobian is Hurwitz for the full LQ-GAN system.
Proving quasiconvexity of 4th degree polynomials has been proven strongly NP-Hard . This implies that proving monotonicity of 3rd degree maps is strongly NP-Hard. The original contains quadratic terms suggesting it may welcome a quasimonotone analysis, however, the remaining maps all contain 3rd degree terms. Unsurprisingly, analyzing quasimonotonicity for represents the most involved of our proofs given in Appendix A.11.
The definition stated in (3) suggests checking the truth of an expression depending on four separate variables: , , , . While we used this definition for certain cases, the following alternate requirements proposed in made the complete analysis of the system tractable. We restate simplified versions of the requirements we leveraged for convenience.
For all and such that , we have that .
is quasimonotone on only if (A) holds, i.e. (A) is necessary but not sufficient.
is pseudomonotone on if (A) and (B) hold, i.e. (A) and (B) are sufficient but not necessary.
Condition (A) says that for a map to be quasimonotone, the map must be monotone along directions orthogonal to the vector field. In addition to this, condition (B) says that for a map to be pseudomonotone, the dynamics, , must not be leading away from the equilibrium anywhere.
Equipped with these definitions, we can conclude the following:
None of the maps, including with any setting of coefficients, is quasimonotone for the full LQ-GAN. See Corollary 5 and Propositions 15 through 17.
None of the maps, including with any setting of coefficients, has a Hurwitz Jacobian for the full LQ-GAN. See Propositions 27 and 15 through 17.
Results from the previous section suggest that we cannot solve the full LQ-GAN, but given that we can solve the ()-subsystem, we shift focus to the ()-subsystem assuming the mean has already been learned exactly, i.e., . We will revisit this assumption later.
We can conclude the following for the ()-subsystem:
, , , , and are not quasimonotone. Also, their Jacobians are not Hurwitz. See Propositions 14 through 19.
and are pseudomonotone which implies an stochastic convergence rate. See Propositions 21 and 24. Their Jacobians are not Hurwitz. See Proposition 27.
No monotone exists. See Proposition 26.
These results are not purely theoretical. Figure 4 displays trajectories resulting from each of the maps.
We can further improve upon and by rescaling with : (12)(13) and (14)(15) respectively. This results in strongly-monotone and strongly-convex systems respectively, improving the stochastic convergence rate to . In deriving these results, we assumed the mean was given. We can relax this assumption and analyze the ()-subsystem under the assumption that the mean is “close enough”. Using a Hoeffding bound, we find that k>\big{(}\frac{y_{hi}-y_{low}}{-|\mu|+\sqrt{\mu^{2}+d\sigma^{2}}}\big{)}^{2}\log[\frac{\sqrt{2}}{\delta^{1/2}}] iterations of are required to achieve a probability of the mean being accurate enough to ensure the ()-subsystem is strongly-monotone. Note that this approach of first learning the mean, then the variance retains the overall stochastic rate. We summarize the main points here.
A nonlinear scaling of and results in strictly monotone and -strongly monotone subsystems respectively. See Proposition 29.
If the mean is first well approximated, i.e., , then remains 1) -strongly-monotone if the ()-subsystem is “shut off” or 2) strictly-monotone if the -subsystem is re-weighted with a high coefficient. See Propositions 30 and 31.
and are not quasimonotone for the 2-d LQ-GAN system (with and without scaling). See Proposition 32.
Several takeaways emerge. One is that the stability of the system is highly dependent on the mean first being learned. In other words, batch norm is required for the monotonicity of LQ-GAN, so it is not surprising that GANs typically fail without these specialized layers.
Second is that stability is achieved by first learning a simple subsystem, (), then learning the more complex, ()-subsystem. This theoretically confirms the intuition behind progressive training of GANs , which have generated the highest quality images to date.
Thirdly, because is symmetric (and ), we can integrate to discover the convex function it is implicitly descending via gradient descent: . Compare this to KL-divergence: . In contrast to , is convex in and may be a desirable alternative due to less extreme gradients near .
After learning both the mean and variance of each dimension, the covariance of separate dimensions can be learned. Proposition 35 in the Appendix states that the subsystem relevant to learning each row of is strictly monotone when all other rows are held fixed. In fact, the maps for these subsystems are affine and skew-symmetric just like the ()-subsystem. This implies that Crossing-the-Curl applied successively to each row of can solve for ; pseudocode is presented in Algorithm 1 in Appendix A.15. Note that this procedure is reminiscent of the Cholesky–Banachiewicz algorithm which computes row by row, beginning with the first row. The resulting algorithm is .
Experiments
The second and third rows of the table reveal that convergence slows considerably for higher dimensions. However, the stagewise procedure discussed in Subsection 6.2 is guaranteed to converge given the mean has been learned to a given accuracy. This procedure solves the 4-d deterministic LQ-GAN in iterations with a success rate. For the 4-d stochastic LQ-GAN using two-sample minibatch estimates, this procedure achieves in 100,000 iterations with a 0.75 success rate.
Conclusion
In this work, we performed the first global convergence analysis for a variety of GAN training algorithms. According to Variational Inequality theory, none of the current GAN training algorithms is globally convergent for the LQ-GAN. We proposed an intuitive technique, Crossing-the-Curl, with the first global convergence guarantees for any generative adversarial network. As a by-product of our analysis, we extract high-level explanations for why the use of batch norm and progressive training schedules for GANs are critical to training. In experiments with the multivariate LQ-GAN, Crossing-the-Curl achieves performance superior to any existing GAN training algorithm.
For future work, we will investigate alternate parameterizations of the discriminator such as . We will also work on devising heuristics for setting the coefficients of .
Acknowledgments
Crossing-the-Curl was independently proposed in called Symplectic Gradient Adjustment (SGA). Like Crossing-the-Curl, this algorithm is motivated by attacking the challenges of rotation in differentiable games, however, it is derived by performing gradient descent on the Hamiltonian as opposed to generalizing a particular perpendicular direction selected from intuition in 2-d. Given the equivalence between SGA and Crossing-the-Curl, our work can also be viewed as proving that a non-trivial application of this algorithm can be used to solve the LQ-GAN. On the other hand, we have also proven in Proposition 7 that a naive application of this algorithm is insufficient for solving LQ-GAN suggesting more research is required to understand and more efficiently solve this complex problem.
References
Appendix A Appendix
Algorithmic Game Theory (AGT) offers results on convergence to equilibria when a game, possibly online, is convex , socially-convex , or smooth . A convex game is one in which all player losses are convex in their respective variables, i.e. is convex in . A socially-convex game adds the additional requirements that 1) there exists a strict convex combination of the player losses that is convex and 2) each player’s loss is concave in the variables of each of the other players. In other words, the players as a whole are cooperative, yet individually competitive. Lastly, smoothness ensures that “the externality imposed on any one player by the actions of the others is bounded” . In a zero-sum game such as (1), one player’s gain is exactly the other player’s loss making smoothness an unlikely fit for studying GANs. See for examples where the three properties above overlap with monotonicity in VIs.
A.1.2 Differential Games
Differential games consider more general dynamics such as , not just first order ODEs, however, the focus is on systems that separate control, , and state , i.e. . More specific to our interests, Differential Nash Games can be expressed as Differential VIs, a specific class of infinite dimensional VIs with explicit state dynamics and explicit controls; these, in turn, can be framed as infinite dimensional VIs without an explicit state.
A.2 Nash Equilibrium vs VI Solution
Repeated from . Let be a cost minimization game with player cost functions and feasible set . Let be a Nash equilibrium. Let . Then
where is the internal cone at . When is pseudoconvex in for all , this condition is also sufficient. Note that this is implied if is pseudomonotone, i.e. pseudomonotonicity of is a stronger condition.
A.3 Table of Maps Considered in Analysis
All maps corresponding to the ()-subsystem in Table 4 maintain the desired unique fixed point, , where .
For the ()-subsystem, all maps except with certain settings of () and maintain the desired unique fixed point, . introduces an additional spurious fixed point at
is a special case of where , , and .
A.4 Minimax Solution to Constrained Multivariate LQ-GAN is Unique
Taking derivatives and setting equal to zero, we find that the fixed point at the interior is unique.
The last implication in Equation (30) follows because is constrained to be of Cholesky form, i.e., lower triangular with positive diagonal, and every symmetric positive definite matrix has a unique Cholesky decomposition.
The second to last implication of Equation (32) follows because is necessarily full rank. Note this implies is also full rank. The null space of a full rank matrix is the zeros vector, which implies . is symmetric, so this implies . ∎
Consider the case where the mean of is zero:
We will show that simultaneous gradient descent always produces an iterate that is farther away from the equilibrium than the previous iterate, i.e. .
Therefore, simultaneous gradient descent diverges from the equilibrium of the ()-subsystem for any step size scheme, .
A.6 Derivation of Crossing-the-Curl
Here, we derive our proposed technique in 3-d, however, the result of the derivation can be computed in arbitrary dimensions:
A.7 Monotonicity: Definitions and Requirements
For all and ,
While we used these definitions in our analysis for certain cases, the following alternate requirements proposed in made the complete analysis of the system tractable. We restate them here for convenience. Note that we what we refer to as condition (B) in the main body of the paper is actually a stronger version of condition (C) below with .
is quasimonotone on if and only if (A) and (B’) hold.
is pseudomonotone on if and only if (A) and (C’) hold.
A.8 A Comparison of Monotonicity and Hurwitz
The monotonicity and Hurwitz properties are complementary.
Let , , , and . Then so is Hurwitz, and
which, by condition (A), implies is not quasimonotone.
A.8.2 Monotonicity Does Not Imply Hurwitz
Let and . Then so is not Hurwitz, but
A.8.3 Monotonicity and Hurwitz Can Overlap
Let and . Then so is Hurwitz and
If is (strictly,strongly)-monotone, then the Jacobian of is a real, square, (positive definite,strongly-positive definite) matrix, therefore, it matches the above assumptions. Hence, the conclusion follows. ∎
A.9 Crossing-the-Curl Can Make Monotone Fields, Non-Monotone
Here, we provide examples of negative results for Crossing-the-Curl. This is to emphasize that our proposed technique can cause problems if not used with caution. The headings below describe the before and afters when applying our proposed technique to the map .
Increase in condition number: .
becomes non-monotone.
Crossing-the-Curl forces monotonicity for normal, affine fields.
Let and assume is normal, i.e., . Then
Unrolled GANs and Alternating Updates are Monotone for the ()-subsystem.
In Unrolled GANs, the generator computes the gradient of assuming the discriminator has already made several updates. Define the discriminator’s update as
and denote the composition of , -times as
where is some positive integer. Then the update for Unrolled GANs is
In the case of the ()-subsystem, we can write these unrolled updates out explicitly. Remember , so
where the corresponding map is . Taking a look at the Jacobian, we find
Here, we considered updating first, but the ()-subsystem is perfectly symmetric, so the analysis holds either way. If is updated first, this is equivalent to Unrolled GAN with (see Equation 104). The Jacobian is
The Jacobian’s for Unrolled GAN and alternating descent are both positive semidefinite, therefore, their maps are monotone (but not strictly-monotone). Note that these results imply neither is Hurwitz either because both Jacobians exhibit a zero eigenvalue. ∎
, , , and are strongly-monotone for the ()-subsystem (includes multivariate case). and are monotone, but not strictly monotone. Moreover, , , , , and are Hurwitz for the ()-subsystem (includes multivariate case). is not Hurwitz.
We start with the original map, , and its Jacobian.
The symmetrized Jacobian is positive semidefinite, therefore this system is monotone. Also, the real parts of the eigenvalues of its Jacobian are zero, therefore, is not Hurwitz.
Now we analyze , , and , which as discussed in the main body, are equivalent.
The symmetrized Jacobian is positive definite with a minimum eigenvalue of , therefore this system is -strongly-monotone. By Proposition 10, the Jacobians of these maps are Hurwitz for the -subsystem.
Now we analyze the generalization .
The symmetrized Jacobian is positive definite with a minimum eigenvalue of , therefore this system is -strongly-monotone. By Proposition 10, is Hurwitz for the -subsystem.
Now we analyze the regularized-gradient algorithm, .
Therefore, this map is monotone (but not strictly or strongly-monotone). Also, the real parts of the eigenvalues of its Jacobian are strictly positive, therefore, is Hurwitz.
Note that for , , , and , is symmetric, therefore, is the gradient of some function, . Also, note that the standard algorithm with step size is equivalent to the standard running estimate of the mean: where is the -th sample.
Specifically, we first consider the sign of . Lemma 1 rules out negative values. Lemma 2 rules out positive values when , and Lemma 3 rules out positive values when . Corollary 2 concludes that .
Next, given , we consider the sign of . Lemmas 4 and 5 rule out positive values of when is greater than or less than or equal to zero respectively, i.e., cannot be positive. Similarly, Lemmas 6 and 7 rule out negative values of when is less than or greater than or equal to zero respectively, i.e., cannot be negative. Corollary 3 concludes that .
Lastly, given that , Lemmas 8 and 9 prove that cannot be greater than or less than or equal to zero respectively. Corollary 4 concludes that . Therefore, the only quasimonotone linear combination is the trivial one resulting in , which completes the proof.
For to be quasimonotone, must not be strictly less than zero, i.e. .
If , then this system is not quasimonotone. Therefore, assume from now on. ∎
If , for to be quasimonotone, must not be strictly greater than zero, i.e. .
We will use a different parameterization of for this part of the proof.
In order for a system to be quasimonotone, we require condition (A) (among other properties). We will now show that this property is not satisfied for with by considering two different cases.
Case 1: Consider the -subsystem. Let
Above, we premultiply by a skew symmetric matrix, which ensures .
The relevant portion of the Jacobian of is
Consider and both and fixed.
This implies either or for the system to be quasimonotone.
Case 2: Consider the -subsystem. Let
The relevant portion of the Jacobian of is
Consider and . Then
Then . In either case, must be nonpositive. Therefore, . ∎
Part of the proof in Lemma 2 looks at the limit in which approaches 0. One might presume a simple fix is to constrain to be larger than some small value, e.g., 1e-10, and use a large value. Here, we show that even using breaks quasimonotonicity. The variance of the data distribution is assumed to be unknown, which would make it very difficult to select a proper lower bound for that maintains quasimonotonicity within the feasible region.
Consider and the -subsystem as in 2. Then
If , then this is a concave quadratic form in . To find where this function is positive, we need to find its roots.
The root with smaller magnitude provides an upper bound for .
Now consider again and equation 141 with .
This provides a lower bound for .
The upper bound we require for is greater than the lower bound, therefore, no will satisfy quasimonotonicity. ∎
If , for to be quasimonotone, must not be strictly greater than zero, i.e. .
For this proof, we make use of the traditional definition of quasimonotonicity. Consider
If , then this system is not quasimonotone. In either case, . ∎
𝛽𝛾0\beta+\gamma=0 for quasimonotonicity.). Together, Lemmas 1, 2 and 3 imply that must be to satisfy quasimonotonicity.
If and , for to be quasimonotone, must not be strictly greater than zero, i.e. .
For this proof, we make use of the traditional definition of quasimonotonicity. Consider
If and , then for the system to be quasimonotone. ∎
If , for to be quasimonotone, must not be strictly greater than zero, i.e. .
We will assume , which by Lemma 4 implies . This will lead to a contradiction. Consider
If and (implies ), then and , which breaks quasimonotonicity. Therefore, . ∎
If and , for to be quasimonotone, must not be strictly less than zero, i.e. .
If , then to maintain quasimonotonicity. ∎
If , for to be quasimonotone, must not be strictly less than zero, i.e. .
We will assume , which by 6 implies . This will lead to a contradiction.
If and (implies ), then and , which breaks quasimonotonicity. Therefore, . ∎
Together, Corollary 2 and Lemmas 4-7 imply that must equal zero for to be quasimonotone.
If and , for to be quasimonotone, must not be strictly greater than zero, i.e. .
If , then this system is not quasimonotone. Therefore, . ∎
If and , for to be quasimonotone, must not be strictly less than zero, i.e. .
If , then this system is not quasimonotone. Therefore, . ∎
𝛽𝛾0𝛼0⇒𝛽𝛾0(\beta+\gamma)=0,\alpha=0\Rightarrow\beta=\gamma=0). Together, Lemmas 8 and 9 imply that , which, along with Corollary 2, imply that as well.
[] Together, Corollaries 2 and 3, and 4 imply that there is no non-trivial linear combination that induces a quasimonotone LQ-GAN system.
, , , and are not quasimonotone for the LQ-GAN system.
These maps are all linear combinations of , and , therefore, by Corollary 5, they are not quasimonotone for the LQ-GAN system. ∎
Note that if a map is not quasimonotone for the -subsystem, then it is not quasimonotone for the full system. This is because an analysis of the -subsystem is equivalent to an analysis of a subspace of the full system with .
is not quasimontone for the -subsystem. Also, its Jacobian is not Hurwitz.
The Jacobian of for the ()-subsystem is
The trace of is strictly negative for , which implies has an eigenvalue with strictly negative real part. Therefore, is not Hurwitz. ∎
is not quasimonotone for the -subsystem. Also, its Jacobian is not Hurwitz.
In order for a system to be quasimonotone, we require condition (A) (among other properties). We will now show that this property is not satisfied for the gradient-regularized system.
Consider the point and let be defined as follows:
where is actually derived by considering the field formed by crossing the curl for the 2-d subspace with and only.
It suffices to consider the submatrix of the Jacobian corresponding to and only when computing :
If and , then there isn’t an that will make this system quasimonotone.
The Jacobian of for the ()-subsystem is
The trace of is strictly negative for and , which implies has an eigenvalue with strictly negative real part. Therefore, is not Hurwitz. ∎
is not quasimonotone or Hurwitz for the -subsystem. Also, its Jacobian is not Hurwitz.
We consider Unrolled GAN as described in . Some of the necessary arithmetic can be found in the supplementary Mathematica notebook. Define the discriminator’s update as
where is a step size, and denote the composition of , -times as
where is some positive integer. Then the update for Unrolled GANs is
In the case of the ()-subsystem, we can write these unrolled updates out explicitly. Remember , so
We will use the following vector to test condition (A) for quasimonotonicity of :
Computing and evaluating at gives
therefore, is not quasimonotone.
If we examine the determinant of and evaluate it at , we get
which is less than zero for positive . Therefore, the Jacobian exhibits negative eigenvalues which means the system is not Hurwitz. ∎
is not quasimonotone or Hurwitz for the -subsystem. Also, its Jacobian is not Hurwitz.
We consider an alternating gradient descent scheme. Some of the necessary arithmetic can be found in the supplementary Mathematica notebook. First, we begin with the case where the discriminator updates first. The updates are
where is a step size. The corresponding map is
Note the similarity to the Unrolled GAN map Equation (242). The maps are equivalent if . Unrolled GANs was shown to be not quasimonotone for any , therefore, is not quasimonotone as well.
If we examine the trace of and evaluate it at (), we get
which is strictly negative. Therefore, the Jacobian exhibits negative eigenvalues which means the system is not Hurwitz.
Now, consider the generator updating first. The updates are
Testing for condition (A) as before (see Equations (242)- (244)), we find that
Using Descartes’ Rule of Signs , we can determine that this expression has exactly one positive root for . This implies that changes sign locally around this root when varying , which means for some positive . Therefore is not quasimonotone.
If we examine the determinant of and evaluate it at (), we get
which is less than zero for positive . Therefore, the Jacobian exhibits negative eigenvalues which means the system is not Hurwitz. ∎
The following propositions concern the monotonicity of , , and for the ()-subsystem. The field and Jacobian for will be helpful for proofs of their properties.
is not quasimontone for the ()-subsystem. Also, its Jacobian is not Hurwitz.
This corresponds to with . We consider three cases. Let
which implies for the system to be quasimonotone.
which, combined with above, implies for the system to be quasimonotone.
Case 3: Consider . Then
The quantity in parentheses must be positive for this system to be quasimonotone. This quantity is a concave quadratic form with an upper root of . This implies for the system to be quasimonotone.
The last two results cannot be satisfied by a single , therefore, this system is not quasimonotone.
For completeness, we analyze the limit where the term is ignored. Consider .
This is negative for , therefore, this system is not quasimonotone.
The trace of is strictly negative for and , which implies has an eigenvalue with strictly negative real part. Therefore, is not Hurwitz. ∎
is not quasimontone for the ()-subsystem. Also, its Jacobian is not Hurwitz.
This corresponds to with . We consider two cases.
which, for , implies for the system to be quasimonotone.
Case 2: Consider . Then
which, for , implies for the system to be quasimonotone. Combined with above, this implies for the system to be quasimonotone. In conclusion, is not quasimonotone.
The trace of is strictly negative for and , which implies has an eigenvalue with strictly negative real part. Therefore, is not Hurwitz. ∎
requires to be pseudomonotone for ()-subsystem
This corresponds to with . We consider two cases.
Case 1: Consider and . Then
Then or for the system to be quasimonotone.
Case 2: Consider and . Then
Then or for the system to be quasimonotone. The latter condition is more lenient, so the former is unnecessary.
For the system to be quasimonotone in both scenarios, we require that . This implies must be arbitrarily large for small . In the limit, the effect of on the system is negligible. We consider this limit next. ∎
is pseudomonotone for ()-subsystem.
Note this system is 2-d, therefore, there is only 1 vector (aside from scaling) that is perpendicular to .
This satisfies conditions (A) and (C), therefore, this system is pseudomonotone. ∎
is pseudomonotone for the constrained -subsystem.
We consider in this case and let the user define a feasible region for which they are confident the equilibrium exists: and —the most important bounds being those on . We will attempt to find a value for that ensures the system is pseudomonotone within this region.
A partially sufficient (and necessary) condition for pseudomonotonicity is the following (see condition (C)).
We can find the that maximizes this equation for a given by setting the derivative equal to zero and taking the positive root of the resulting quadratic. The denominator of the derivative is non-negative and only zero at equilibrium—this is not a concern because at equilibrium. Continuing and looking at the numerator of the derivative, we find
If we plug that back into the lower bound for , we get
The condition above along with the following (see condition (A)) are sufficient to ensure pseudomonotonicity.
If , then this quantity is greater than or equal to zero due to the result in equation (283), which we have already shown to be greater than zero. Therefore, we focus on . We can divide the analysis into two cases.
Consider . In this case, all coefficients of terms except a term and the last term (the constant) are positive. For simplicity, we can find the value for such that the first part of the coefficient is greater than the two negative terms.
Now consider . One of the terms in the coefficient is now negative. We will find a value for such that the term can drown out that negative term.
Note this bound is not tight; it is just meant to provide a satisfactory estimate. ∎
requires to be pseudomonotone for the ()-subsystem.
This corresponds to with .
this, combined with above, implies that .
This implies must be arbitrarily large for small . In the limit, the effect of on the system is negligible. We consider this limit in Subsubsection 24. ∎
is pseudomonotone for the -subsystem.
Note that the skew part of the Jacobian of is full rank except at the boundary (), so maintains the same fixed points. This can be seen by looking at above. We will simply need to constrain to be greater than 0.
In order for a system to be quasimonotone, we require condition (A) (among other properties). We will now show that this property is satisfied for the ()-subsystem.
Case 1: Consider the point and let be defined as follows:
is 0 as expected.
Now, we will compute to see if it is greater than zero.
In addition to this, proving that is sufficient for proving condition (C).
The last two terms of the sum are always the same sign due to the square function being “monotone” and the fact that is constrained to be non-negative. Therefore, is pseudomonotone. ∎
is pseudomonotone for the constrained -subsystem.
We consider in this case and let the user define a feasible region for which they are confident the equilibrium exists: and —the most important bounds being those on . We will attempt to find a value for that ensures the system is pseudomonotone within this region.
A partially sufficient (and necessary) condition for pseudomonotonicity is the following (see condition (C)).
We can find the that maximizes this equation for a given by setting the derivative equal to zero and taking the positive root of the resulting quadratic. The denominator of the derivative is non-negative and only zero at equilibrium—this is not a concern because at equilibrium. Continuing and looking at the numerator of the derivative, we find
If we plug that back into the lower bound for , we get
The condition above along with the following (see condition (A)) are sufficient to ensure pseudomonotonicity.
If , then this quantity is greater than or equal to zero due to the result in equation (322), which we have already shown to be greater than zero. Therefore, we focus on . We can divide the analysis into two cases.
Consider . In this case, all coefficients of terms except the last term (the constant) are positive. For simplicity, we can find the value for such that the first part of the coefficient is greater than the last term (the constant).
Now consider . One of the terms in the coefficient is now negative. We will find a value for such that the term can drown out the two negative terms.
This last lower bound is the greatest of the three, so it suffices to set greater than this value to ensure the system is pseudomonotone within the given feasible region. ∎
is not monotone for the (-subsystem (before scaling).
Let be defined as follows:
The trace of the symmetrized Jacobian must be non-negative to ensure monotonicity because a negative trace implies the existence of a negative eigenvalue:
Assume . If and , then the trace is less than zero.
Assume . If and , then the trace is less than zero.
If , then . If , then . Therefore, , however, and are constants while is a variable. Therefore, and must equal zero to satisfy this for all proving that no monotone linear combination exists. ∎
is not Hurwitz for the (-subsystem.
Consider at .
If , then implies the existence of an eigenvalue with negative real part. If , then implies the existence of an eigenvalue with negative real part. If , then the real part is zero. ∎
There exists an family after scaling by that exhibits strict-monotonicity.
If we consider the same linear combinations above, but divide by , we can obtain a family of monotone fields (see Mathematica notebook).
The trace of the corresponding symmetrized Jacobian is
For constant and and nonzero , there exists a value for that will force the trace to be negative, therefore must be zero. Note that must be greater than or equal to to ensure that the trace cannot be made negative in the limit as grows to infinity.
Case 1: Consider the case where . Then for any fixed , , and nonzero ,
Case 2: Otherwise, consider solving the quadratic form for when :
For the trace to be non-negative, we need the leading coefficient of the quadratic to be positive, i.e., . We also need there to be at most 1 real root, meaning the square root must be non-positive. If , then setting and using the following formula will force the root to be positive:
For example, set , and then set and as follows to force the trace to be negative:
Case 3: If , then the root is necessarily positive. Therefore, must be set to zero.
The field and Jacobian are now wieldy enough to state:
and is non-negative as long as both and .
which is non-negative as long as, in addition to the previous conditions, we have . The trace and determinant are both strictly positive if .
In summary, is strictly-monotone, i.e., , if and . ∎
The family includes and . By Proposition 28, and are at least strictly-monotone.
is -strongly monotone and is only strictly-monotone.
Case : The eigenvalues of are and \lambda_{2}=\frac{1}{2}\Big{(}1+\frac{\sigma^{2}}{a^{2}}\Big{)}. Therefore, and is -strongly monotone.
Case : The eigenvalues of a matrix can be written in terms of the trace and determinant as
Therefore, if the term can be made arbitrarily small, then one of the eigenvalues can made arbitrarily close to zero. On the other hand, if this quantity has a finite lower bound, then the eigenvalues are lower bounded as a constant multiple of the trace.
The trace and determinant of are
This term can be made arbitrarily small as goes to infinity. To be more rigorous, let so that and . Then
An application of L’Hopital’s rule shows that
The minimum eigenvalue only approaches zero in the limit, so is strictly-monotone. ∎
is the gradient of the following convex function: f^{w_{2},a}_{cc^{\prime}}=w_{2}^{2}+1/2\Big{(}(a^{2}-\sigma^{2})-\sigma^{2}\log(\frac{a^{2}}{\sigma^{2}}\Big{)}.
The Jacobian of is symmetric and PSD, therefore it is the Hessian of some convex function. We can integrate to arrive at a convex function (with arbitrary constant). Integrating results in the following:
Note that must be convex along the subspace with as well, which implies that
is convex as well. This function is of individual interest because it may serve as a preferred alternative to KL-divergence. ∎
A.13 Progressive Learning of LQ-GAN
Here, we consider the stochastic setting where the GAN is trained using samples from and . There are two ways to learn both the mean and variance of a distribution using . One is to first learn the mean to a high degree of accuracy, then stop learning the mean and start learning the variance. The other is to keep learning the mean with an appropriate weighting of the two systems to maintain stability. We discuss the former option first.
Assume all lie in . After k>\big{(}\frac{y_{hi}-y_{low}}{-|\mu|+\sqrt{\mu^{2}+d\sigma^{2}}}\big{)}^{2}\log[\frac{\sqrt{2}}{\delta^{1/2}}] iterations, with probability, , the -subsystem can be “shut-off” and the ()-subsystem safely “turned-on” resulting in a -strongly-monotone .
We begin by observing the symmetrized Jacobian of :
where . In order for to be strongly monotone, we require . In other words, the square of the generator’s estimate of the mean, , learned from training the ()-subsystem needs to be less than or equal to .
Assume and introduce a scalar: . Remember, we require . And we know which implies
This expression has two roots for , one positive and one negative. can only be upper bounded by a positive number, so we select the positive root.
Plugging back into equation (377) for , we find that
Rearranging (376) and plugging in , we can derive the number of iterations required:
If we assume and use a Chernoff bound, we find
The number of samples needed to maintain stability of the system grows as the true mean deviates from zero. This is not an artifact of the concentration inequalities (it occurs with both), but of the parameterization of the LQ-GAN—the samples are not mean centered before being passed to the quadratic discriminator, i.e., rather than . This may explain why batch norm is so helpful (almost required) in stabilizing training.
Assume all lie in . After k>\big{(}\frac{y_{hi}-y_{low}}{-|\mu|+\sqrt{\mu^{2}+d\sigma^{2}}}\big{)}^{2}\log[\frac{\sqrt{2}}{\delta^{1/2}}] iterations, with probability, , the ()-subsystem can be up-weighted and the ()-subsystem “turned-on”, resulting in a strictly-monotone LQ-GAN.
As before, assume we are running on the ()-subsystem and on the ()-subsystem. Also, multiply by , i.e., increase the learning rate by or divide the learning rate of by . The full symmetrized Jacobian of this system is:
The upper left block of this matrix is positive definite. In order to show the whole matrix is positive definite, it suffices to prove the lower right block is positive definite. The trace and determinant of that block are
where as before. We need for (for ) and for . As before, Hoeffding’s inequality says iterations are required for an accurate estimate of the mean (see Equation (383)). And as before, we find that . We will focus on the determinant condition here. Let
More simply, let . Then set . This ensures the trace and determinant are both strictly positive which implies that the resulting system is at least strictly monotone.
We can show that this system is not strongly-monotone by upper bounding the minimum eigenvalue. To ease the analysis, let and note that (see Equation (389)), i.e., is finite. This allows us to upper bound the determinant, in turn, upper bounding the minimum eigenvalue. The determinant simplifies to
The minimum eigenvalue is upper bounded as follows:
As the system continues learning a more accurate mean (iterations, , is increasing), is effectively decreasing towards zero. In the limit .
Given, , we can set . Also, note that if the distribution is known to support balls at the ends of the specified interval, , with some nonzero probabilities, and , then we can lower bound the variance as well. Specifically, let P_{low}=\frac{\epsilon}{2}\big{(}p(y_{low})+p(y_{low}+\epsilon)\big{)} and P_{hi}=\frac{\epsilon}{2}\big{(}p(y_{hi})+p(y_{hi}-\epsilon)\big{)}. Then
Let be a lower triangular matrix with positive diagonal— represents the generator’s guess at the square root of .
The 2-d LQ-GAN is not quasimonotone for or with or without scaling.
We will show that this system fails condition (A). Please refer to the Mathematica notebook for our derivations of these results.
Define the following skew symmetric matrix.
Let and . Similarly, with scaling, let and . Let
This implies that neither system is quasimonotone (with, , or without, , scaling). ∎
The 2-d LQ-GAN with and already learned, i.e., and , is not quasimonotone for or .
We will show that this system fails condition (A). Please refer to the Mathematica notebook for our derivations of these results.
Define the following skew symmetric matrix.
Let and . Let
This implies that neither system is quasimonotone. ∎
The 3-d LQ-GAN with the diagonal of already learned, i.e., , is not quasimonotone for or with or without scaling.
We will show that this system fails condition (A). Please refer to the Mathematica notebook for our derivations of these results.
Define the following skew symmetric matrix.
Let and . Let
This implies that neither system is quasimonotone. ∎
The N-d LQ-GAN with all but a single row of fixed is strictly-monotone for , , and .
First, note that the Cholesky decomposition of , denoted by , obeys the follow equation:
where . is symmetric, so can be recovered as . This allows us to remove 1 degree of freedom from the system by defining the diagonal term in a single row of in terms of the other entries in the row:
where as before must be greater than zero. We assume that has already been learned by Crossing-the-Curl as described in the main body. The condition can be ensured by constraining with —this can be achieved with a simple ball projection.
We will begin by writing down the map for the entire system and then simplifying using the constraints and assumptions discussed above:
We are only interested in learning the th row of . Take for example. Notice that the rd row of , only contains the following terms: . The rest are set to zero as mentioned earlier. The reason for this will become apparent soon. We fix all other entries to zero to highlight the relevant subsystem below:
Notice that the map is zero only if Equation (421) is satisfied for and for all . Therefore, setting all other entries of as prescribed simplified the system, while maintaining the correct fixed point.
In order to determine the monotonicity of this system, we need to compute the Jacobian of :
which is skew-symmetric and constant with respect to the variables being learned: and . Therefore, is PSD, which implies is monotone. The fact that is constant along with Proposition 11 imply that are also monotone:
Note that the component of corresponding to the dynamics of , is independent of . This means the dynamics are now decoupled from and can be run separately. By inspecting the symmetrized Jacobian of we can show that it is a block matrix composed of positive definite matrices:
is positive definite because is constrained to be of Cholesky form. Moreover, the eigenvalues of are the same as , therefore both blocks are positive definite. This implies the entire matrix is positive definite which means are strictly monotone. Note that we do not require for strict monotonicity. In practice, the system will actually be both strongly-monotone and smooth. This is because is constrained with a projection onto a ball and the diagonal of is restricted to be larger than . These two conditions guarantee a nonzero, finite minimum and maximum value for the eigenvalues of —the minimum corresponds to strong-monotonicity and the maximum corresponds to smoothness. ∎
Unlike the ()-subsystem where monotonicity depends on the accuracy of the learned mean, this system is monotone as long as is PSD which is guaranteed from the form we have prescribed to . This result suggests learning the rows of in succession, and each subsystem is guaranteed to be strictly monotone. Note that the variance, i.e., diagonal of , will be slightly off the true value if the mean, , is not first learned perfectly. The learned will then be slightly off the true and errors will compound, but still not affect monotonicity. The subsystems corresponding to each row of can be revisited to learn the entries of more accurately. Permuting the dimensions of such that the dimensions corresponding to highest variance are learned first may ensure subsystems with maximal strong-monotonicity. We leave a detailed examination to future research.
A.15 An 𝒪(N/k)𝒪𝑁𝑘\mathcal{O}(N/k) Algorithm for LQ-GAN
Here we present pseudocode for solving the stochastic LQ-GAN. The maps corresponding to learning the mean and variance by Crossing-the-Curl are both strongly convex and can therefore be solved with a simple projected gradient method. We argued in the previous subsection that the map associated with learning the covariance terms is strongly-monotone and smooth, not only strictly monotone. In practice, we found that a projected Extragradient algorithm gave better results. The full procedure is outlined in Algorithm 1. Replace sample estimates with the true and for the deterministic LQ-GAN.
As mentioned above, the maps for learning the mean and variance are both strongly convex which implies a stochastic convergence rate for each, the sum of which is still .
In practice, the maps for learning each row of are strongly-monotone and smooth (see last paragraph of proof of Proposition 35) which implies a stochastic convergence rate for each as well. Because this technique consists of steps for learning the full -d LQ-GAN, it requires iterations which, in total, implies a stochastic convergence rate.
Hidden within this analysis is the fact that each iteration of learning the mean and variance is in terms of time-complexity and each iteration for learning each row of is , therefore this entire procedure is in terms of FLOPS. This is expected as the complexity of a Cholesky decomposition to compute is also . Note that unlike the complexity of computing each iteration which can be mitigated with parallel computation, the sequential nature of the stagewise procedure cannot be amortized which is why we report a convergence rate and not .
Another subtle point is that the LQ-GAN is locally monotone about the equilibrium. Recall from Theorem D.1 on p.26 in that the Jacobian at the equilibrium is of the following form (remember our definition for the Jacobian is the negative of theirs):
where is positive definite. The symmetrized Jacobian is then
This implies is monotone where . Therefore, we can use stagewise procedure in Algorithm 1 to converge to a local neighborhood about the equilibrium, constrain the system to this neighborhood with a projection (which will guarantee smoothness of the map), and then continue with an extragradient method applied to the full system. The local convergence rate will still be with iteration complexity due to the matrix multiplications required in computing (see Proposition 9).
A.16 Deep Learning Specifications and Results
We also experimented on common neural-net driven tasks. We tested with on a mixture of Gaussians and on CIFAR10 against , i.e., . Introducing a small term can help accelerate training (see Figure 5).
A.16.2 Mixture of Gaussians Network Architectures
Both the generator and discriminator are fully connected neural networks. The relevant hyperparameters for setting up the GAN are itemized below.
was used with and was used with .
A.16.3 Images at End of Training for CIFAR10
A.16.4 CIFAR10 Network Architectures
Both the generator and discriminator are convolutional neural networks; we copied the architectures used in . The generator consists of a linear layer, followed by 4 deconvolution layers ( kernel, stride, leaky ReLU, 64 hidden channels), followed by a final linear layer with a tanh nonlinearity. The discriminator consists of 4 convolution layers ( kernel, stride, leaky ReLU, 64 hidden channels) followed by a linear layer. The relevant hyperparameters for setting up the GAN are itemized below.
was used with and was used with .