Year: 2020

Abstract:
Adaptive gradient algorithms perform gradient-based updates using the history of gradients and are ubiquitous in training deep neural networks. While adaptive gradient methods theory is well understood for minimization problems, the underlying factors driving their empirical success in min-max problems such as GANs remain unclear. In this paper, we aim at bridging this gap from both theoretical and empirical perspectives. First, we analyze a variant of Optimistic Stochastic Gradient (OSG) proposed in (Daskalakis et al., 2017) for solving a class of nonconvex non-concave min-max problem and establish O( −4) complexity for finding -first-order stationary point, in which the algorithm only requires invoking one stochastic first-order oracle while enjoying state-of-the-art iteration complexity achieved by stochastic extragradient method by (Iusem et al., 2017). Then we propose an adaptive variant of OSG named Optimistic Adagrad (OAdagrad) and reveal an improved adaptive complexity O ( − 2 1−α ) , where α characterizes the growth rate of the cumulative stochastic gradient and 0 ≤ α ≤ 1/2. To the best of our knowledge, this is the first work for establishing adaptive complexity in nonconvex non-concave min-max optimization. Empirically, our experiments show that indeed adaptive gradient algorithms outperform their non-adaptive counterparts in GAN training. Moreover, this observation can be explained by the slow growth rate of the cumulative stochastic gradient, as observed empirically.

text: ( −
2 1−α ) , where α characterizes the
growth rate of the cumulative stochastic gradient and 0 ≤ α ≤ 1/2. To the best of our knowledge, this is the first work for establishing adaptive complexity in nonconvex non-concave min-max optimization. Empirically, our experiments show that indeed adaptive gradient algorithms outperform their non-adaptive counterparts in GAN training. Moreover, this observation can be explained by the slow growth rate of the cumulative stochastic gradient, as observed empirically.

heading: 1 INTRODUCTION
text: Adaptive gradient algorithms (Duchi et al., 2011; Tieleman & Hinton, 2012; Kingma & Ba, 2014; Reddi et al., 2019) are very popular in training deep neural networks due to their computational efficiency and minimal need for hyper-parameter tuning (Kingma & Ba, 2014). For example, Adagrad (Duchi et al., 2011) automatically adjusts the learning rate for each dimension of the model parameter according to the information of history gradients, while its computational cost is almost the same as Stochastic Gradient Descent (SGD). However, in supervised deep learning (for example, image classification tasks using a deep convolutional neural network), there is not enough evidence showing that adaptive gradient methods converge faster than its non-adaptive counterpart (i.e., SGD) on benchmark datasets. For example, it is argued in (Wilson et al., 2017) that adaptive gradient methods often find a solution with worse performance than SGD. Specifically, Wilson et al. (2017) observed that Adagrad has slower convergence than SGD in terms of both training and testing error, while using VGG (Simonyan & Zisserman, 2014) on CIFAR10 data.
GANs (Goodfellow et al., 2014) are a popular class of generative models. In a nutshell, they consist of a generator and a discriminator, both of which are defined by deep neural networks. The generator and the discriminator are trained under an adversarial cost, corresponding to a non-convex non-concave min-max problem. GANs are known to be notoriously difficult to train. In practice, Adam (Kingma & Ba, 2014) is the defacto optimizer used for GAN training. The common optimization strategy is to alternatively update the discriminator and the generator (Arjovsky et al.,
∗Correspondence to mingrui-liu@uiowa.edu
2017; Gulrajani et al., 2017). Using Adam is important in GAN training, since replacing it with non-adaptive methods (e.g. SGD) would significantly deteriorate the performance. This paper studies and attempts to answer the following question:
Why do adaptive gradient methods outperform their non-adaptive counterparts in GAN training?
We analyze a variant of Optimistic Stochastic Gradient (OSG) in (Daskalakis & Panageas, 2018) and propose an adaptive variant named Optimistic Adagrad (OAdagrad) for solving a class of nonconvex non-concave min-max problems. Both of them are shown to enjoy state-of-the-art complexities. We further prove that the convergence rate of OAdagrad to an -first-order stationary point depends on the growth rate of the cumulative stochastic gradient. In our experiments, we observed an interesting phenomenon while using adaptive gradient methods for training GANs: the cumulative stochastic gradient grows at a slow rate. This observation is in line with the prediction of our theory suggesting improved convergence rate for OAdagrad in GAN training, when the growth rate of the cumulative stochastic gradient is slow.
Since GAN is a min-max optimization problem in nature, our problem of interest is to solve the following stochastic optimization problem:
min u∈U max v∈V
F (u,v) := Eξ∼D [f(u,v; ξ)] , (1)
where U , V are closed and convex sets, F (u,v) is possibly non-convex in u and non-concave in v. ξ is a random variable following an unknown distribution D. In GAN training, u, v represent the parameters of generator and discriminator respectively.
The ideal goal for solving (1) is to find a saddle point (u∗,v∗) ∈ U × V such that F (u∗,v) ≤ F (u∗,v∗) ≤ F (u,v∗) for ∀u ∈ U ,∀v ∈ V . To achieve this goal, the typical assumption usually made is that the objective function is convexconcave. When F (u,v) is convex in u and concave in v, non-asymptotic guarantee in terms of the duality gap is well established by a series of work (Nemirovski & Yudin, 1978; Nemirovski, 2004; Nesterov, 2007; Nemirovski et al., 2009; Juditsky et al., 2011). However, when F (u,v) is non-convex in u and non-concave in v, finding the saddle point is NP-hard in general. Instead, we focus on finding the first-order stationary point provided that the objective function is smooth. I.e. we aim to find (u,v) ∈ U × V such that ∇uF (u,v) = 0, ∇vF (u,v) = 0. Note that this is a necessary condition for finding the (local) saddle point.
Related Work. Several works designed iterative first-order deterministic (Dang & Lan, 2015) and stochastic (Iusem et al., 2017; Lin et al., 2018) algorithms for achieving the -first-order stationary point with non-asymptotic guarantee. The goal is to find x such that ‖T (x)‖ ≤ or E [‖T (x)‖] ≤ , where the first-order oracle is defined as T (x) = [∇uF (u,v),−∇vF (u,v)]> with x = (u,v) and the first-order stochastic oracle is the noisy observation of T , i.e. T (x; ξ) = [∇uF (u,v; ξ),−∇vF (u,v; ξ)]>. For instance, Dang & Lan (2015) focuses on the deterministic setting. On the other hand, (Iusem et al., 2017) develops a stochastic extra-gradient algorithm that enjoys O( −4) iteration complexity. The extra-gradient method requires two stochastic first-order oracles in one iteration, which can be computationally expensive in deep learning applications such as GANs. The inexact proximal point method developed in (Lin et al., 2018) has iteration complexity O( −6) for finding an -first-order stationary point 1.
To avoid the cost of an additional oracle call in extragradient step, several studies (Chiang et al., 2012; Rakhlin & Sridharan, 2013; Daskalakis et al., 2017; Gidel et al., 2018; Xu et al., 2019) proposed single-call variants of the extragradient algorithm. Some of them focus on the convex setting (e.g. (Chiang et al., 2012; Rakhlin & Sridharan, 2013)), while others focus on the non-convex setting (Xu et al., 2019). The closest to our work is the work by (Daskalakis et al., 2017; Gidel et al., 2018), where the min-max setting and GAN training are considered. However, the convergence of those algorithms is only shown for a class of bilinear problems in (Daskalakis et al., 2017) and for monotone variational inequalities in (Gidel et al., 2018). Hence a big gap remains between the specific settings studied in (Daskalakis et al., 2017; Gidel et al., 2018) and more general non-convex
1The result in (Lin et al., 2018) assumes the first-order oracle T is a weakly-monotone operator, which is milder than the Lipschitz-continuity assumption as assumed in Iusem et al. (2017). However, simply applying the Lipschitz-continuity condition in their proof does not change their iteration complexity.
non-concave min-max problems. Table 1 provides a complete overview of our results and existing results. It is hard to give justice to the large body of work on min-max optimization, so we refer the interested reader to Appendix B that gives a comprehensive survey of related previous methods that are not covered in this Table.
Our main goal is to design stochastic first-order algorithms with low iteration complexity, low periteration cost and suitable for a general class of non-convex non-concave min-max problems. The main tool we use in our analysis is variational inequality.
Let T : Rd 7→ Rd be an operator and X ⊂ Rd is a closed convex set. The Stampacchia Variational Inequality (SVI) problem (Hartman & Stampacchia, 1966) is defined by the operator T and X and denoted by SVI(T,X ). It consists of finding x∗ ∈ X such that 〈T (x∗),x− x∗〉 ≥ 0 for ∀x ∈ X . A similar one is Minty Variational Inequality (MVI) problem (Minty et al., 1962) denoted by MVI(T,X ), which consists of finding x∗ such that 〈T (x),x− x∗〉 ≥ 0 for ∀x ∈ X . Min-max optimization is closely related to variational inequalities. The corresponding SVI and MVI for the min-max problem are defined through T (x) = [∇uF (u,v),−∇vF (u,v)]> with x = (u,v). Our main contributions are summarized as follows:
• Following (Daskalakis et al., 2017), we extend optimistic stochastic gradient (OSG) analysis beyond the bilinear and unconstrained case, by assuming the Lipschitz continuity of the operator T and the existence of a solution for the variational inequality MVI(T,X ). These conditions were considered in the analysis of the stochastic extragradient algorithm in (Iusem et al., 2017). We analyze a variant of Optimistic Stochastic Gradient (OSG) under these conditions, inspired by the analysis of (Iusem et al., 2017). We show that OSG achieves state-of-the-art iteration complexityO(1/ 4) for finding an -first-order stationary point. Note that our OSG variant only requires invoking one stochastic first-order oracle
3Note that the pseudo-monotonicity assumption used by (Iusem et al., 2017) can also be replaced by our MVI assumption in their proof. The main difference between our OSG and the stochastic extragradient method in (Iusem et al., 2017) is the number of stochastic gradient calculations in each iteration.
4Here the bilinear game is defined as minu∈Rp maxv∈Rq u>Av+u>a+bv>, where the smallest singular value of A ∈ Rp×q is positive, a ∈ Rp×1, b ∈ R1×q .
5Here n, L, µ denote the number of components in the finite sum structure, Lipschitz constant and strongmonotonicity parameter of the operator of variational inequality respectively.
while enjoying the state-of-the-art iteration complexity achieved by stochastic extragradient method (Iusem et al., 2017).
• Under the same conditions, we design an adaptive gradient algorithm named Optimistic Adagrad (OAdagrad), and show that it enjoys better adaptive complexity O
( −
2 1−α ) ,
where α characterizes the growth rate of cumulative stochastic gradient and 0 ≤ α ≤ 1/2. Similar to Adagrad (Duchi et al., 2011), our main innovation is in considering variable metrics according to the geometry of the data in order to achieve potentially faster convergence rate for a class of nonconvex-nonconcave min-max games. Note that this adaptive complexity improves upon the non-adaptive one (i.e., O(1/ 4)) achieved by OSG. To the best of our knowledge, we establish the first known adaptive complexity for adaptive gradient algorithms in a class of non-convex non-concave min-max problems.
• We demonstrate the effectiveness of our algorithms in GAN training on CIFAR10 data. Empirical results identify an important reason behind why adaptive gradient methods behave well in GANs, which is due to the fact that the cumulative stochastic gradient grows in a slow rate. We also show that OAdagrad outperforms Simultaneous Adam in sample quality in ImageNet generation using self-attention GANs (Zhang et al., 2018). This confirms the superiority of OAdagrad in min-max optimization.

heading: 2 PRELIMINARIES AND NOTATIONS
text: In this section, we fix some notations and give formal definitions of variational inequalities, and their relationship to the min-max problem (1).
Notations. LetX ⊂ Rd be a closed convex set, and ‖·‖ the euclidean norm. We note ΠX the projection operator, i.e. ΠX (y) = arg minx∈X ‖y − x‖2. Define T (x) = [∇uF (u,v),−∇vF (u,v)]> with x = (u,v) in problem (1). At every point x ∈ X , we don’t have access to T (x) and have only access to a noisy observations of T (x). That is, T (x; ξ), where ξ is a random variable with distribution D. For the ease of presentation, we use the terms stochastic gradient and stochastic first-order oracle interchangeably to stand for T (x; ξ) in the min-max setting.
Definition 1 (Monotonicity). An operator T is monotone if 〈T (x)−T (y),x−y〉 ≥ 0 for ∀x,y ∈ X . An operator T is pseudo-monotone if 〈T (x),y − x〉 ≥ 0⇒ 〈T (y),y − x〉 ≥ 0 for ∀x,y ∈ X . An operator T is γ-strongly-monotone if 〈T (x)− T (y),x− y〉 ≥ γ2 ‖x− y‖ 2 for ∀x,y ∈ X .
We give here formal definitions of monotonic operators T and the -first-order stationary point.
Definition 2 ( -First-Order Stationary Point). A point x ∈ X is called -first-order stationary point if ‖T (x)‖ ≤ .
Remark: We make the following observations:
(a). From the definition, it is evident that strong-monotonicity ⇒ monotonicity ⇒pseudomonotonicity. Assuming SVI has a solution and pseudo-monotonicity of the operator T imply that MVI(T,X ) has a solution. To see that, assume that SVI has a nonempty solution set, i.e. there exists x∗ such that 〈T (x∗),y − x∗〉 ≥ 0 for any y. Noting that pseudomonotonicity means that for every y,x, 〈T (x),y − x〉 ≥ 0 implies 〈T (y),y − x〉 ≥ 0, we have 〈T (y),y − x∗〉 ≥ 0 for any y, which means that x∗ is the solution of Minty variational inequality. Note that the reverse may not be true and an example is provided in Appendix G.
(b). For the min-max problem (1), when F (u,v) is convex in u and concave in v, T is monotone. And, therefore solving SVI(T,X ) is equivalent to solving (1). When T is not monotone, by assuming T is Lipschitz continuous, it can be shown that the solution set of (1) is a subset of the solution set of SVI(T,X ). However, even solving SVI(T,X ) is NP-hard in general and hence we resort to finding an -first-order stationary point.
Throughout the paper, we make the following assumption:
Assumption 1. (i). T is L-Lipschitz continuous, i.e. ‖T (x1) − T (x2)‖2 ≤ L‖x1 − x2‖2 for ∀x1,x2 ∈ X .
(ii). MVI(T,X ) has a solution, i.e. there exists x∗ such that 〈T (x),x− x∗〉 ≥ 0 for ∀x ∈ X .
(iii). For ∀x ∈ X , E [T (x; ξ)] = T (x), E ‖T (x; ξ)− T (x)‖2 ≤ σ2.
Remark: Assumptions (i) and (iii) are commonly used assumptions in the literature of variational inequalities and non-convex optimization (Juditsky et al., 2011; Ghadimi & Lan, 2013; Iusem et al., 2017). Assumption (ii) is used frequently in previous work focusing on analyzing algorithms that solve non-monotone variational inequalities (Iusem et al., 2017; Lin et al., 2018; Mertikopoulos et al., 2018). Assumption (ii) is weaker than other assumptions usually considered, such as pseudomonotonicity, monotonicity, or coherence as assumed in (Mertikopoulos et al., 2018). For nonconvex minimization problem, it has been shown that this assumption holds while using SGD to learn neural networks (Li & Yuan, 2017; Kleinberg et al., 2018; Zhou et al., 2019).

heading: 3 OPTIMISTIC STOCHASTIC GRADIENT
text: This section serves as a warm-up and motivation of our main theoretical contribution presented in the next section. Inspired by (Iusem et al., 2017), we present an algorithm called Optimistic Stochastic Gradient (OSG) that saves the cost of the additional oracle call as required in (Iusem et al., 2017) and maintains the same iteration complexity. The main algorithm is described in Algorithm 1, where mt denotes the minibatch size for estimating the first-order oracle. It is worth mentioning that Algorithm 1 becomes stochastic extragradient method if one changes T (zk−1; ξik−1) to T (xk−1; ξ i k−1) in line 3. Stochastic extragradient method requires to compute stochastic gradient over both sequences {xk} and {zk}. In contrast, {xk} is an ancillary sequence in OSG and the stochastic gradient is only computed over the sequence of {zk}. Thus, stochastic extragradient method is twice as expensive as OSG in each iteration. In some tasks (e.g. training GANs) where the stochastic gradient computation is expensive, OSG is numerically more appealing.
Algorithm 1 Optimistic Stochastic Gradient (OSG) 1: Input: z0 = x0 = 0 2: for k = 1, . . . , N do 3: zk = ΠX [ xk−1 − η · 1mk−1 ∑mk−1 i=1 T (zk−1; ξ i k−1)
] 4: xk = ΠX [ xk−1 − η · 1mk ∑mk i=1 T (zk; ξ i k) ] 5: end for
Remark: When X = Rd, the update in Algorithm 1 becomes the algorithm in (Daskalakis et al., 2017), i.e.
zk+1 = zk − 2η · 1
mk−1 mk∑ i=1 T (zk; ξ i k) + η · 1 mk−1 mk−1∑ i=1 T (zk−1; ξ i k−1) (2)
The detailed derivation of (2) can be found in Appendix F.
Theorem 1. Suppose that Assumption 1 holds. Let rα(zk) = ‖zk −ΠX (zk − αT (zk))‖. Let η ≤ 1/9L and run Algorithm 1 for N iterations. Then we have
1
N N∑ k=1 E [ r2η(zk) ] ≤ 8‖x0 − x∗‖ 2 N + 100η2 N N∑ k=0 σ2 mk ,
Corollary 1. Consider the unconstrained case where X = Rd. Let η ≤ 1/9L, and we have
1
N N∑ k=1 E‖T (zk)‖22 ≤ 8‖x0 − x∗‖2 η2N + 100 N N∑ k=0 σ2 mk , (3)
Remark: There are two implications of Corollary 1.
• (Increasing Minibatch Size) Let η = 19L , mk = k + 1. To guarantee 1 N ∑N k=1 E‖T (zk)‖22 ≤ 2, the total number of iterations is N = Õ( −2), and the total
complexity is ∑N k=1mk = Õ( −4), where Õ(·) hides a logarithmic factor of .
• (Constant Minibatch Size) Let η = 19L , mk = 1/ 2. To guarantee 1N ∑N k=1 E‖T (zk)‖22 ≤
2, the total number of iterations is N = O( −2), and the total complexity is ∑N k=0mk =
O( −4).

heading: 4 OPTIMISTIC ADAGRAD

heading: 4.1 ADAGRAD FOR MINIMIZATION PROBLEMS
text: Before introducing Optimistic Adagrad, we present here a quick overview of Adagrad (Duchi et al., 2011). The main objective in Adagrad is to solve the following minimization problem:
min w∈Rd
F (w) = Eζ∼Pf(w; ζ) (4)
where w is the model parameter, and ζ is an random variable following distribution P . The update rule of Adagrad is
wt+1 = wt − ηH−1t ĝt, (5)
where η > 0, ĝt = ∇f(wt; ζt), Ht = diag ((∑t i=1 ĝi ◦ ĝi ) 1 2 ) with ◦ denoting the Hadamard
product. Adagrad when taking Ht = I reduces to SGD. Different from SGD, Adagrad dynamically incorporates knowledge of history gradients to perform more informative gradient-based learning. When solving a convex minimization problem and the gradient is sparse, Adagrad converges faster than SGD. There are several variants of Adagrad, including Adam (Kingma & Ba, 2014), RMSProp (Tieleman & Hinton, 2012), and AmsGrad (Reddi et al., 2019). All of them share the spirit, as they take advantage of the information provided by the history of gradients. Wilson et al. (2017) provide a complete overview of different adaptive gradient methods in a unified framework. It is worth mentioning that Adagrad can not be directly applied to solve non-convex non-concave minmax problems with provable guarantee.

heading: 4.2 OPTIMISTIC ADAGRAD FOR MIN-MAX OPTIMIZATION
text: Our second algorithm named Optimistic Adagrad (OAdagrad) is an adaptive variant of OSG, which also updates minimization variable and maximization variable simultaneously. The key difference between OSG and OAdagrad is that OAdagrad inherits ideas from Adagrad to construct variable metric based on history gradients information, while OSG only utilizes a fixed metric. This difference helps us establish faster adaptive convergence under some mild assumptions. Note that in OAdagrad we only consider the unconstrained case, i.e. X = Rd. Assumption 2. (i). There exists G > 0 and δ > 0 such that ‖T (z; ξ)‖2 ≤ G, ‖T (z; ξ)‖∞ ≤ δ
for all z almost surely.
(ii). There exists a universal constant D > 0 such that ‖xk‖2 ≤ D/2 for k = 1, . . . , N , and ‖x∗‖2 ≤ D/2.
Remark: Assumption 2 (i) is a standard one often made in literature (Duchi et al., 2011). Assumption 2 (ii) holds when we use normalization layers in the discriminator and generator such as spectral normalization of weights (Miyato et al., 2018; Zhang et al., 2018), that will keep the norms of the weights bounded. Regularization techniques such as weight decay also ensure that the weights of the networks remain bounded throughout the training.
Define ĝk = 1m ∑m i=1 T (zk; ξ i k), ‖x‖H = √ 〈x, Hx〉. Denote ĝ0:k by the concatenation of ĝ0, . . . , ĝk, and denote ĝ0:k,i by the i-th row of ĝ0:k.
Algorithm 2 Optimistic AdaGrad (OAdagrad) 1: Input: z0 = x0 = 0, H0 = δI 2: for k = 1, . . . , N do 3: zk = xk−1 − ηH−1k−1ĝk−1 4: xk = xk−1 − ηH−1k−1ĝk 5: Update ĝ0:k = [ĝ0:k−1 ĝk], sk,i = ‖ĝ0:k,i‖, i = 1, . . . , d and set Hk = δI + diag(sk−1) 6: end for
Theorem 2. Suppose Assumption 1 and 2 hold. Suppose ‖ĝ1:k,i‖2 ≤ δkα with 0 ≤ α ≤ 1/2 for every i = 1, . . . , d and every k = 1, . . . , N . When η ≤ δ9L , after running Algorithm 2 for N iterations, we have
1
N N∑ k=1 E‖T (zk)‖2H−1k−1 ≤ 8D2δ2(1 + d(N − 1)α) η2N +
100 ( σ2/m+ d ( 2δ2Nα +G2 )) N . (6)
To make sure 1N ∑N k=1 E‖T (zk)‖2H−1k−1 ≤ 2, the number of iterations is N = O ( − 2 1−α ) .
Remark:
• Note that the convergence measure used in Theorem 2 is different from that in Corollary 1. However we show that under the measure used in Theorem 2, OSG (Algorithm 1) still has complexity O(1/ 4). By the construction of Hk in Algorithm 2, we know that ‖T (z)‖H−1k ≤ ‖T (z)‖H−10 for any k ≥ 0 and any z, and hence 1 N ∑N k=1 E‖T (zk)‖2H−1k−1
≤ 1 N ∑N k=1 E‖T (zk)‖2H−10 = 1 δ · 1 N ∑N k=1 E‖T (zk)‖22. By Corollary 1, we know that OSG
still requires O(1/ 4) complexity to guarantee that 1N ∑N k=1 E‖T (zk)‖2H−1k−1 ≤ 2.
• We denote ĝ1:k by the cumulative stochastic gradient, where ‖ĝ1:k,i‖2 ≤ δkα characterizes the growth rate of the gradient in terms of i-th coordinate. In our proof, a key quantity is∑d i=1 ‖ĝ1:k,i‖2 that crucially affects the computational complexity of Algorithm 2. Since∑d i=1 ‖ĝ1:k,i‖2 ≤ δdkα, in the worst case, α = 1 2 . But in practice, the stochastic gradient
is usually sparse, and hence α can be strictly smaller than 12 . • As shown in Theorem 2, the minibatch size used in Algorithm 2 for estimating the first-
order oracle can be any positive constant and independent of . This is more practical than the results established in Theorem 1, since the minibatch size in Theorem 1 does either increase in terms of number of iterations or is dependent on . When α = 12 , the complexity of Algorithm 2 isO(1/ 4), which matches the complexity stated in Theorem 1. When α < 12 , the complexity of OAdagrad given in Algorithm 2 isO ( − 2 1−α ) , i.e., strictly better than that of OSG given in Algorithm 1.
Comparison with Alternating Adam and Optimistic Adam Alternating Adam is very popular in GAN training (Goodfellow et al., 2014; Arjovsky et al., 2017; Gulrajani et al., 2017; Brock et al., 2018). In Alternating Adam, one alternates between multiple steps of Adam on the discriminator and a single step of Adam on the generator. The key difference between OAdagrad and Alternating Adam is that OAdagrad updates the discriminator and generator simultaneously. It is worth mentioning that OAdagrad naturally fits into the framework of Optimistic Adam proposed in (Daskalakis et al., 2017). Taking β1 = 0, β2 → 1 in their Algorithm 1 reduces to OAdagrad with annealing learning rate. To the best of our knowledge, there is no convergence proof for Alternating Adam for non-convex non-concave problems. Our convergence proof for OAdagrad provides a theoretical justification of a special case of Optimistic Adam.

heading: 5 EXPERIMENTS
text: WGAN-GP on CIFAR10 In the first experiment, we verify the effectiveness of the proposed algorithms in GAN training using the PyTorch framework (Paszke et al., 2017). We use Wasserstein
GAN with gradient penalty (WGAN-GP) (Gulrajani et al., 2017) and CIFAR10 data in our experiments. The architectures of discriminator and generator, and the penalty parameter in WGAN-GP are set to be same as in the original paper. We compare Alternating Adam, OSG and OAdagrad, where the Alternating Adam is to run 5 steps of Adam on the discriminator before performing 1 step of Adam on the generator. We try different batch sizes (64, 128, 256) for each algorithm. For each algorithm, we tune the learning rate in the range of {1×10−3, 2×10−4, 1×10−4, 2×10−5, 1×10−5} when using batch size 64, and use the same learning rate for batch size 128 and 256. We report Inception Score (IS) (Salimans et al., 2016) as a function of number of iterations. Figure 1 suggests that OAdagrad performs better than OSG and Alternating Adam, and OAdagrad results in higher IS. We compare the generated CIFAR10 images associated with these three methods, which is included in Appendix A. We also provide experimental results to compare the performance of different algorithms using different minibatch sizes, which are included in Appendix E.
Growth Rate of Cumulative Stochastic Gradient In the second experiment, we employ OAdagrad to train GANs and study the growth rate of the cumulative stochastic gradient (i.e.,∑d i=1 ‖ĝ1:N,i‖2). We tune the learning rate from {1×10−3, 2×10−4, 1×10−4, 2×10−5, 1×10−5} and choose batch size to be 64. In Figure 2, the blue curve and red curve stand for the growth rate for OAdagrad and its corresponding tightest polynomial growth upper bound respectively. N is the number of iterations, and c is a multiplicative constant such that the red curve and blue curve overlaps at the starting point of the training. The degree of the polynomial is determined using binary search. We can see that the growth rate of cumulative stochastic gradient grows very slowly in GANs (the worst-case polynomial degree is 0.5, but it is 0.2 for WGAN-GP on CIFAR10 and 0.07 for WGAN on LSUN Bedroom dataset). As predicted by our theory, this behavior explains the faster convergence of OAdagrad versus OSG, consistent with what is observed empirically in Figure 1.
Self-Attention GAN on ImageNet In the third experiment, we consider GAN training on largescale dataset. We use the model from Self-Attention GAN (Zhang et al., 2018) (SA-GAN) and ImageNet as our dataset. Note that in this setting the boundedness of both generator (G) and discriminator (D) is ensured by spectral normalization of bothG andD. Three separate experiments are performed, including Alternating Adam (baseline), Simultaneous Adam (Mescheder et al., 2017), and OAdagrad. It should be mentioned that the update rule of Simultaneous Adam involves performing Adam-type update for discriminator and generator simultaneously. Training is performed with batch size 128 for all experiments.
For the baseline experiment (Alternating Adam) we use the default settings and hyper parameters reported in SA-GAN (Zhang et al., 2018) (note that we are not using the same batch size of 256 as in (Zhang et al., 2018) due to limited computational resources). In our experience, Alternating Adam training for a batch size of 128 with same learning rate as in SA-GAN (0.0001 for generator and 0.0004 for discriminator) collapsed. This does not mean that Alternating Adam fails, it just needs more tuning to find the correct range of learning rates for the particular batch size we have. With the hyperparameters ranges we tried Alternating Adam collapsed, with extra tuning efforts and an expensive computational budget Alternating Adam would eventually succeed. This is inline with the large scale study in (Lucic et al., 2018) that states that given a large computational budget for tuning hyper-parameters most GANs training succeed equally.
For both OAdagrad and Simultaneous Adam, we use different learning rate for generator and discriminator, as suggested in (Heusel et al., 2017). Specifically, the learning rates used are 10−3 for the generator and 4 × 10−5 for the discriminator. We report both Inception Score (IS) and Fréchet Inception Distance (Heusel et al., 2017) (FID) as a function of number of iterations.
We compare the generated ImageNet images associated with the three optimization methods in Appendix A. Since Alternating Adam collapsed we don’t report its Inception Score or FID. As it can be seen in Figure 3 and Appendix A, OAdagrad outperforms simultaneous Adam in quantitative metrics (IS and FID) and in sample quality generation. Future work will include investigating whether OAdagrad would benefit from training with larger batch size, in order to achieve state-of-the-art results.

heading: 6 CONCLUSION
text: In this paper, we explain the effectiveness of adaptive gradient methods in training GANs from both theoretical and empirical perspectives. Theoretically, we provide two efficient stochastic algorithms for solving a class of min-max non-convex non-concave problems with state-of-the-art computational complexities. We also establish adaptive complexity results for an Adagrad-style algorithm by using coordinate-wise stepsize according to the geometry of the history data. The algorithm is proven to enjoy faster adaptive convergence than its non-adaptive counterpart when the gradient is
sparse, which is similar to Adagrad applied to convex minimization problem. We have conducted extensive empirical studies to verify our theoretical findings. In addition, our experimental results suggest that the reason why adaptive gradient methods deliver good practical performance for GAN training is due to the slow growth rate of the cumulative stochastic gradient.

heading: ACKNOWLEDGMENTS
text: The authors thank the anonymous reviewers for their helpful comments. M. Liu and T. Yang are partially supported by National Science Foundation CAREER Award 1844403. M. Liu would like to thank Xiufan Yu from Pennsylvania State University and Zehao Dou from Yale University for helpful discussions.

heading: A MORE EXPERIMENTAL RESULTS
text: Comparison of Generated CIFAR10 Images by Different Optimization Methods In this section, we report the generated CIFAR10 images during the training of WGAN-GP by three optimization methods (OSG, OAdagrad, Alternating Adam). Every method uses batch size 64, and 1 iteration represents calculating the stochastic gradient with minibatch size 64 once. Figure 4 consists of images by three optimization methods at iteration 8000. Visually we can see that OAdagrad is better than Alternating Adam, and both of them are significantly better than OSG. It is consistent with the inception score results reported in Figure 1, and it also illustrates the tremendous benefits delivered by adaptive gradient methods when training GANs.
Comparison of Generated ImageNet Images by Different Optimization Methods In this section, we report the generated ImageNet images during the training of Self-Attention GAN by three optimization methods (OAdagrad, Simultaneous Adam, Alternating Adam). Every method uses batch size 128 and 1 iteration represents calculating the stochastic gradient with minibatch 128 once. Figure 5 consists of images by three optimization methods at iteration 135000. Visually it is apparent that OAdagrad is better than Simultaneous Adam, and both of them are significantly than Alternating Adam.
Unofficial PyTorch Inception Score and FID results for SA-GAN on ImageNet

heading: B RELATED WORK
text: Min-max Optimization and GAN Training For convex-concave min-max optimization, the extragradient method was first proposed by (Korpelevich, 1976). Later on, under gradient Lipschitz condition, Nemirovski (2004) extended the idea of extragradient to mirror-prox and obtained the O(1/N) convergence rate in terms of the duality gap (see also (Nesterov, 2007)), where N is the number of iterations. When only the stochastic first-order oracle is available, the stochastic mirrorprox was analyzed by (Juditsky et al., 2011). The convergence rates for both deterministic and stochastic mirror-prox are optimal (Nemirovsky & Yudin, 1983). Recently, Zhao (2019) developed a nearly-optimal stochastic first-order algorithm when the primal variable is strongly convex in the primal variable. Bach & Levy (2019) proposed a universal algorithm that is adaptive to smoothness and noise, and simultaneously achieves optimal convergence rate.
There is a plethora of work analyzing one-sided nonconvex min-max problem, where the objective function is nonconvex in the minimization variable but concave in maximization variable. When the function is weakly-convex in terms of the minimization variable, Rafique et al. (2018) propose a stage-wise stochastic algorithm that approximately solves a convex-concave subproblem by adding a quadratic regularizer and show the first-order convergence of the equivalent minimization problem. Under the same setting, Lu et al. (2019) utilize block-based optimization strategy and show the convergence of the stationarity gap. By further assuming that the function is smooth in the minimization variable, Lin et al. (2019) show that (stochastic) gradient descent ascent is able to converge to the first-order stationary point of the equivalent minimization problem. Liu et al. (2020) cast the problem of stochastic AUC maximization with deep neural networks into a nonconvex-concave min-max problem, show the PL (Polyak-Łojasiewicz) condition holds for the objective of the outer minimization problem, and propose an algorithm and establish its fast convergence rate.
A more challenging problem is the non-convex non-concave min-max problem. Dang & Lan (2015) demonstrate that the deterministic extragradient method is able to converge to -first-order stationary point with non-asymptotic guarantee. Under the condition that the objective function is weaklyconvex and weakly-concave, Lin et al. (2018) designs a stage-wise algorithm, where in each stage a strongly-convex strongly-concave subproblem is constructed by adding quadratic terms and appropriate stochastic algorithms can be employed to approximately solve it. They also show the convergence to the stationary point. Sanjabi et al. (2018) design an alternating deterministic optimization algorithm, in which multiple steps of gradient ascent for dual variable are conducted before one step of gradient descent for primal variable is performed. They show the convergence to stationary point based on the assumption that the inner maximization problem satisfies PL condition (Polyak, 1969). Our work is different from these previous methods in many aspects. In comparison to (Lin et al., 2018), our result does not need the bounded domain assumption. Furthermore, our iteration complexity is O(1/ 4) to achieve -first-order stationary point while the corresponding complexity
in (Lin et al., 2018) isO(1/ 6). When comparing to (Sanjabi et al., 2018), we do not assume that the PL (Polyak-Łojasiewicz) condition holds. Additionally, our algorithm is stochastic and not restricted to the deterministic case. Apparently the most related work to the present one is (Iusem et al., 2017). The stochastic extragradient method analyzed in (Iusem et al., 2017) requires calculation of two stochastic gradients per iteration, while the present algorithm only needs one since it memorizes the stochastic gradient in the previous iteration to guide the update in the current iteration. Nevertheless, we achieve the same iteration complexity as in (Iusem et al., 2017).
There are a body of work analyzing the convergence behavior of min-max optimization algorithms and its application in training GANs (Heusel et al., 2017; Daskalakis & Panageas, 2018; Nagarajan & Kolter, 2017; Grnarova et al., 2017; Yadav et al., 2017; Gidel et al., 2018; Mertikopoulos et al., 2018; Mazumdar et al., 2019). A few of them (Heusel et al., 2017; Daskalakis & Panageas, 2018; Mazumdar et al., 2019) only have asymptotic convergence. Others (Nagarajan & Kolter, 2017; Grnarova et al., 2017; Daskalakis et al., 2017; Yadav et al., 2017; Gidel et al., 2018; Mertikopoulos et al., 2018) focus on more restricted settings. For example, Nagarajan & Kolter (2017); Grnarova et al. (2017) require the concavity of the objective function in terms of dual variable. Yadav et al. (2017); Gidel et al. (2018) assume the objective to be convex-concave. Mertikopoulos et al. (2018) imposes the so-called coherence condition which is stronger than our assumption. Daskalakis et al. (2017) analyze the last-iteration convergence for bilinear problem. Recently, Gidel et al. (2019) analyze the benefits of using negative momentum in alternating gradient descent to improve the training of a bilinear game. Chavdarova et al. (2019) develop a variance-reduced extragradient method and shows its linear convergence under strong monotonicity and finite-sum structure assumptions. Azizian et al. (2019) provide a unified analysis of extragradient for bilinear game, strongly monotone case, and their intermediate cases. However, none of them give non-asymptotic convergence results for the class of non-convex non-concave min-max problem considered in our paper.

heading: C PROOF OF THEOREM 1
text: C.1 FACTS
Suppose X ⊂ Rd is closed and convex set, then we have
Fact 1. For all x ∈ Rd and y ∈ X , ‖ΠX (x)− y‖2 + ‖ΠX (x)− x‖2 ≤ ‖x− y‖2.
Fact 2. For all x ∈ Rd and y ∈ X , 〈x−ΠX (x),y −ΠX (x)〉 ≤ 0.
C.2 LEMMAS
Lemma 1. For η ≤ 19L , we have
1
2 N∑ k=1 ‖xk−1−zk‖2+ 1 2 N∑ k=1 ‖xk−zk‖2 ≤ ‖x0−x∗‖2−‖xN−x∗‖2+12η2 N∑ k=0 ‖ k‖2+ N∑ k=1 Λk (7)
Proof. Let x∗ ∈ X ∗, where X ∗ is the set of optimal solutions of MVI(T,X ), i.e. 〈T (x),x− x∗〉 ≥ 0 holds for ∀x ∈ X . Define k = 1mk ∑mk i=1 T (zk, ξ i k) − T (zk), and T̂ ( k, zk) = T (zk) + k. For
any x ∈ X , we have ‖xk − x‖2 = ‖ΠX ( xk−1 − ηT̂ ( k, zk) ) − x‖2
(a) ≤ ∥∥∥xk−1 − ηT̂ ( k, zk)− x∥∥∥2 − ∥∥∥xk−1 − ηT̂ ( k, zk)−ΠX (xk−1 − ηT̂ ( k, zk))∥∥∥2
= ∥∥∥xk−1 − ηT̂ ( k, zk)− x∥∥∥2 − ∥∥∥xk−1 − ηT̂ ( k, zk)− xk∥∥∥2
= ‖xk−1 − x‖2 − ‖xk−1 − xk‖2 + 2 〈 x− xk, ηT̂ ( k, zk) 〉 = ‖xk−1 − x‖2 − ‖xk−1 − xk‖2 + 2 〈 x− zk, ηT̂ ( k, zk) 〉 + 2 〈 zk − xk, ηT̂ ( k, zk)
〉 = ‖xk−1 − x‖2 − ‖xk−1 − zk + zk − xk‖2 + 2 〈 x− zk, ηT̂ ( k, zk) 〉 + 2 〈 zk − xk, ηT̂ ( k, zk)
〉 = ‖xk−1 − x‖2 − ‖xk−1 − zk‖2 − ‖zk − xk‖2 − 2 〈xk−1 − zk, zk − xk〉+
2 〈 x− zk, ηT̂ ( k, zk) 〉 + 2 〈 zk − xk, ηT̂ ( k, zk) 〉 = ‖xk−1 − x‖2 − ‖xk−1 − zk‖2 − ‖zk − xk‖2 + 2 〈 x− zk, ηT̂ ( k, zk) 〉 + 2 〈 xk − zk,xk−1 − ηT̂ ( k, zk)− zk
〉 (8)
where (a) holds by using Fact 1. Note that 2 〈 x∗ − zk, ηT̂ ( k, zk) 〉 = 2 〈x∗ − zk, η (T (zk) + k)〉 ≤ 2 〈x∗ − zk, η k〉 , (9)
where the last inequality holds by the fact that 〈x∗ − zk, T (zk)〉 ≤ 0 since x∗ is a solution of MVI(T,X ). Note that
2 〈 xk − zk,xk−1 − ηT̂ ( k, zk)− zk 〉 = 2 〈 xk − zk,xk−1 − ηT̂ ( k−1, zk−1)− zk 〉 + 2 〈 xk − zk, η ( T̂ ( k−1, zk−1)− T̂ ( k, zk)
)〉 (a)
≤ 2η ‖xk − zk‖ · ∥∥∥T̂ ( k−1, zk−1)− T̂ ( k, zk)∥∥∥
(b) ≤ 2η ∥∥∥ΠX (xk−1 − η · T̂ ( k, zk))−ΠX (xk−1 − η · T̂ ( k−1, zk−1))∥∥∥ · ∥∥∥T̂ ( k−1, zk−1)− T̂ ( k, zk)∥∥∥
(c) ≤ 2η2 ∥∥∥T̂ ( k−1, zk−1)− T̂ ( k, zk)∥∥∥2 = 2η2 ‖T (zk−1) + k−1 − (T (zk) + k)‖2
≤ 2η2 (‖T (zk−1)− T (zk)‖+ ‖ k−1‖+ ‖ k‖)2 (d) ≤ 2η2 (L‖zk−1 − zk‖+ ‖ k−1‖+ ‖ k‖)2
(e) ≤ 6η2 ( L2‖zk−1 − zk‖2 + ‖ k−1‖2 + ‖ k‖2 ) (10)
where (a) holds by 〈 xk − zk,xk−1 − ηT̂ ( k−1, zk−1)− zk 〉 ≤ 0 and Cauchy-Schwartz inequality,
where the former inequality comes from Fact 2 and the update rules of the algorithm, (b) holds by the update rule of zk and xk, (c) holds by the nonexpansion property of the projection operator, (d) holds since T is L-Lipschitz continuous, (e) holds since (a+ b+ c)2 ≤ 3a2 + 3b2 + 3c2. Define Λk = 2〈x∗ − zk, η k〉. Taking x = x∗ in (8) and combining (9) and (10), we have ‖xk − x∗‖2
≤ ‖xk−1 − x∗‖2 − ‖xk−1 − zk‖2 − ‖zk − xk‖2 + 6η2L2‖zk−1 − zk‖2 + 6η2‖ k−1‖2 + 6η2‖ k‖2 + Λk (11)
Noting that ‖zk−1 − zk‖2 = ‖zk−1 − xk−1 + xk−1 − zk‖2 ≤ 3‖zk−1 − xk−1‖2 + 3‖xk−1 − zk‖2 + 3‖xk − zk‖2, we rearrange terms in (11), which yields ‖xk−1 − zk‖2 + ‖zk − xk‖2 − 6η2L2 ( 3‖zk−1 − xk−1‖2 + 3‖xk−1 − zk‖2 + 3‖zk − xk‖2
) ≤ ‖xk−1 − x∗‖2 − ‖xk − x∗‖2 + 6η2‖ k−1‖2 + 6η2‖ k‖2 + Λk
(12)
Take summation over k = 1, . . . , N in (12) and note that x0 = z0, which yields
( 1− 18η2L2 ) N∑ k=1 ‖xk−1 − zk‖2 + ( 1− 36η2L2 ) N∑ k=1 ‖xk − zk‖2
≤ ‖x0 − x∗‖2 − ‖xN − x∗‖2 + 12η2 N∑ k=0 ‖ k‖2 + N∑ k=1 Λk
(13)
By taking η ≤ 19L , we have 1− 36η 2L2 ≥ 12 , and we have the result.
C.3 MAIN PROOF OF THEOREM 1
Proof. Define rη(zk) = ‖zk −ΠX (zk − ηT (zk))‖. Our goal is to get a bound on rη(zk). We have:
r2η(zk) = ‖zk −ΠX (zk − ηT (zk))‖ 2 = ‖zk − xk + xk −ΠX (zk − ηT (zk))‖2
(a) ≤ 2 ‖zk − xk‖2 + 2 ‖xk −ΠX (zk − ηT (zk))‖2
= 2 ‖zk − xk‖2 + 2 ∥∥∥ΠX (xk−1 − ηT̂ ( k, zk))−ΠX (zk − ηT (zk))∥∥∥2
(b) ≤ 2 ‖zk − xk‖2 + 4 ‖xk−1 − zk‖2 + 4η2 ∥∥∥T (zk)− T̂ ( k, zk)∥∥∥2
≤ 4 ‖zk − xk‖2 + 4 ‖xk−1 − zk‖2 + 4η2‖ k‖2
(14)
where (a) holds since (a + b)2 ≤ 2a2 + 2b2, (b) holds by the non-expansion property of the projection operator and (a+ b)2 ≤ 2a2 + 2b2.
Let x∗ ∈ X ∗, where X ∗ is the set of optimal solutions of MVI(T,X ), i.e. 〈T (x),x− x∗〉 ≥ 0 holds for ∀x ∈ X . Define k = 1mk ∑mk i=1 T (zk, ξ i k)− T (zk), and T̂ ( k, zk) = T (zk) + k. Define Λk = 2〈x∗ − zk, η k〉.
By summing over k in Equation (14) and using Equation (7) in Lemma 1, we have
N∑ k=1 r2η(zk) ≤ 4 N∑ k=1 ‖zk − xk‖2 + 4 N∑ k=1 ‖xk−1 − zk‖2 + 4η2 N∑ k=1 ‖ k‖2
= 8
( 1
2 N∑ k=1 ‖zk − xk‖2 + 1 2 N∑ k=1
‖xk−1 − zk‖2 )
+ 4η2 N∑ k=0 ‖ k‖2
By (7) ≤ 8 ( ‖x0 − x∗‖2 + 12η2
N∑ k=0 ‖ k‖2 + N∑ k=1 Λk
) + 4η2
N∑ k=0 ‖ k‖2
(15)
Taking expectation and divided by N on both sides, we have
1
N N∑ k=1 E [ r2η(zk) ] ≤ 8 N
( ‖x0 − x∗‖2 + 12η2
N∑ k=0 E‖ k‖2 + N∑ k=1 E(Λk)
) + 4η2
N N∑ k=1 E‖ k‖2
≤ 8 N
( ‖x0 − x∗‖2 + 12η2
N∑ k=0 σ2 mk
) + 4η2
N N∑ k=0 σ2 mk
= 8‖x0 − x∗‖2
N +
100η2
N N∑ k=0 σ2 mk .
(16)

heading: D PROOF OF THEOREM 2
text: In this section, we define gk = T (zk), k = ĝk − gk.
D.1 LEMMAS
Lemma 2. For any positive definite diagonal matrix H satisfying H δI with δ > 0, if ‖T (x1)− T (x2)‖2 ≤ L‖x1 − x2‖2 for x1,x2 ∈ X , then
‖T (x1)− T (x2)‖H−1 ≤ L
δ ‖x1 − x2‖H .
Proof. Note that H δI , we have 0 < H−1 1δ I . Noting that ‖x‖H = √ x>Hx, we have
‖T (x1)− T (x2)‖H−1 ≤ 1√ δ ‖T (x1)− T (x2)‖2 ≤ L√ δ ‖x1 − x2‖2 ≤ L δ ‖x1 − x2‖H .
Lemma 3. When η ≤ δ9L , we have
1
2 N∑ k=1 ‖xk−1 − zk‖2Hk−1 + 1 2 N∑ k=1 ‖xk − zk‖2Hk−1
≤ N∑ k=1 ( ‖xk−1 − x∗‖2Hk−1 − ‖xk − x∗‖ 2 Hk−1 ) + 12η2 ( ‖ 0‖2H−10 + N∑ k=1 ‖ k‖2H−1k−1 ) + N∑ k=1 Λk
(17)
Proof. Define k = ĝk − gk. For any x ∈ X , we have ‖xk − x‖2Hk−1 = ∥∥xk−1 − ηH−1k−1ĝk − x∥∥2Hk−1 = ∥∥xk−1 − ηH−1k−1ĝk − x∥∥2Hk−1 − ∥∥xk−1 − ηH−1k−1ĝk − xk∥∥2Hk−1 = ‖xk−1 − x‖2Hk−1 − ‖xk−1 − xk‖ 2 Hk−1
+ 2 〈x− xk, ηĝk〉 = ‖xk−1 − x‖2Hk−1 − ‖xk−1 − xk‖ 2 Hk−1
+ 2 〈x− zk, ηĝk〉+ 2 〈zk − xk, ηĝk〉 = ‖xk−1 − x‖2Hk−1 − ‖xk−1 − zk + zk − xk‖ 2 Hk−1
+ 2 〈x− zk, ηĝk〉+ 2 〈zk − xk, ηĝk〉 = ‖xk−1 − x‖2Hk−1 − ‖xk−1 − zk‖ 2 Hk−1
− ‖zk − xk‖2Hk−1 − 2 〈Hk−1(xk−1 − zk), zk − xk〉+ 2 〈x− zk, ηĝk〉+ 2 〈zk − xk, ηĝk〉
= ‖xk−1 − x‖2Hk−1 − ‖xk−1 − zk‖ 2 Hk−1 − ‖zk − xk‖2Hk−1 + 2 〈x− zk, ηĝk〉 + 2 〈xk − zk, Hk−1(xk−1 − zk)− ηĝk〉
(18) Note that 2 〈x∗ − zk, ηĝk〉 = 2 〈x∗ − zk, η (gk + k)〉 ≤ 2 〈x∗ − zk, η k〉 , (19) where the last inequality holds by the fact that 〈x∗ − zk,gk〉 ≤ 0 since x∗ is a solution of MVI(T,X ). Note that
2 〈xk − zk, Hk−1(xk−1 − zk)− ηĝk〉 = 2 〈 xk − zk, Hk−1(xk−1 − zk − ηH−1k−1ĝk−1) 〉 + 2 〈xk − zk, η (ĝk−1 − ĝk)〉
(a) ≤ 2 〈( xk−1 − ηH−1k−1ĝk ) − ( xk−1 − ηH−1k−1ĝk−1 ) , η (ĝk−1 − ĝk) 〉 = 2η2 ‖ĝk−1 − ĝk‖2H−1k−1 = 2η
2 ‖gk−1 − gk + k−1 + k‖2H−1k−1 (b) ≤ 2η2 ( ‖gk−1 − gk‖H−1k−1 + ‖ k−1‖H−1k−1 + ‖ k‖H−1k−1 )2 (c)
≤ 2η2 ( L
δ ‖zk−1 − zk‖Hk−1 + ‖ k−1‖H−1k−1 + ‖ k‖H−1k−1 )2 (d)
≤ 6η2 ( L2
δ2 ‖zk−1 − zk‖2Hk−1 + ‖ k−1‖ 2 H−1k−1 + ‖ k‖2H−1k−1
)
(20)
where (a) holds by the update rule of zk and xk in Algorithm 2, (b) holds by the triangle inequality, (c) holds by utilizing the Lipschitz continuity of T , Lemma 2 and the fact that Hk−1 δI for any k, (d) holds since (a+ b+ c)2 ≤ 3a2 + 3b2 + 3c2. Define Λk = 2〈x∗ − zk, η k〉. Taking x = x∗ in (18) and combining (19) and (20), we have
‖xk − x∗‖2Hk−1 ≤ ‖xk−1 − x∗‖ 2 Hk−1 − ‖xk−1 − zk‖2Hk−1 − ‖zk − xk‖ 2 Hk−1
+ 6η2L2
δ2 ‖zk−1 − zk‖2Hk−1
+ 6η2‖ k−1‖2H−1k−1 + 6η 2‖ k‖2H−1k−1 + Λk
(21)
Noting that
‖zk−1 − zk‖2Hk−1 = ‖zk−1 − xk−1 + xk−1 − zk‖ 2 Hk−1
≤ 3‖zk−1 − xk−1‖2Hk−1 + 3‖xk−1 − zk‖ 2 Hk−1 + 3‖zk − xk‖2Hk−1 ,
we rearrange terms in (21), which yields
‖xk−1 − zk‖2Hk−1 + ‖zk − xk‖ 2 Hk−1
− 6η 2L2
δ2
( 3‖zk−1 − xk−1‖2Hk−1 + 3‖xk−1 − zk‖ 2 Hk−1 + 3‖zk − xk‖2Hk−1 )
≤ ‖xk−1 − x∗‖2Hk−1 − ‖xk − x∗‖ 2 Hk−1 + 6η2‖ k−1‖2H−1k−1 + 6η 2‖ k‖2H−1k−1 + Λk
(22) Taking summation over k = 1, . . . , N in (22), and noting that x0 = z0, ‖x‖2H−1t−1 ≥ ‖x‖2 H−1t for all x and t ≥ 1, we have( 1− 18η 2L2
δ2 ) N∑ k=1 ‖xk−1 − zk‖2Hk−1 + ( 1− 36η 2L2 δ2 ) N∑ k=1 ‖xk − zk‖2Hk−1
≤ N∑ k=1 ( ‖xk−1 − x∗‖2Hk−1 − ‖xk − x∗‖ 2 Hk−1 ) + 12η2 ( ‖ 0‖2H−10 + N∑ k=1 ‖ k‖2H−1k−1 ) + N∑ k=1 Λk
(23) By taking η ≤ δ9L , we have 1− 36η2L2 δ2 ≥ 1 2 , and we have the result.
Lemma 4. When ‖ĝ1:N,i‖2 ≤ δNα with 0 ≤ α ≤ 1/2 for every i, we have
N∑ k=1 ( ‖xk−1 − x∗‖2Hk−1 − ‖xk − x∗‖ 2 Hk−1 ) ≤ D2δ +D2 · dδ(N − 1)α (24)
Proof.
N∑ k=1 ( ‖xk−1 − x∗‖2Hk−1 − ‖xk − x∗‖ 2 Hk−1 ) = ‖x0 − x∗‖2H0 − ‖x1 − x∗‖ 2 H0 + ‖x1 − x∗‖ 2 H1 − ‖x2 − x∗‖ 2 H1 + . . .+ ‖xN−1 − x∗‖ 2 HN−1
− ‖xN − x∗‖2HN−1 ≤ ‖x0 − x∗‖2H0 + ( −‖x1 − x∗‖2H0 + ‖x1 − x∗‖ 2 H1 ) + . . .+ (−‖xN−1 − x∗‖2HN−2 + ‖xN−1 − x∗‖ 2 HN−1 )
≤ ‖x0 − x∗‖2H0 +D 2 (tr(H1 −H0) + tr(H2 −H1) + . . .+ tr(HN−1 −HN−2)) = ‖x0 − x∗‖2H0 +D 2 (tr(HN−1 −H0)) ≤ ‖x0 − x∗‖2H0 +D
2tr(HN−1) ≤ D2δ +D2 · dδ(N − 1)α (25)
Lemma 5. When ‖ĝ1:N,i‖2 ≤ δNα with 0 ≤ α ≤ 1/2 for every i, we have
E [ 96η2‖ 0‖2H−10 + 100η 2 N∑ k=1 ‖ k‖2H−1k−1 ] ≤ 96η 2σ2 mδ + 100η2 ( 2δdNα + G2d δ ) (26)
Proof. Note that
E [ N∑ k=1 ‖ k‖2H−1k−1 ] = E [ N∑ k=1 ‖ĝk − gk‖2H−1k−1 ] = N∑ k=1 E‖ĝk − gk‖2H−1k−1 = N∑ k=1 ( E‖ĝk‖2H−1k−1 − ‖gk‖ 2 H−1k−1 ) ≤
N∑ k=1 E‖ĝk‖2H−1k−1 = N∑ k=1 E‖ĝk‖2H−1k + N∑ k=1 ( E‖ĝk‖2H−1k−1 − E‖ĝk‖ 2 H−1k ) =
N∑ k=1 E‖ĝk‖2H−1k + N∑ k=1 E 〈 ĝk, (H −1 k−1 −H −1 k )ĝk 〉 ≤ N∑ k=1 E‖ĝk‖2H−1k + N∑ k=1 E [ tr(H−1k−1 −H −1 k )G 2 ]
(a) ≤ E [ 2
d∑ i=1 ‖ĝ1:N,i‖2 ] + tr(H−10 )G 2 (b) ≤ 2δdNα + G 2d δ
(27) where (a) holds since we have ∑N k=1 ‖ĝk‖2H−1k ≤ 2 ∑d i=1 ‖ĝ1:N,i‖2 by the setting of Hk and utilizing Lemma 4 of (Duchi et al., 2011), (b) holds because of ‖ĝ1:N,i‖2 ≤ δNα.
In addition, we have E ‖ 0‖2H−10 ≤ σ2 mδ , and hence
E [ 96η2‖ 0‖2H−10 + 100η 2 N∑ k=1 ‖ k‖2H−1k−1 ] ≤ 96η 2σ2 mδ + 100η2 ( 2δdNα + G2d δ ) (28)
D.2 MAIN PROOF OF THEOREM 2
Proof. Our goal is to bound 1N ∑N k=1 E‖T (zk)‖22. Note that
‖ηT (zk)‖2H−1k−1 = ∥∥∥H1/2k−1 (zk − (zk − ηH−1k−1T (zk)))∥∥∥2 = ∥∥∥H1/2k−1 (zk − xk + xk − (zk − ηH−1k−1T (zk)))∥∥∥2
(a) ≤ (∥∥∥H1/2k−1 (zk − xk)∥∥∥+ ∥∥∥H1/2k−1 [xk − (zk − ηH−1k−1T (zk))]∥∥∥)2
(b) ≤ 2 ∥∥∥H1/2k−1 (zk − xk)∥∥∥2 + 2 ∥∥∥H1/2k−1 [xk − (zk − ηH−1k−1T (zk))]∥∥∥2
(c) = 2 ‖zk − xk‖2Hk−1 + 2 ∥∥∥H1/2k−1 [xk−1 − ηH−1k−1ĝk − (zk − ηH−1k−1T (zk))]∥∥∥2 (d) ≤ 2 ‖zk − xk‖2Hk−1 + 4 ‖xk−1 − zk‖ 2 Hk−1
+ 4η2 ‖ĝk − T (zk)‖2H−1k−1 = 2 ‖zk − xk‖2Hk−1 + 4 ‖xk−1 − zk‖ 2 Hk−1
+ 4η2 ‖ k‖2H−1k−1 (29)
where (a) holds by the triangle inequality, (b) is due to (a+ b)2 ≤ 2a2 +2b2, (c) holds by the update rule of xk of Algorithm 2, (d) comes from the triangle inequality and (a+ b)2 ≤ 2a2 + 2b2.
Taking summation over k = 1, . . . , N over (29) and invoking Lemma 3, we have
N∑ k=1 ‖ηT (zk)‖2H−1k−1 ≤ N∑ k=1 ( 2 ‖zk − xk‖2Hk−1 + 4 ‖xk−1 − zk‖ 2 Hk−1 + 4η2 ‖ k‖2H−1k−1 )
≤ 8
( 1
2 N∑ k=1 ‖zk − xk‖2Hk−1 + 1 2 N∑ k=1 ‖xk−1 − zk‖2Hk−1
) + 4η2
N∑ k=1 ‖ k‖2H−1k−1
By (17) ≤ 8 ( N∑ k=1 ( ‖xk−1 − x∗‖2Hk−1 − ‖xk − x∗‖ 2 Hk−1 ) + 12η2 ( ‖ 0‖2H−10 + N∑ k=1 ‖ k‖2H−1k−1 ) + N∑ k=1 Λk )
+ 4η2 N∑ k=1 ‖ k‖2H−1k−1
= 8 N∑ k=1 ( ‖xk−1 − x∗‖2Hk−1 − ‖xk − x∗‖ 2 Hk−1 ) + 96η2‖ 0‖2H−10 + 100η 2 N∑ k=1 ‖ k‖2H−1k−1 + 8 N∑ k=1 Λk
(30) Taking expectation on both sides, and invoking Lemma 4 and Lemma 5, and noting that E [∑N
k=1 Λk
] = 0, we have
N∑ k=1 E ‖ηT (zk)‖2H−1k−1 ≤ 8 ( D2δ +D2 · dδ(N − 1)α ) + 96η2σ2 mδ +100η2 ( 2δdNα + G2d δ ) (31)
Dividing η2N on both sides, we have
1
N N∑ k=1 E ‖T (zk)‖2H−1k−1 ≤ 8D2δ2(1 + d(N − 1)α) η2N +
100 ( σ2/m+ d ( 2δ2Nα +G2 )) N
(32)

heading: E MORE EXPERIMENTAL RESULTS ON CIFAR10
text: In Figure 7, we compare the performance of OSG, Alternating Adam (AlterAdam) and OAdagrad under the same minibatch size setting on CIFAR10 dataset, where one epoch means one pass of the dataset. We can see that OAdagrad and Alternating Adam behave consistently better than OSG. When the minibatch size is small (e.g., 64), OAdagrad and Alternating Adam have comparable performance, but when the minibatch size is large (e.g., 128, 256), OAdagrad converges faster than Alternating Adam. This phenomenon shows the benefits of OAdagrad when large minibatch size is used.

heading: F THE EQUIVALENCE BETWEEN OSG IN UNCONSTRAINED CASE AND THE
text: ALGORITHM IN DASKALAKIS ET AL. (2017) Define ĝk = 1mk ∑mk i=1 T (zk; ξ i k), then the update rule of Algorithm 1 becomes
zk = xk−1 − ηĝk−1 (33)
and xk = xk−1 − ηĝk. (34)
These two equalities together imply that
zk+1 = xk − ηĝk = xk−1 − 2ηĝk = zk + ηĝk−1 − 2ηĝk, (35)
where the first equality comes from (33) by replacing k to k + 1, the second equality holds by (34), and the third equality holds by using (33) again. (35) is the algorithm in (Daskalakis et al. 2017).

heading: G THE EXISTENCE OF MVI SOLUTION MAY NOT IMPLY PSEUDO-MONOTONICITY
text: Consider the function f : R→ R, where
f(x) = { cos(x) if 0 ≤ x ≤ 2π 1 if x ≤ 0 or x ≥ 2π
Define T (x) = ∇f(x). Then T (x) = − sin(x) if 0 ≤ x ≤ 2π and T (x) = 0 if x ≤ 0 or x ≥ 2π. Then we know that π is the solution of both SVI (i.e. 〈T (π), x − π〉 ≥ 0 for any x ∈ X ) and MVI (i.e. 〈T (x), x − π〉 ≥ 0 for any x ∈ X ). However T is not pseudo-monotone. To see this, take x = 0 and y = π4 and we have 〈T (x), y− x〉 = 0 and 〈T (y), y− x〉 < 0, which means that T is not pseudo-monotone.

Review summaries:
This paper proposes two methods named OSG and OAdagrad for solving stochastic non-convex non-concave min-max problems. In theoretical analyses, a convergence rate of $O(\epsilon^{-4})$ and a much better rate are provided for OSG and OAdagrad, respectively, to find the $\epsilon$-accurate first-order stationary point. Finally, the superior performance of the proposed method is empirically verified on training generative adversarial networks (GANs).