Global Convergence to the Equilibrium of GANs using Variational Inequalities

Ian Gemp, Sridhar Mahadevan

Introduction

When minimizing f(x)f(x) over x∈Xx\in\mathcal{X}, it is known that ff decreases fastest if xx moves in the direction −∇f(x)-\nabla f(x). In addition, any direction orthogonal to −∇f(x)-\nabla f(x) will leave f(x)f(x) 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 ii in a game updates with x(i)←x(i)−ρ∇x(i)f(i)(x)x^{(i)}\leftarrow x^{(i)}-\rho\nabla_{x^{(i)}}f^{(i)}(x), x=[x(1);x(2);…]⊤x=[x^{(1)};x^{(2)};\ldots]^{\top} 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 O(N/k)\mathcal{O}(N/k) 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 (min⁡\min player) aims to synthesize realistic data samples by transforming vectors drawn from a fixed source distribution, e.g., N(0,Id)\mathcal{N}(\mathbf{0},I_{d}). The discriminator (max⁡\max 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, GG, and discriminator’s scoring function, DD, are typically chosen to be neural networks parameterized by weights θ\theta and ϕ\phi respectively. The minimax objective of the original GAN is

where p(z)p(z) is the source distribution, p(y)p(y) is the true data distribution, and g(x)=−log⁡(1+e−x)g(x)=-\log(1+e^{-x}).

In practice, finding the solution to (1) consists of local updates, e.g., SGD, to θ\theta and ϕ\phi. This continues until 1) VV 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 GG is a linear function, DD is a quadratic function, g(x)=xg(x)=x, and p(z)p(z) is Gaussian . Follow-up research has shown that, in this setting, the optimal generator distribution is a rank-kk Gaussian containing the top-kk principal components of the data . Furthermore, it is shown that if the dimensionality of p(z)p(z) matches that of p(y)p(y), 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 x∈Xx\in\mathcal{X} and x′∈Xx^{\prime}\in\mathcal{X}, FF is

If, in Equation (2), “≥\geq” is replaced by “>>”, then FF is strictly-monotone; if “≥\geq” is replaced by “s∣∣x−x′∣∣2s||x-x^{\prime}||^{2}”, then FF is ss-strongly-monotone. If FF 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 FF evaluated at x∗x^{*} is HurwitzOur definition of Hurwitz is equivalent to the more standard: −J-J is Hurwitz if max⁡i[Re(λi(−J))]<0\max_{i}[\text{Re}(\lambda_{i}(-J))]<0. , 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 ϵ\epsilon-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: x(t)=∑icivieλitx(t)=\sum_{i}c_{i}v_{i}e^{\lambda_{i}t}; Euler’s formula: e(a+ib)t=eat(cos⁡(bt)+isin⁡(bt))e^{(a+ib)t}=e^{at}(\cos(bt)+i\sin(bt))..

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: G(z)=Az+bG(z)=Az+b and D(y)=y⊤W2y+w1⊤yD(y)=y^{\top}W_{2}y+w_{1}^{\top}y. Equation (1) becomes

The map FF associated with this zero-sum game is constructed by concatenating the gradients of the two players’ losses (fG=V,fD=−Vf_{G}=V,f_{D}=-V):

Crossing-the-Curl

In this section, we will derive our proposed technique, Crossing-the-Curl, motivated by an examination of the (w1,bw_{1},b)-subsystem of LQ-GAN, i.e., (w2,a)(w_{2},a) fixed at (0,a0)(0,a_{0}) for any a0a_{0}. 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 Fw1,bF^{w_{1},b} is not Hurwitz, and simultaneous gradient descent, defined in Equation (7), will diverge for this problem (see A.5). However, Fw1,bF^{w_{1},b} is monotone (J+J⊤=0)(J+J^{\top}=\mathbf{0}) and 1−1-Lipschitz in the sense that ∣∣Fw1,b(x)−Fw1,b(x′)∣∣2≤1∣∣x−x′∣∣2||F^{w_{1},b}(x)-F^{w_{1},b}(x^{\prime})||^{2}\leq 1||x-x^{\prime}||^{2}. Table 1 offers an extragradient method (see Figure 2) with O(1/k)\mathcal{O}(1/k) 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 FF 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 FF and the axis of rotation, we can take their cross product:

where ∇F\nabla_{F} is Feynman notation for the gradient with respect to FF only and ∣v=F|_{v=F} means evaluate the expression at v=Fv=F. The \nicefrac−12\nicefrac{{-1}}{{2}} 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 (I−η(J−J⊤))F(I-\eta(J-J^{\top}))F as FηccF_{\eta cc}.

Note that the fixed point of FccF_{cc} remains the same as the original field FF. Furthermore, the reader may recognize FccF_{cc} as the gradient of the function 12(w12+(b−μ)2)\frac{1}{2}(w_{1}^{2}+(b-\mu)^{2}), which is strongly convex, allowing an O(e−k)\mathcal{O}(e^{-k}) convergence rate in the deterministic setting. FccF_{cc} 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 (w1,bw_{1},b)-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 (J⊤<spanclass="katex−display"><spanclass="katex"><spanclass="katex−mathml"><mathxmlns="http://www.w3.org/1998/Math/MathML"display="block"><semantics><mrow><mo>=</mo></mrow><annotationencoding="application/x−tex">=</annotation></semantics></math></span><spanclass="katex−html"aria−hidden="true"><spanclass="base"><spanclass="strut"style="height:0.3669em;"></span><spanclass="mrel">=</span></span></span></span></span>−JJ^{\top}<span class="katex-display"><span class="katex"><span class="katex-mathml"><math xmlns="http://www.w3.org/1998/Math/MathML" display="block"><semantics><mrow><mo>=</mo></mrow><annotation encoding="application/x-tex">=</annotation></semantics></math></span><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.3669em;"></span><span class="mrel">=</span></span></span></span></span>-J) for the (w1,bw_{1},b)-subsystem. In the more general case, where JJ 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 FccF_{cc}. To our knowledge, FegF_{eg} 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 FF is (n−1)(n-1) dimensional, which means (J⊤<spanclass="katex−display"><spanclass="katex"><spanclass="katex−mathml"><mathxmlns="http://www.w3.org/1998/Math/MathML"display="block"><semantics><mrow><mo>−</mo></mrow><annotationencoding="application/x−tex">−</annotation></semantics></math></span><spanclass="katex−html"aria−hidden="true"><spanclass="base"><spanclass="strut"style="height:0.6667em;vertical−align:−0.0833em;"></span><spanclass="mord">−</span></span></span></span></span>J)F(J^{\top}<span class="katex-display"><span class="katex"><span class="katex-mathml"><math xmlns="http://www.w3.org/1998/Math/MathML" display="block"><semantics><mrow><mo>−</mo></mrow><annotation encoding="application/x-tex">-</annotation></semantics></math></span><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.6667em;vertical-align:-0.0833em;"></span><span class="mord">−</span></span></span></span></span>J)F is no longer the unique direction orthogonal to FF. However, every matrix can be decomposed into a symmetric part with real eigenvalues, \nicefrac12(J+J⊤)\nicefrac{{1}}{{2}}(J+J^{\top}), and a skew-symmetric part with purely imaginary eigenvalues, \nicefrac12(J−J⊤)\nicefrac{{1}}{{2}}(J-J^{\top}). Notice that for an optimization problem, J<spanclass="katex−display"><spanclass="katex"><spanclass="katex−mathml"><mathxmlns="http://www.w3.org/1998/Math/MathML"display="block"><semantics><mrow><mo>−</mo></mrow><annotationencoding="application/x−tex">−</annotation></semantics></math></span><spanclass="katex−html"aria−hidden="true"><spanclass="base"><spanclass="strut"style="height:0.6667em;vertical−align:−0.0833em;"></span><spanclass="mord">−</span></span></span></span></span>J⊤<spanclass="katex−display"><spanclass="katex"><spanclass="katex−mathml"><mathxmlns="http://www.w3.org/1998/Math/MathML"display="block"><semantics><mrow><mo>=</mo></mrow><annotationencoding="application/x−tex">=</annotation></semantics></math></span><spanclass="katex−html"aria−hidden="true"><spanclass="base"><spanclass="strut"style="height:0.3669em;"></span><spanclass="mrel">=</span></span></span></span></span>H<spanclass="katex−display"><spanclass="katex"><spanclass="katex−mathml"><mathxmlns="http://www.w3.org/1998/Math/MathML"display="block"><semantics><mrow><mo>−</mo></mrow><annotationencoding="application/x−tex">−</annotation></semantics></math></span><spanclass="katex−html"aria−hidden="true"><spanclass="base"><spanclass="strut"style="height:0.6667em;vertical−align:−0.0833em;"></span><spanclass="mord">−</span></span></span></span></span>H⊤J<span class="katex-display"><span class="katex"><span class="katex-mathml"><math xmlns="http://www.w3.org/1998/Math/MathML" display="block"><semantics><mrow><mo>−</mo></mrow><annotation encoding="application/x-tex">-</annotation></semantics></math></span><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.6667em;vertical-align:-0.0833em;"></span><span class="mord">−</span></span></span></span></span>J^{\top}<span class="katex-display"><span class="katex"><span class="katex-mathml"><math xmlns="http://www.w3.org/1998/Math/MathML" display="block"><semantics><mrow><mo>=</mo></mrow><annotation encoding="application/x-tex">=</annotation></semantics></math></span><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.3669em;"></span><span class="mrel">=</span></span></span></span></span>H<span class="katex-display"><span class="katex"><span class="katex-mathml"><math xmlns="http://www.w3.org/1998/Math/MathML" display="block"><semantics><mrow><mo>−</mo></mrow><annotation encoding="application/x-tex">-</annotation></semantics></math></span><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.6667em;vertical-align:-0.0833em;"></span><span class="mord">−</span></span></span></span></span>H^{\top}== where HH 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 FccF_{cc}. In addition, FccF_{cc} 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 FccF_{cc} as preconditioning FF 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 x⊤Ax≡0x^{\top}Ax\equiv 0 for any skew-symmetric matrix AA, so calling (J⊤−J)(J^{\top}-J) 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, JFJF, can be approximated accurately and efficiently with finite differences. Likewise, J⊤FJ^{\top}F can be computed efficiently with double backprop by taking the gradient of \nicefrac12∣∣F∣∣2\nicefrac{{1}}{{2}}||F||^{2}. In total, three backprops are required, one for F(xk)F(x_{k}), one for F(x^k+1)F(\hat{x}_{k+1}), and one for \nicefrac12∣∣F(xk)∣∣2\nicefrac{{1}}{{2}}||F(x_{k})||^{2}.

In our analysis, we also consider the gradient regularization proposed in , FregF_{reg}, the Unrolled GAN proposed in , FunrF_{unr}, alternating gradient descent, FaltF_{alt}, as well as any linear combination of FF, JFJF, and J⊤FJ^{\top}F, deemed FlinF_{lin}, which forms a family of maps that includes FegF_{eg}, FconF_{con}, and FccF_{cc}:

Keep in mind that we are proposing FlinF_{lin} as a generalization of Crossing-the-Curl. We state our main results here for the (w1,b)(w_{1},b)-subsystem.

For any α\alpha, Flinw1,bF^{w_{1},b}_{lin} with at least one of β\beta and γ\gamma positive and both non-negative is strongly monotone. Also, its Jacobian is Hurwitz. See Proposition 13.

Fccw1,bF^{w_{1},b}_{cc}, Fηccw1,bF^{w_{1},b}_{\eta cc}, Fegw1,bF^{w_{1},b}_{eg}, and Fconw1,bF^{w_{1},b}_{con} with η>0\eta>0 are strongly-monotone with Hurwitz Jacobians. See Proposition 1.

Faltw1,bF^{w_{1},b}_{alt}, Funrw1,bF^{w_{1},b}_{unr}, Fw1,bF^{w_{1},b}, and Fregw1,bF^{w_{1},b}_{reg} with any η\eta are monotone, but not strictly monotone. Of these maps, only Fregw1,bF^{w_{1},b}_{reg}’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 FF contains quadratic terms suggesting it may welcome a quasimonotone analysis, however, the remaining maps all contain 3rd degree terms. Unsurprisingly, analyzing quasimonotonicity for FlinF_{lin} 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: xx, x′x^{\prime}, yy, y′y^{\prime}. 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 x∈Xx\in\mathcal{X} and x∗∈Xx^{*}\in\mathcal{X} such that F(x∗)=0F(x^{*})=0, we have that F(x)⊤(x−x∗)≥0F(x)^{\top}(x-x^{*})\geq 0.

FF is quasimonotone on X\mathcal{X} only if (A) holds, i.e. (A) is necessary but not sufficient.

FF is pseudomonotone on X\mathcal{X} 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, −F-F, must not be leading away from the equilibrium anywhere.

Equipped with these definitions, we can conclude the following:

None of the maps, including FlinF_{lin} 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 FlinF_{lin} 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 (w1,bw_{1},b)-subsystem, we shift focus to the (w2,aw_{2},a)-subsystem assuming the mean has already been learned exactly, i.e., b=μb=\mu. We will revisit this assumption later.

We can conclude the following for the (w2,aw_{2},a)-subsystem:

Fw2,aF^{w_{2},a}, Fregw2,aF^{w_{2},a}_{reg}, Funrw2,aF^{w_{2},a}_{unr}, Faltw2,aF^{w_{2},a}_{alt}, and Fconw2,aF^{w_{2},a}_{con} are not quasimonotone. Also, their Jacobians are not Hurwitz. See Propositions 14 through 19.

Fegw2,aF^{w_{2},a}_{eg} and Fccw2,aF^{w_{2},a}_{cc} are pseudomonotone which implies an O(1/k)\mathcal{O}(1/\sqrt{k}) stochastic convergence rate. See Propositions 21 and 24. Their Jacobians are not Hurwitz. See Proposition 27.

No monotone Flinw2,aF^{w_{2},a}_{lin} 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 Fegw2,aF^{w_{2},a}_{eg} and Fccw2,aF^{w_{2},a}_{cc} by rescaling with \nicefrac14a2\nicefrac{{1}}{{4a^{2}}}: (12)→\rightarrow(13) and (14)→\rightarrow(15) respectively. This results in strongly-monotone and strongly-convex systems respectively, improving the stochastic convergence rate to O(1/k)\mathcal{O}(1/k). In deriving these results, we assumed the mean was given. We can relax this assumption and analyze the (w2,aw_{2},a)-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 Fccw1,bF^{w_{1},b}_{cc} are required to achieve a 1−δ1-\delta probability of the mean being accurate enough to ensure the (w2,aw_{2},a)-subsystem is strongly-monotone. Note that this approach of first learning the mean, then the variance retains the overall O(1/k)\mathcal{O}(1/k) stochastic rate. We summarize the main points here.

A nonlinear scaling of Fegw2,aF^{w_{2},a}_{eg} and Fccw2,aF^{w_{2},a}_{cc} results in strictly monotone and \nicefrac12\nicefrac{{1}}{{2}}-strongly monotone subsystems respectively. See Proposition 29.

If the mean is first well approximated, i.e., b2≤μ2+σ2b^{2}\leq\mu^{2}+\sigma^{2}, then Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} remains 1) \nicefrac12\nicefrac{{1}}{{2}}-strongly-monotone if the (w1,bw_{1},b)-subsystem is “shut off” or 2) strictly-monotone if the (w1,b)(w_{1},b)-subsystem is re-weighted with a high coefficient. See Propositions 30 and 31.

FegW2,AF^{W_{2},A}_{eg} and FccW2,AF^{W_{2},A}_{cc} are not quasimonotone for the 2-d LQ-GAN system (with and without (AA⊤)−1(AA^{\top})^{-1} 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, (w1,bw_{1},b), then learning the more complex, (w2,aw_{2},a)-subsystem. This theoretically confirms the intuition behind progressive training of GANs , which have generated the highest quality images to date.

Thirdly, because Jw2,acc′J^{cc^{\prime}}_{w_{2},a} is symmetric (and ≻0\succ 0), we can integrate Fw2,acc′F^{cc^{\prime}}_{w_{2},a} to discover the convex function it is implicitly descending via gradient descent: fw2,acc′=1/2[(a2−σ2)−σ2log⁡(\nicefraca2σ2)]f^{cc^{\prime}}_{w_{2},a}=1/2[(a^{2}-\sigma^{2})-\sigma^{2}\log(\nicefrac{{a^{2}}}{{\sigma^{2}}})]. Compare this to KL-divergence: KL(σ∣∣a)=1/2[(σ2/a2)+log⁡(\nicefraca2σ2)−1]KL(\sigma||a)=1/2[(\sigma^{2}/a^{2})+\log(\nicefrac{{a^{2}}}{{\sigma^{2}}})-1]. In contrast to KLKL, fw2,acc′f^{cc^{\prime}}_{w_{2},a} is convex in aa and may be a desirable alternative due to less extreme gradients near a=0a=0.

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 AA 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 (w1,bw_{1},b)-subsystem. This implies that Crossing-the-Curl applied successively to each row of AA can solve for A∗A^{*}; pseudocode is presented in Algorithm 1 in Appendix A.15. Note that this procedure is reminiscent of the Cholesky–Banachiewicz algorithm which computes AA row by row, beginning with the first row. The resulting algorithm is O(N/k)\mathcal{O}(N/k).

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 20549\mathbf{20549} iterations with a 0.88\mathbf{0.88} success rate. For the 4-d stochastic LQ-GAN using two-sample minibatch estimates, this procedure achieves ∣∣xk−x∗∣∣/∣∣x0−x∗∣∣<0.1||x_{k}-x^{*}||/||x_{0}-x^{*}||<0.1 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 D(y)=w2(y−w1)2D(y)=w_{2}(y-w_{1})^{2}. We will also work on devising heuristics for setting the coefficients of FlinF_{lin}.

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. fi(xi,x−i)f_{i}(x_{i},x_{-i}) is convex in xix_{i}. 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 x¨=−F(x)\ddot{x}=-F(x), not just first order ODEs, however, the focus is on systems that separate control, uu, and state xx, i.e. x˙=−F(x(t),u(t),t)\dot{x}=-F(x(t),u(t),t). 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 (C,K)(\mathbf{C},K) be a cost minimization game with player cost functions CiC_{i} and feasible set KK. Let x∗x^{*} be a Nash equilibrium. Let F=[∂C1∂x1,…,∂CN∂xN]F=[\frac{\partial C_{1}}{\partial x_{1}},\ldots,\frac{\partial C_{N}}{\partial x_{N}}]. Then

where IK(x∗)\mathbf{I}_{K}(x^{*}) is the internal cone at x∗x^{*}. When Ci(xi,x−i)C_{i}(\mathbf{x}_{i},\mathbf{x}_{-i}) is pseudoconvex in xi\mathbf{x}_{i} for all ii, this condition is also sufficient. Note that this is implied if FF is pseudomonotone, i.e. pseudomonotonicity of FF is a stronger condition.

A.3 Table of Maps Considered in Analysis

All maps corresponding to the (w1,bw_{1},b)-subsystem in Table 4 maintain the desired unique fixed point, F(x∗)=0F(x^{*})=0, where x∗=(w1∗,b∗)=(0,μ)x^{*}=(w_{1}^{*},b^{*})=(0,\mu).

For the (w2,aw_{2},a)-subsystem, all maps except FlinF_{lin} with certain settings of (α,β,γ\alpha,\beta,\gamma) and FconF_{con} maintain the desired unique fixed point, x∗=(w2∗,a∗)=(0,σ)x^{*}=(w_{2}^{*},a^{*})=(0,\sigma). FconF_{con} introduces an additional spurious fixed point at

FconF_{con} is a special case of FlinF_{lin} where α=1\alpha=1, β=1\beta=1, and γ=0\gamma=0.

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 AA 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 A=Σ1/2A=\Sigma^{1/2} is necessarily full rank. Note this implies A⊤A^{\top} is also full rank. The null space of a full rank matrix is the zeros vector, which implies W2+W2⊤=0W_{2}+W_{2}^{\top}=0. W2W_{2} is symmetric, so this implies W2=0W_{2}=0. ∎

Consider the case where the mean of p(z)p(z) 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. ∣∣xk+1−x∗∣∣2/∣∣xk−x∗∣∣2>1||x_{k+1}-x^{*}||^{2}/||x_{k}-x^{*}||^{2}>1.

Therefore, simultaneous gradient descent diverges from the equilibrium of the (w1,bw_{1},b)-subsystem for any step size scheme, ρk\rho_{k}.

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 x∈Xx\in\mathcal{X} and x′∈Xx^{\prime}\in\mathcal{X},

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 v=(x∗−x)/tv=(x^{*}-x)/t.

FF is quasimonotone on X\mathcal{X} if and only if (A) and (B’) hold.

FF is pseudomonotone on X\mathcal{X} if and only if (A) and (C’) hold.

A.8 A Comparison of Monotonicity and Hurwitz

The monotonicity and Hurwitz properties are complementary.

Let F(x)=JxF(x)=Jx, J=[14−11]J=\begin{bmatrix}1&4\\ -1&1\end{bmatrix}, S=[01−10]S=\begin{bmatrix}0&1\\ -1&0\end{bmatrix}, and v=SJx=[−x1+x2,−x1−4x2]⊤v=SJx=[-x_{1}+x_{2},-x_{1}-4x_{2}]^{\top}. Then λ1,2(J)=1±2i\lambda_{1,2}(J)=1\pm 2i so JJ is Hurwitz, and

which, by condition (A), implies FF is not quasimonotone.

A.8.2 Monotonicity Does Not Imply Hurwitz

Let F(x)=JxF(x)=Jx and J=[01−10]J=\begin{bmatrix}0&1\\ -1&0\end{bmatrix}. Then λ1,2(J)=±i\lambda_{1,2}(J)=\pm i so JJ is not Hurwitz, but

A.8.3 Monotonicity and Hurwitz Can Overlap

Let F(x)=JxF(x)=Jx and J=[1001]J=\begin{bmatrix}1&0\\ 0&1\end{bmatrix}. Then λ1,2(J)=1\lambda_{1,2}(J)=1 so JJ is Hurwitz and

If FF is (strictly,strongly)-monotone, then the Jacobian of FF 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 F(x)=JxF(x)=Jx.

Increase in condition number: κ=\nicefrac115→4\kappa=\nicefrac{{11}}{{5}}\rightarrow 4.

Feg′w2,aF^{w_{2},a}_{eg^{\prime}} becomes non-monotone.

Crossing-the-Curl forces monotonicity for normal, affine fields.

Let F=Jx+bF=Jx+b and assume JJ is normal, i.e., JJ⊤=J⊤JJJ^{\top}=J^{\top}J. Then

Unrolled GANs and Alternating Updates are Monotone for the (w1,bw_{1},b)-subsystem.

In Unrolled GANs, the generator computes the gradient of VV assuming the discriminator has already made several updates. Define the discriminator’s update as

and denote the composition of UU, Δk\Delta k-times as

where Δk\Delta k is some positive integer. Then the update for Unrolled GANs is

In the case of the (w1,bw_{1},b)-subsystem, we can write these unrolled updates out explicitly. Remember F=[b−μ,−w1]⊤F=[b-\mu,-w_{1}]^{\top}, so

where the corresponding map is Funr=[bk−μ,ρΔk(bk−μ)−w1,k]F^{unr}=[b_{k}-\mu,\rho\Delta k(b_{k}-\mu)-w_{1,k}]. Taking a look at the Jacobian, we find

Here, we considered updating bb first, but the (w1,bw_{1},b)-subsystem is perfectly symmetric, so the analysis holds either way. If w1w_{1} is updated first, this is equivalent to Unrolled GAN with Δk=1\Delta k=1 (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. ∎

FlinF_{lin}, FccF_{cc}, FegF_{eg}, and FconF_{con} are strongly-monotone for the (w1,bw_{1},b)-subsystem (includes multivariate case). FF and FregF_{reg} are monotone, but not strictly monotone. Moreover, FlinF_{lin}, FccF_{cc}, FegF_{eg}, FconF_{con}, and FregF_{reg} are Hurwitz for the (w1,bw_{1},b)-subsystem (includes multivariate case). FF is not Hurwitz.

We start with the original map, Fw1,bF^{w_{1},b}, 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, JJ is not Hurwitz.

Now we analyze Fccw1,bF^{w_{1},b}_{cc}, Fegw1,bF^{w_{1},b}_{eg}, and Fconw1,bF^{w_{1},b}_{con}, which as discussed in the main body, are equivalent.

The symmetrized Jacobian is positive definite with a minimum eigenvalue of 11, therefore this system is 11-strongly-monotone. By Proposition 10, the Jacobians of these maps are Hurwitz for the (w1,b)(w_{1},b)-subsystem.

Now we analyze the generalization Flinw1,b=(αI−βJ⊤−γJ)Fw1,bF^{w_{1},b}_{lin}=(\alpha I-\beta J^{\top}-\gamma J)F^{w_{1},b}.

The symmetrized Jacobian is positive definite with a minimum eigenvalue of (β+γ)(\beta+\gamma), therefore this system is (β+γ)(\beta+\gamma)-strongly-monotone. By Proposition 10, Jlinw1,bJ^{w_{1},b}_{lin} is Hurwitz for the (w1,b)(w_{1},b)-subsystem.

Now we analyze the regularized-gradient algorithm, Fregw1,bF^{w_{1},b}_{reg}.

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, Jregw1,bJ^{w_{1},b}_{reg} is Hurwitz.

Note that for FccF_{cc}, FegF_{eg}, FconF_{con}, and FlinF_{lin}, JJ is symmetric, therefore, FF is the gradient of some function, f(w1,b)=12(w12+(b−μ)2)f(w_{1},b)=\frac{1}{2}(w_{1}^{2}+(b-\mu)^{2}). Also, note that the standard algorithm with step size ρk=1k+1\rho_{k}=\frac{1}{k+1} is equivalent to the standard running estimate of the mean: μk+1=kk+1μk+1k+1xk\mu_{k+1}=\frac{k}{k+1}\mu_{k}+\frac{1}{k+1}x_{k} where xkx_{k} is the kk-th sample.

Specifically, we first consider the sign of β+γ\beta+\gamma. Lemma 1 rules out negative values. Lemma 2 rules out positive values when σ2≤\nicefrac12\sigma^{2}\leq\nicefrac{{1}}{{2}}, and Lemma 3 rules out positive values when σ2>\nicefrac12\sigma^{2}>\nicefrac{{1}}{{2}}. Corollary 2 concludes that β+γ=0\beta+\gamma=0.

Next, given β+γ=0\beta+\gamma=0, we consider the sign of α\alpha. Lemmas 4 and 5 rule out positive values of α\alpha when β\beta is greater than or less than or equal to zero respectively, i.e., α\alpha cannot be positive. Similarly, Lemmas 6 and 7 rule out negative values of α\alpha when β\beta is less than or greater than or equal to zero respectively, i.e., α\alpha cannot be negative. Corollary 3 concludes that α=0\alpha=0.

Lastly, given that β+γ=α=0\beta+\gamma=\alpha=0, Lemmas 8 and 9 prove that β\beta cannot be greater than or less than or equal to zero respectively. Corollary 4 concludes that β=γ=0\beta=\gamma=0. Therefore, the only quasimonotone linear combination is the trivial one resulting in F=0F=0, which completes the proof.

For FlinF_{lin} to be quasimonotone, β+γ\beta+\gamma must not be strictly less than zero, i.e. β+γ<0\beta+\gamma\cancel{<}0.

If (β+γ)<0(\beta+\gamma)<0, then this system is not quasimonotone. Therefore, assume (β+γ)≥0(\beta+\gamma)\geq 0 from now on. ∎

If σ2≤12\sigma^{2}\leq\frac{1}{2}, for FlinF_{lin} to be quasimonotone, β+γ\beta+\gamma must not be strictly greater than zero, i.e. β+γ>0\beta+\gamma\cancel{>}0.

We will use a different parameterization of FlinF_{lin} 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 FlinF_{lin} with β^>0\hat{\beta}>0 by considering two different cases.

Case 1: Consider the (w2,a)(w_{2},a)-subsystem. Let

Above, we premultiply FlinF_{lin} by a skew symmetric matrix, which ensures v⊤Flin=Flin⊤AskewFlin=0v^{\top}F_{lin}=F_{lin}^{\top}A_{skew}F_{lin}=0.

The relevant portion of the Jacobian of FlinF_{lin} is

Consider x=[0,0,cσ,0]x=[0,0,c\sigma,0] and both β^\hat{\beta} and α\alpha fixed.

This implies either α=0\alpha=0 or β^≤0\hat{\beta}\leq 0 for the system to be quasimonotone.

Case 2: Consider the (a,b)(a,b)-subsystem. Let

The relevant portion of the Jacobian of FlinF_{lin} is

Consider α=0\alpha=0 and x=[0,0,σ10,σ2]x=[0,0,\frac{\sigma}{10},\frac{\sigma}{2}]. Then

Then α=0⇒β^≤0\alpha=0\Rightarrow\hat{\beta}\leq 0. In either case, β^\hat{\beta} must be nonpositive. Therefore, β^=β+γ>0\hat{\beta}=\beta+\gamma\cancel{>}0. ∎

Part of the proof in Lemma 2 looks at the limit in which aa approaches 0. One might presume a simple fix is to constrain aa to be larger than some small value, e.g., 1e-10, and use a large β^\hat{\beta} value. Here, we show that even using a=σ100a=\frac{\sigma}{100} 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 aa that maintains quasimonotonicity within the feasible region.

Consider x=[0,−1,σ100,σ2]x=[0,-1,\frac{\sigma}{100},\frac{\sigma}{2}] and the (a,b)(a,b)-subsystem as in 2. Then

If β^>0\hat{\beta}>0, then this is a concave quadratic form in α\alpha. To find where this function is positive, we need to find its roots.

The α\alpha root with smaller magnitude provides an upper bound for β^2\hat{\beta}^{2}.

Now consider again x=[0,0,σ100,0]x=[0,0,\frac{\sigma}{100},0] and equation 141 with c=1100c=\frac{1}{100}.

This provides a lower bound for β^2\hat{\beta}^{2}.

The upper bound we require for β^\hat{\beta} is greater than the lower bound, therefore, no β^\hat{\beta} will satisfy quasimonotonicity. ∎

If σ2>12\sigma^{2}>\frac{1}{2}, for FlinF_{lin} to be quasimonotone, β+γ\beta+\gamma must not be strictly greater than zero, i.e. β+γ>0\beta+\gamma\cancel{>}0.

For this proof, we make use of the traditional definition of quasimonotonicity. Consider

If (β+γ)>0(\beta+\gamma)>0, then this system is not quasimonotone. In either case, (β+γ)>0(\beta+\gamma)\cancel{>}0. ∎

𝛽𝛾0\beta+\gamma=0 for quasimonotonicity.). Together, Lemmas 1, 2 and 3 imply that (β+γ)(\beta+\gamma) must be to satisfy quasimonotonicity.

If (β+γ)=0(\beta+\gamma)=0 and α>0\alpha>0, for FlinF_{lin} to be quasimonotone, β\beta must not be strictly greater than zero, i.e. β>0\beta\cancel{>}0.

For this proof, we make use of the traditional definition of quasimonotonicity. Consider

If (β+γ)=0(\beta+\gamma)=0 and α>0\alpha>0, then β≤0\beta\leq 0 for the system to be quasimonotone. ∎

If (β+γ)=0(\beta+\gamma)=0, for FlinF_{lin} to be quasimonotone, α\alpha must not be strictly greater than zero, i.e. α>0\alpha\cancel{>}0.

We will assume α>0\alpha>0, which by Lemma 4 implies β≤0\beta\leq 0. This will lead to a contradiction. Consider

If (β+γ)=0(\beta+\gamma)=0 and α>0\alpha>0 (implies β≤0\beta\leq 0), then ⟨F(y),x−y⟩>0\langle F(y),x-y\rangle>0 and ⟨F(x),x−y⟩<0\langle F(x),x-y\rangle<0, which breaks quasimonotonicity. Therefore, α>0\alpha\cancel{>}0. ∎

If (β+γ)=0(\beta+\gamma)=0 and α<0\alpha<0, for FlinF_{lin} to be quasimonotone, β\beta must not be strictly less than zero, i.e. β<0\beta\cancel{<}0.

If α<0\alpha<0, then β≥0\beta\geq 0 to maintain quasimonotonicity. ∎

If (β+γ)=0(\beta+\gamma)=0, for FlinF_{lin} to be quasimonotone, α\alpha must not be strictly less than zero, i.e. α<0\alpha\cancel{<}0.

We will assume α<0\alpha<0, which by 6 implies β≥0\beta\geq 0. This will lead to a contradiction.

If (β+γ)=0(\beta+\gamma)=0 and α<0\alpha<0 (implies β≥0\beta\geq 0), then ⟨F(y),x−y⟩>0\langle F(y),x-y\rangle>0 and ⟨F(x),x−y⟩<0\langle F(x),x-y\rangle<0, which breaks quasimonotonicity. Therefore, α≥0\alpha\geq 0. ∎

Together, Corollary 2 and Lemmas 4-7 imply that α\alpha must equal zero for FlinF_{lin} to be quasimonotone.

If (β+γ)=0(\beta+\gamma)=0 and α=0\alpha=0, for FlinF_{lin} to be quasimonotone, β\beta must not be strictly greater than zero, i.e. β>0\beta\cancel{>}0.

If β>0\beta>0, then this system is not quasimonotone. Therefore, β≤0\beta\leq 0. ∎

If (β+γ)=0(\beta+\gamma)=0 and α=0\alpha=0, for FlinF_{lin} to be quasimonotone, β\beta must not be strictly less than zero, i.e. β<0\beta\cancel{<}0.

If β<0\beta<0, then this system is not quasimonotone. Therefore, β≥0\beta\geq 0. ∎

𝛽𝛾0𝛼0⇒𝛽𝛾0(\beta+\gamma)=0,\alpha=0\Rightarrow\beta=\gamma=0). Together, Lemmas 8 and 9 imply that β=0\beta=0, which, along with Corollary 2, imply that γ=0\gamma=0 as well.

[α=β=γ=0\alpha=\beta=\gamma=0] Together, Corollaries 2 and 3, and 4 imply that there is no non-trivial linear combination that induces a quasimonotone LQ-GAN system.

FccF_{cc}, FegF_{eg}, FconF_{con}, and FF are not quasimonotone for the LQ-GAN system.

These maps are all linear combinations of FF, JFJF and J⊤FJ^{\top}F, therefore, by Corollary 5, they are not quasimonotone for the LQ-GAN system. ∎

Note that if a map is not quasimonotone for the (w2,a)(w_{2},a)-subsystem, then it is not quasimonotone for the full system. This is because an analysis of the (w2,a)(w_{2},a)-subsystem is equivalent to an analysis of a subspace of the full system with w1=b=0w_{1}=b=0.

FF is not quasimontone for the (w2,a)(w_{2},a)-subsystem. Also, its Jacobian is not Hurwitz.

The Jacobian of FF for the (w2,aw_{2},a)-subsystem is

The trace of Jw2,aJ^{w_{2},a} is strictly negative for w2>0w_{2}>0, which implies Jw2,aJ^{w_{2},a} has an eigenvalue with strictly negative real part. Therefore, Jw2,aJ^{w_{2},a} is not Hurwitz. ∎

FregF_{reg} is not quasimonotone for the (w2,a)(w_{2},a)-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 x=[w2,0,a,0]x=[w_{2},0,a,0] and let vv be defined as follows:

where vv is actually derived by considering the field formed by crossing the curl for the 2-d subspace with w2w_{2} and aa only.

It suffices to consider the submatrix of the Jacobian corresponding to w2w_{2} and aa only when computing v⊤Jvv^{\top}Jv:

If w2>0w_{2}>0 and a<σ3a<\frac{\sigma}{\sqrt{3}}, then there isn’t an η≥0\eta\geq 0 that will make this system quasimonotone.

The Jacobian of Fregw2,aF^{w_{2},a}_{reg} for the (w2,aw_{2},a)-subsystem is

The trace of Jw2,aJ^{w_{2},a} is strictly negative for w2>0w_{2}>0 and a<σ/3a<\sigma/\sqrt{3}, which implies Jregw2,aJ^{w_{2},a}_{reg} has an eigenvalue with strictly negative real part. Therefore, Jregw2,aJ^{w_{2},a}_{reg} is not Hurwitz. ∎

FunrF_{unr} is not quasimonotone or Hurwitz for the (w2,a)(w_{2},a)-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 α>0\alpha>0 is a step size, and denote the composition of UU, Δk\Delta k-times as

where Δk\Delta k is some positive integer. Then the update for Unrolled GANs is

In the case of the (w2,aw_{2},a)-subsystem, we can write these unrolled updates out explicitly. Remember F=[a2−σ2,−2aw2]F=[a^{2}-\sigma^{2},-2aw_{2}], so

We will use the following vector to test condition (A) for quasimonotonicity of FunrF_{unr}:

Computing v⊤Junrvv^{\top}J_{unr}v and evaluating at (w2=1,a=σ23)(w_{2}=1,a=\frac{\sigma^{2}}{\sqrt{3}}) gives

therefore, FunrF_{unr} is not quasimonotone.

If we examine the determinant of JunrJ_{unr} and evaluate it at a=σ3a=\frac{\sigma}{\sqrt{3}}, we get

which is less than zero for positive w2w_{2}. Therefore, the Jacobian exhibits negative eigenvalues which means the system is not Hurwitz. ∎

FaltF_{alt} is not quasimonotone or Hurwitz for the (w2,a)(w_{2},a)-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 α>0\alpha>0 is a step size. The corresponding map is

Note the similarity to the Unrolled GAN map Equation (242). The maps are equivalent if Δk=\nicefrac12\Delta k=\nicefrac{{1}}{{2}}. Unrolled GANs was shown to be not quasimonotone for any Δk\Delta k, therefore, FaltF_{alt} is not quasimonotone as well.

If we examine the trace of JaltJ_{alt} and evaluate it at (w2=5ασ2,a=σw_{2}=5\alpha\sigma^{2},a=\sigma), 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 w2w_{2}. This implies that v⊤Jalt′vv^{\top}J_{alt^{\prime}}v changes sign locally around this root when varying w2w_{2}, which means v⊤Jalt′v<0v^{\top}J_{alt^{\prime}}v<0 for some positive w2w_{2}. Therefore Falt′F_{alt^{\prime}} is not quasimonotone.

If we examine the determinant of Jalt′J_{alt^{\prime}} and evaluate it at (w2=1,a=σw_{2}=1,a=\sigma), we get

which is less than zero for positive w2w_{2}. Therefore, the Jacobian exhibits negative eigenvalues which means the system is not Hurwitz. ∎

The following propositions concern the monotonicity of FccF_{cc}, FegF_{eg}, and FconF_{con} for the (w2,aw_{2},a)-subsystem. The field and Jacobian for FlinF_{lin} will be helpful for proofs of their properties.

Fcon=F+βJ⊤FF_{con}=F+\beta J^{\top}F is not quasimontone for the (w2,aw_{2},a)-subsystem. Also, its Jacobian is not Hurwitz.

This corresponds to FlinF_{lin} with α=1,β=β,γ=0\alpha=1,\beta=\beta,\gamma=0. We consider three cases. Let

which implies β≥0\beta\geq 0 for the system to be quasimonotone.

which, combined with above, implies β≥12σ≈0.707σ\beta\geq\frac{1}{\sqrt{2}\sigma}\approx\frac{0.707}{\sigma} for the system to be quasimonotone.

Case 3: Consider x=[2σ,σ]x=[2\sigma,\sigma]. 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 ≈0.273σ\approx\frac{0.273}{\sigma}. This implies β≤≈0.273σ\beta\leq\approx\frac{0.273}{\sigma} for the system to be quasimonotone.

The last two results cannot be satisfied by a single β\beta, therefore, this system is not quasimonotone.

For completeness, we analyze the limit where the FF term is ignored. Consider a=cσa=c\sigma.

This is negative for c=1c=1, therefore, this system is not quasimonotone.

The trace of Jconw2,aJ^{w_{2},a}_{con} is strictly negative for w2=0w_{2}=0 and a<σ/5a<\sigma/\sqrt{5}, which implies Jconw2,aJ^{w_{2},a}_{con} has an eigenvalue with strictly negative real part. Therefore, Jconw2,aJ^{w_{2},a}_{con} is not Hurwitz. ∎

Fcon=βJ⊤FF_{con}=\beta J^{\top}F is not quasimontone for the (w2,aw_{2},a)-subsystem. Also, its Jacobian is not Hurwitz.

This corresponds to FlinF_{lin} with α=0,β=β,γ=0\alpha=0,\beta=\beta,\gamma=0. We consider two cases.

which, for c≠1c\neq 1, implies β≥0\beta\geq 0 for the system to be quasimonotone.

Case 2: Consider x=[2cσ,cσ]x=[2c\sigma,c\sigma]. Then

which, for c=1c=1, implies β≤0\beta\leq 0 for the system to be quasimonotone. Combined with above, this implies β=0\beta=0 for the system to be quasimonotone. In conclusion, βJ⊤F\beta J^{\top}F is not quasimonotone.

The trace of Jconw2,aJ^{w_{2},a}_{con} is strictly negative for w2=0w_{2}=0 and a<σ/5a<\sigma/\sqrt{5}, which implies Jconw2,aJ^{w_{2},a}_{con} has an eigenvalue with strictly negative real part. Therefore, Jconw2,aJ^{w_{2},a}_{con} is not Hurwitz. ∎

Feg=F−γJFF_{eg}=F-\gamma JF requires γ→∞\gamma\rightarrow\infty to be pseudomonotone for (w2,aw_{2},a)-subsystem

This corresponds to FlinF_{lin} with α=1,β=0,γ=γ\alpha=1,\beta=0,\gamma=\gamma. We consider two cases.

Case 1: Consider y=[σ,3σ]y=[\sigma,3\sigma] and x=[3σ,5σ]x=[3\sigma,5\sigma]. Then

Then γ≤−136σ≈−0.027σ\gamma\leq-\frac{1}{36\sigma}\approx-\frac{0.027}{\sigma} or γ≥160σ≈0.017σ\gamma\geq\frac{1}{60\sigma}\approx\frac{0.017}{\sigma} for the system to be quasimonotone.

Case 2: Consider y=[σ,20σ]y=[\sigma,20\sigma] and x=[20σ,5σ]x=[20\sigma,5\sigma]. Then

Then γ≥8181207800σ≈0.039σ\gamma\geq\frac{8181}{207800\sigma}\approx\frac{0.039}{\sigma} or γ≥1084825σ≈−0.022σ\gamma\geq\frac{108}{4825\sigma}\approx-\frac{0.022}{\sigma} 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 γ≥160σ\gamma\geq\frac{1}{60\sigma}. This implies γ\gamma must be arbitrarily large for small σ\sigma. In the limit, the effect of FF on the system is negligible. We consider this limit next. ∎

Feg=−γJFF_{eg}=-\gamma JF is pseudomonotone for (w2,aw_{2},a)-subsystem.

Note this system is 2-d, therefore, there is only 1 vector vv (aside from scaling) that is perpendicular to FF.

This satisfies conditions (A) and (C), therefore, this system is pseudomonotone. ∎

Feg=F−γJFF_{eg}=F-\gamma JF is pseudomonotone for the constrained (w2,a)(w_{2},a)-subsystem.

We consider α=1\alpha=1 in this case and let the user define a feasible region for which they are confident the equilibrium exists: w2∈[w2min⁡,w2max⁡]w_{2}\in[w_{2}^{\min},w_{2}^{\max}] and a∈[amin⁡,amax⁡]a\in[a_{\min},a_{\max}]—the most important bounds being those on aa. We will attempt to find a value for γ\gamma 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 w2w_{2} that maximizes this equation for a given aa 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 ⟨F(x),x−x∗⟩=0\langle F(x),x-x^{*}\rangle=0 at equilibrium. Continuing and looking at the numerator of the derivative, we find

If we plug that back into the lower bound for γ\gamma, we get

The condition above along with the following (see condition (A)) are sufficient to ensure pseudomonotonicity.

If w2≤0w_{2}\leq 0, 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 w2>0w_{2}>0. We can divide the analysis into two cases.

Consider 3a2≥σ23a^{2}\geq\sigma^{2}. In this case, all coefficients of γ\gamma terms except a γ1\gamma^{1} term and the last term (the constant) are positive. For simplicity, we can find the value for γ\gamma such that the first part of the β2\beta^{2} coefficient is greater than the two negative terms.

Now consider 3a2<σ23a^{2}<\sigma^{2}. One of the terms in the γ1\gamma^{1} coefficient is now negative. We will find a value for γ\gamma such that the γ3\gamma^{3} term can drown out that negative term.

Note this bound is not tight; it is just meant to provide a satisfactory estimate. ∎

Fcc=F+β(J⊤−J)FF_{cc}=F+\beta(J^{\top}-J)F requires β→∞\beta\rightarrow\infty to be pseudomonotone for the (w2,aw_{2},a)-subsystem.

This corresponds to FlinF_{lin} with α=1,γ=β/2,β=β/2\alpha=1,\gamma=\beta/2,\beta=\beta/2.

this, combined with above, implies that β≥12σ\beta\geq\frac{1}{\sqrt{2}\sigma}.

This implies β\beta must be arbitrarily large for small σ\sigma. In the limit, the effect of FF on the system is negligible. We consider this limit in Subsubsection 24. ∎

Fcc=(J⊤−J)FF_{cc}=(J^{\top}-J)F is pseudomonotone for the (w2,a)(w_{2},a)-subsystem.

Note that the skew part of the Jacobian of FF is full rank except at the boundary (a=0a=0), so Fcc=(J⊤−J)FF_{cc}=(J^{\top}-J)F maintains the same fixed points. This can be seen by looking at FccF_{cc} above. We will simply need to constrain aa 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 (w2,aw_{2},a)-subsystem.

Case 1: Consider the point x=[w2,a]x=[w_{2},a] and let vv be defined as follows:

v⊤Fccw2,av^{\top}F^{w_{2},a}_{cc} is 0 as expected.

Now, we will compute v⊤Jccw2,avv^{\top}J^{w_{2},a}_{cc}v to see if it is greater than zero.

In addition to this, proving that ⟨F(x),x−x∗⟩≥0\langle F(x),x-x^{*}\rangle\geq 0 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 aa is constrained to be non-negative. Therefore, FccF_{cc} is pseudomonotone. ∎

Fcc=F+β(J⊤−J)FF_{cc}=F+\beta(J^{\top}-J)F is pseudomonotone for the constrained (w2,a)(w_{2},a)-subsystem.

We consider α=1\alpha=1 in this case and let the user define a feasible region for which they are confident the equilibrium exists: w2∈[w2min⁡,w2max⁡]w_{2}\in[w_{2}^{\min},w_{2}^{\max}] and a∈[amin⁡,amax⁡]a\in[a_{\min},a_{\max}]—the most important bounds being those on aa. We will attempt to find a value for β\beta 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 w2w_{2} that maximizes this equation for a given aa 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 ⟨F(x),x−x∗⟩=0\langle F(x),x-x^{*}\rangle=0 at equilibrium. Continuing and looking at the numerator of the derivative, we find

If we plug that back into the lower bound for β\beta, we get

The condition above along with the following (see condition (A)) are sufficient to ensure pseudomonotonicity.

If w2≤0w_{2}\leq 0, 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 w2>0w_{2}>0. We can divide the analysis into two cases.

Consider 3a2≥σ23a^{2}\geq\sigma^{2}. In this case, all coefficients of β\beta terms except the last term (the constant) are positive. For simplicity, we can find the value for β\beta such that the first part of the β3\beta^{3} coefficient is greater than the last term (the constant).

Now consider 3a2<σ23a^{2}<\sigma^{2}. One of the terms in the β1\beta^{1} coefficient is now negative. We will find a value for β\beta such that the β3\beta^{3} term can drown out the two negative terms.

This last lower bound is the greatest of the three, so it suffices to set β\beta greater than this value to ensure the system is pseudomonotone within the given feasible region. ∎

FlinF_{lin} is not monotone for the (w2,a)w_{2},a)-subsystem (before scaling).

Let Flinw2,aF^{w_{2},a}_{lin} 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 β+γ>0\beta+\gamma>0. If a<σ/5a<\sigma/\sqrt{5} and w2=0w_{2}=0, then the trace is less than zero.

Assume β+γ<0\beta+\gamma<0. If a>σ/5a>\sigma/\sqrt{5} and w2=0w_{2}=0, then the trace is less than zero.

If w2<0w_{2}<0, then β≤α4w2\beta\leq\frac{\alpha}{4w_{2}}. If w2>0w_{2}>0, then β≥α4w2\beta\geq\frac{\alpha}{4w_{2}}. Therefore, β=α4w2\beta=\frac{\alpha}{4w_{2}}, however, β\beta and α\alpha are constants while w2w_{2} is a variable. Therefore, α\alpha and β\beta must equal zero to satisfy this for all w2w_{2} proving that no monotone linear combination exists. ∎

FlinF_{lin} is not Hurwitz for the (w2,a)w_{2},a)-subsystem.

Consider Jlinw2,aJ^{w_{2},a}_{lin} at w2=0w_{2}=0.

If β+γ<0\beta+\gamma<0, then a>σ/5a>\sigma/\sqrt{5} implies the existence of an eigenvalue with negative real part. If β+γ>0\beta+\gamma>0, then a<σ/5a<\sigma/\sqrt{5} implies the existence of an eigenvalue with negative real part. If β+γ=0\beta+\gamma=0, then the real part is zero. ∎

There exists an Flin′F_{lin^{\prime}} family after scaling by \nicefrac14a2\nicefrac{{1}}{{4a^{2}}} that exhibits strict-monotonicity.

If we consider the same linear combinations above, but divide FF by 4a24a^{2}, we can obtain a family of monotone fields (see Mathematica notebook).

The trace of the corresponding symmetrized Jacobian is

For constant β\beta and γ\gamma and nonzero α\alpha, there exists a value for w2w_{2} that will force the trace to be negative, therefore α\alpha must be zero. Note that γ\gamma must be greater than or equal to β\beta to ensure that the trace cannot be made negative in the limit as w22w_{2}^{2} grows to infinity.

Case 1: Consider the case where β=γ\beta=\gamma. Then for any fixed β\beta, γ\gamma, and nonzero α\alpha,

Case 2: Otherwise, consider solving the quadratic form for w2w_{2} when β+γ>0\beta+\gamma>0:

For the trace to be non-negative, we need the leading coefficient of the quadratic to be positive, i.e., γ−β>0\gamma-\beta>0. We also need there to be at most 1 real root, meaning the square root must be non-positive. If β+γ>0\beta+\gamma>0, then setting aa and σ\sigma using the following formula will force the root to be positive:

For example, set a=σa=\sigma, and then set σ\sigma and w2w_{2} as follows to force the trace to be negative:

Case 3: If β+γ≤0\beta+\gamma\leq 0, then the root is necessarily positive. Therefore, α\alpha must be set to zero.

The field and Jacobian are now wieldy enough to state:

and is non-negative as long as both β+γ≥0\beta+\gamma\geq 0 and γ−β≥0\gamma-\beta\geq 0.

which is non-negative as long as, in addition to the previous conditions, we have β≥0\beta\geq 0. The trace and determinant are both strictly positive if β+γ>0\beta+\gamma>0.

In summary, Flin′w2,aF^{w_{2},a}_{lin^{\prime}} is strictly-monotone, i.e., Jlin′w2,a≻0J^{w_{2},a}_{lin^{\prime}}\succ 0, if γ≥β≥0\gamma\geq\beta\geq 0 and γ>0\gamma>0. ∎

The Flin′F_{lin^{\prime}} family includes Feg′F_{eg^{\prime}} (γ=γ,β=0)(\gamma=\gamma,\beta=0) and Fcc′F_{cc^{\prime}} (γ=β)(\gamma=\beta). By Proposition 28, Feg′F_{eg^{\prime}} and Fcc′F_{cc^{\prime}} are at least strictly-monotone.

Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} is \nicefrac12\nicefrac{{1}}{{2}}-strongly monotone and Feg′w2,aF^{w_{2},a}_{eg^{\prime}} is only strictly-monotone.

Case Fcc′w2,aF^{w_{2},a}_{cc^{\prime}}: The eigenvalues of Jcc′w2,aJ^{w_{2},a}_{cc^{\prime}} are λ1=1\lambda_{1}=1 and \lambda_{2}=\frac{1}{2}\Big{(}1+\frac{\sigma^{2}}{a^{2}}\Big{)}. Therefore, Jcc′w2,a⪰12J^{w_{2},a}_{cc^{\prime}}\succeq\frac{1}{2} and Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} is \nicefrac12\nicefrac{{1}}{{2}}-strongly monotone.

Case Feg′w2,aF^{w_{2},a}_{eg^{\prime}}: The eigenvalues of a 2×22\times 2 matrix can be written in terms of the trace and determinant as

Therefore, if the term 4DetTr2\frac{4Det}{Tr^{2}} 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 Jeg′w2,aJ^{w_{2},a}_{eg^{\prime}} are

This term can be made arbitrarily small as w2w_{2} goes to infinity. To be more rigorous, let a=σ=1a=\sigma=1 so that Tr=2+w22Tr=2+w_{2}^{2} and Det=1Det=1. Then

An application of L’Hopital’s rule shows that

The minimum eigenvalue only approaches zero in the limit, so Feg′w2,aF^{w_{2},a}_{eg^{\prime}} is strictly-monotone. ∎

Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} 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 Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} is symmetric and PSD, therefore it is the Hessian of some convex function. We can integrate Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} to arrive at a convex function (with arbitrary constant). Integrating Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} results in the following:

Note that fcc′w2,af^{w_{2},a}_{cc^{\prime}} must be convex along the subspace with w2=0w_{2}=0 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 p(y)p(y) and p(z)p(z). There are two ways to learn both the mean and variance of a distribution using Fccw2,aF^{w_{2},a}_{cc}. 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 y∼p(y)y\sim p(y) lie in [ylow,yhi][y_{low},y_{hi}]. 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, 1−δ1-\delta, the (w1,b)(w_{1},b)-subsystem can be “shut-off” and the (w2,aw_{2},a)-subsystem safely “turned-on” resulting in a \nicefrac12\nicefrac{{1}}{{2}}-strongly-monotone Fcc′w2,aF^{w_{2},a}_{cc^{\prime}}.

We begin by observing the symmetrized Jacobian of Fcc′w2,aF^{w_{2},a}_{cc^{\prime}}:

where G=μ2+σ2−b2G=\mu^{2}+\sigma^{2}-b^{2}. In order for Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} to be strongly monotone, we require G≥0G\geq 0. In other words, the square of the generator’s estimate of the mean, bkb_{k}, learned from training the (w1,bw_{1},b)-subsystem needs to be less than or equal to μ2+σ2\mu^{2}+\sigma^{2}.

Assume ∣bk−μ∣<t|b_{k}-\mu|<t and introduce a scalar: 0<d<10<d<1. Remember, we require bk2<μ2+σ2b_{k}^{2}<\mu^{2}+\sigma^{2}. And we know μ−t<bk<μ+t\mu-t<b_{k}<\mu+t which implies

This expression has two roots for tt, one positive and one negative. ∣bk−μ∣|b_{k}-\mu| can only be upper bounded by a positive number, so we select the positive root.

Plugging t+t_{+} back into equation (377) for tt, we find that

Rearranging (376) and plugging in tt, we can derive the number of iterations required:

If we assume p(y)∼N(μ,σ2)p(y)\sim\mathcal{N}(\mu,\sigma^{2}) and use a Chernoff bound, we find

The number of samples needed to maintain stability of the system grows as the true mean μ\mu 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., w2y2w_{2}y^{2} rather than w2(y−μ)2w_{2}(y-\mu)^{2}. This may explain why batch norm is so helpful (almost required) in stabilizing training.

Assume all y∼p(y)y\sim p(y) lie in [ylow,yhi][y_{low},y_{hi}]. 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, 1−δ1-\delta, the (w1,bw_{1},b)-subsystem can be up-weighted and the (w2,aw_{2},a)-subsystem “turned-on”, resulting in a strictly-monotone LQ-GAN.

As before, assume we are running Fccw1,bF^{w_{1},b}_{cc} on the (w1,bw_{1},b)-subsystem and Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} on the (w2,aw_{2},a)-subsystem. Also, multiply Fccw1,bF^{w_{1},b}_{cc} by e>0e>0, i.e., increase the learning rate by ee or divide the learning rate of Fcc′w2,aF^{w_{2},a}_{cc^{\prime}} by ee. The full symmetrized Jacobian of this system is:

The upper left 2×22\times 2 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 G=μ2+σ2−b2G=\mu^{2}+\sigma^{2}-b^{2} as before. We need G≥0G\geq 0 for Trab>0Tr_{ab}>0 (for lim⁡a→0+\lim_{a\rightarrow 0+}) and 2eG≥b22eG\geq b^{2} for Det>0Det>0. As before, Hoeffding’s inequality says kk iterations are required for an accurate estimate of the mean (see Equation (383)). And as before, we find that G=(1−d)σ2G=(1-d)\sigma^{2}. We will focus on the determinant condition here. Let

More simply, let d=1/2d=1/2. Then set e>μmax⁡2σmin⁡2+12e>\frac{\mu_{\max}^{2}}{\sigma_{\min}^{2}}+\frac{1}{2}. 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 H=2eG−b2H=2eG-b^{2} and note that H<2eσ2H<2e\sigma^{2} (see Equation (389)), i.e., HH 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, kk, is increasing), dd is effectively decreasing towards zero. In the limit lim⁡d→0+e≥μ22σ2\lim_{d\rightarrow 0+}e\geq\frac{\mu^{2}}{2\sigma^{2}}.

Given, [ylow,yhi][y_{low},y_{hi}], we can set μmax=max⁡(∣ylow∣,∣yhi∣)\mu_{max}=\max(|y_{low}|,|y_{hi}|). Also, note that if the distribution is known to support ϵ\epsilon balls at the ends of the specified interval, [ylow,yhi][y_{low},y_{hi}], with some nonzero probabilities, PlowP_{low} and PhiP_{hi}, 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 AA be a lower triangular matrix with positive diagonal—AA represents the generator’s guess at the square root of Σ\Sigma.

The 2-d LQ-GAN is not quasimonotone for FccF_{cc} or FegF_{eg} 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 vcc=KFccv_{cc}=KF_{cc} and veg=KFegv_{eg}=KF_{eg}. Similarly, with scaling, let vcc′=KFcc′v_{cc^{\prime}}=KF_{cc^{\prime}} and veg′=KFeg′v_{eg^{\prime}}=KF_{eg^{\prime}}. Let

This implies that neither system is quasimonotone (with, cc′/eg′cc^{\prime}/eg^{\prime}, or without, cc/egcc/eg, scaling). ∎

The 2-d LQ-GAN with W11W_{11} and A11A_{11} already learned, i.e., W11=0W_{11}=0 and A11=A11∗A_{11}=A_{11}^{*}, is not quasimonotone for FccF_{cc} or FegF_{eg}.

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 vcc=KFccv_{cc}=KF_{cc} and veg=KFegv_{eg}=KF_{eg}. Let

This implies that neither system is quasimonotone. ∎

The 3-d LQ-GAN with the diagonal of AA already learned, i.e., Aii=Aii∗A_{ii}=A_{ii}^{*}, is not quasimonotone for FccF_{cc} or FegF_{eg} 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 vcc=KFccv_{cc}=KF_{cc} and veg=KFegv_{eg}=KF_{eg}. Let

This implies that neither system is quasimonotone. ∎

The N-d LQ-GAN with all but a single row of AA fixed is strictly-monotone for FccF_{cc}, FegF_{eg}, and FconF_{con}.

First, note that the Cholesky decomposition of Σ\Sigma, denoted by A∗A^{*}, obeys the follow equation:

where i<ji<j. Σ\Sigma is symmetric, so Σji\Sigma_{ji} can be recovered as Σij\Sigma_{ij}. This allows us to remove 1 degree of freedom from the system by defining the diagonal term in a single row of AA in terms of the other entries in the row:

where as before AiiA_{ii} must be greater than zero. We assume that Σii\Sigma_{ii} has already been learned by Crossing-the-Curl as described in the main body. The condition Aii>0A_{ii}>0 can be ensured by constraining ∑d=1i−1Aid2≤Σii−ϵ\sum_{d=1}^{i-1}A^{2}_{id}\leq\Sigma_{ii}-\epsilon with ϵ≪1\epsilon\ll 1—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 NNth row of AA. Take N=3N=3 for example. Notice that the 33rd row of AA, A3:,A_{3:}, only contains the following W2W_{2} terms: W13,W23W_{13},W_{23}. 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 FW2F_{W_{2}} is zero only if Equation (421) is satisfied for ΣiN\Sigma_{iN} and WdN=0W_{dN}=0 for all d<Nd<N. Therefore, setting all other entries of W2W_{2} 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 F=[FW2;FA]F=[F_{W_{2}};F_{A}]:

which is skew-symmetric and constant with respect to the variables being learned: W2,i<NW_{2,i<N} and AN>iA_{N>i}. Therefore, J+J⊤=0J+J^{\top}=0 is PSD, which implies FF is monotone. The fact that JJ is constant along with Proposition 11 imply that Fcc=Feg=Fcon=−JFF_{cc}=F_{eg}=F_{con}=-JF are also monotone:

Note that the component of FccF_{cc} corresponding to the dynamics of AA, is independent of W2W_{2}. This means the dynamics are now decoupled from W2W_{2} and can be run separately. By inspecting the symmetrized Jacobian of FccF_{cc} we can show that it is a block matrix composed of positive definite matrices:

A:d−1A:d−1⊤A_{:d-1}A_{:d-1}^{\top} is positive definite because AA is constrained to be of Cholesky form. Moreover, the eigenvalues of A:d−1⊤A:d−1A_{:d-1}^{\top}A_{:d-1} are the same as A:d−1A:d−1⊤A_{:d-1}A_{:d-1}^{\top}, therefore both blocks are positive definite. This implies the entire matrix JsymJ_{sym} is positive definite which means Fcc=Feg=FconF_{cc}=F_{eg}=F_{con} are strictly monotone. Note that we do not require A:d−1=A:d−1∗A_{:d-1}=A^{*}_{:d-1} for strict monotonicity. In practice, the system will actually be both strongly-monotone and smooth. This is because AA is constrained with a projection onto a ball and the diagonal of AA is restricted to be larger than ϵ\epsilon. These two conditions guarantee a nonzero, finite minimum and maximum value for the eigenvalues of A:d−1A:d−1⊤A_{:d-1}A_{:d-1}^{\top}—the minimum corresponds to strong-monotonicity and the maximum corresponds to smoothness. ∎

Unlike the (w2,aw_{2},a)-subsystem where monotonicity depends on the accuracy of the learned mean, this system is monotone as long as A:d−1A_{:d-1} is PSD which is guaranteed from the form we have prescribed to AA. This result suggests learning the rows of AA in succession, and each subsystem is guaranteed to be strictly monotone. Note that the variance, i.e., diagonal of Σ\Sigma, will be slightly off the true value if the mean, μ\mu, is not first learned perfectly. The learned AA will then be slightly off the true A∗A^{*} and errors will compound, but still not affect monotonicity. The subsystems corresponding to each row of AA can be revisited to learn the entries of AA more accurately. Permuting the dimensions of xx 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 μ\mu and Σ\Sigma for the deterministic LQ-GAN.

As mentioned above, the maps for learning the mean and variance are both strongly convex which implies a O(1/k)\mathcal{O}(1/k) stochastic convergence rate for each, the sum of which is still O(1/k)\mathcal{O}(1/k).

In practice, the maps for learning each row of AA are strongly-monotone and smooth (see last paragraph of proof of Proposition 35) which implies a O(1/k)\mathcal{O}(1/k) stochastic convergence rate for each as well. Because this technique consists of N+1N+1 steps for learning the full NN-d LQ-GAN, it requires k^=Nk\hat{k}=Nk iterations which, in total, implies a O(N/k)\mathcal{O}(N/k) stochastic convergence rate.

Hidden within this analysis is the fact that each iteration of learning the mean and variance is O(N)\mathcal{O}(N) in terms of time-complexity and each iteration for learning each row of AA is O(N2)\mathcal{O}(N^{2}), therefore this entire procedure is O(N3/k)\mathcal{O}(N^{3}/k) in terms of FLOPS. This is expected as the complexity of a Cholesky decomposition to compute A=Σ1/2A=\Sigma^{1/2} is also O(N3)\mathcal{O}(N^{3}). Note that unlike the complexity of computing FF 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 O(N/k)\mathcal{O}(N/k) convergence rate and not O(1/k)\mathcal{O}(1/k).

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 JDDJ_{DD} is positive definite. The symmetrized Jacobian is then

This implies FF is monotone where F=[∇VA,b;−∇VW2,w1]F=[\nabla V_{A,b};-\nabla V_{W_{2},w_{1}}]. 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 O(1/k)\mathcal{O}(1/k) with O(N3)\mathcal{O}(N^{3}) iteration complexity due to the matrix multiplications required in computing FF (see Proposition 9).

A.16 Deep Learning Specifications and Results

We also experimented on common neural-net driven tasks. We tested FlinF_{lin} with (α,β,γ)=(1,10,10−4)(\alpha,\beta,\gamma)=(1,10,10^{-4}) on a mixture of Gaussians and (α,β,γ)=(1,10,0.1)(\alpha,\beta,\gamma)=(1,10,0.1) on CIFAR10 against FconF_{con}, i.e., (α,β,γ)=(1,10,0)(\alpha,\beta,\gamma)=(1,10,0). Introducing a small −JF-JF 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.

FconF_{con} was used with β=1.0\beta=1.0 and FlinF_{lin} was used with (α,β,γ)=(1.0,1.0,0.001)(\alpha,\beta,\gamma)=(1.0,1.0,0.001).

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 (5×55\times 5 kernel, 2×22\times 2 stride, leaky ReLU, 64 hidden channels), followed by a final linear layer with a tanh nonlinearity. The discriminator consists of 4 convolution layers (5×55\times 5 kernel, 2×22\times 2 stride, leaky ReLU, 64 hidden channels) followed by a linear layer. The relevant hyperparameters for setting up the GAN are itemized below.

FconF_{con} was used with β=10.0\beta=10.0 and FlinF_{lin} was used with (α,β,γ)=(1.0,10.0,0.0001)(\alpha,\beta,\gamma)=(1.0,10.0,0.0001).