Fast Extra Gradient Methods for Smooth Structured Nonconvex-Nonconcave Minimax Problems
Sucheol Lee, Donghwan Kim
Introduction
Recently, nonconvex-nonconcave minimax problems have received an increased attention in the optimization community and the machine learning community due to their applications to generative adversarial network and adversarial training . In this paper, we consider a smooth structured nonconvex-nonconcave minimax problem:
So far little is known under the nonconvex-nonconcave setting, compared to the convex-concave setting. Recent works studied extragradient-type methods for minimax problems under various structured nonconvex-nonconcave settings. In other words, they consider various non-monotone conditions on , such as the Minty variational inequality (MVI) condition , the weak MVI condition , and the negative comonotonicity . Relations between the conditions on considered in this paper is summarized in Figure 1. Among them, this paper focuses on the negative comonotonicity condition for a Lipschitz continuous . To the best of our knowledge, the following two-time-scale variant of the extragradient method, named EG+:
is the only known (explicit) A proximal point method converges under the negative comonotonicity , but such implicit method is not preferable over explicit methods in practice due to its implicit nature. method, using , that converges under the considered setting The EG+ was originally shown to work under the weak MVI condition of , which is weaker than the negative comonotonicity. , where . The EG+, however, has a slow rate on the squared gradient norm. Note that a similar two-time-scale approach has been found to stabilize the stochastic extragradient method with unbounded noise variance .
Meanwhile, under the smooth convex-concave setting, recent works suggest that Halpern-type (or anchoring) methods, performing a convex combination of an initial point and the last updated point at each iteration, has a fast rate in terms of the squared gradient norm. In particular, developed the following anchoring variant of the extragradient method, named extra anchored gradient (EAG):
This is the first (explicit) method with a fast rate on the squared gradient norm, when satisfies both the Lipschitz continuity and the monotonicity. also showed that such rate is optimal for first-order methods using a Lipschitz continuous and monotone .
Built upon both EG+ and EAG, this paper studies the following class of two-time-scale anchored extragradient methods, named fast extragradient (FEG):
Note that (Class FEG) reuses the term in the update, unlike the standard extragradient-type methods, which we found essential for handling the negative comonotonicity condition. We leave further understanding the use of and the formulation of (Class FEG) as future work. The proposed FEG method (with appropriately chosen step coefficients , and discussed later) has an rate on the squared gradient norm, under the Lipschitz continuity and the negative comonotonicity conditions on . To the best of our knowledge, this is the first accelerated method under the nonconvex-nonconcave setting. The FEG also has value under the smooth convex-concave setting. First, when is Lipschitz continuous and monotone, the rate bound of FEG is about 27/4 times smaller than that of EAG. Also note that the rate bound of FEG is only about four times larger than the lower complexity bound of first-order methods under such setting , further closing the gap between the lower and upper complexity bounds. Second, when is cocoercive, FEG has a rate faster than that of a version of Halpern iteration in .
We also develop an adaptive variant of FEG, named FEG-A, which updates its parameters, and in (Class FEG), adaptively using a backtracking line-search . FEG requires the knowledge of the two problem parameters for the Lipschitz continuity and the comonotonicity of . However, those global parameters can be conservative, and in practice, they are even usually unknown. For such cases, the FEG-A adaptively and locally estimates the problem parameters, while preserving the fast rate on the squared gradient norm for smooth structured nonconvex-nonconcave minimax problems.
Our main contributions are summarized as follows.
We propose the FEG method that has an accelerated convergence rate on the squared gradient norm for smooth structured nonconvex-nonconcave minimax problems.
We present that the FEG method has a rate faster than those of the EAG and the Halpern iteration for smooth convex-concave problems.
We construct a backtracking line-search version of FEG, named FEG-A, for the case where the Lipschitz constant and comonotonicity parameters of are unavailable.
We analyze a stochastic version of FEG, named S-FEG, for smooth convex-concave problems.
Related work
The extragradient method is one of the widely used methods for solving smooth convex-concave minimax problems (see, e.g., for its extensions and applications). In terms of the duality gap, , where and are compact The convergence analysis on the duality gap of the extragradient type methods are generalized under the unbounded domain assumption in . domains, the ergodic iterate of the extragradient-type methods have an rate. Such rate on the duality gap is order-optimal for the first-order methods , leaving no room for improvement. On the other hand, the last iterate of the extragradient method has a slower rate on the duality gap, under an additional assumption that has a Lipschitz derivative . In terms of the squared gradient norm, , the best iterate of the extragradient-type methods have an rate . The last iterate of the extragradient method also has a rate , when is further assumed to have a Lipschitz derivative . Unlike the duality gap, the rate on the squared gradient norm is not optimal . From now on throughout this paper, we mainly study and compare the convergence rates on the squared gradient norm, which still has room for improvement in convex-concave problems, and has meaning for nonconvex-nonconcave minimax problems, unlike the duality gap.
2 Methods for nonconvex-nonconcave minimax problems
For -Lipschitz continuous , showed that the extragradient-type methods have an rate on the squared gradient norm under the MVI condition, and developed the (EG+) method under the weak MVI condition (and thus under the negative comonotonicty), which also has an rate on the squared gradient norm. To the best of our knowledge, there is no known accelerated method for the nonconvex-nonconcave setting; our proposed FEG method is the first method to have a fast rate under the nonconvex-nonconcave setting. The convergence rates of the existing methods and the FEG on the squared gradient norm are summarized in Table 1.
Preliminaries
For some , satisfies
For some \rho\in\big{(}-\frac{1}{2L},\infty\big{)}, satisfies
The -comonotonicity consists of three cases depending on the choice of ; the negative comonotonicity when , the monotonicity when , and the cocoercivity when . The negative comonotonicity is weaker than the other two, and is the main focus of this paper. The following is an examplary nonconvex-nonconcave condition that is stronger than the negative comonotonicity .
Let be twice continuously differentiable and -weakly-convex-weakly-concave. Further assume that satisfies
for some and , named -interaction dominant condition in . Then, the saddle gradient of satisfies the -negative comonotonicity. (See Appendix A.1.) For any -weakly-convex-weakly-concave function, the condition (2) holds with . Its extreme case is , where there is no interaction between and . On the other hand, when the the second terms in the left-hand side of (2) are sufficently positive definite, a nonconvex-nonconave function satisfies the condition (2) with a nonnegative . In specific, the -interaction dominant condition is satisfied when the interaction term of Hessian is dominating any negative curvature in Hessians and .
We next present our proposed FEG, and illustrate that the FEG outperforms existing methods such as EG+, EAG, and the Halpern iteration, for each three comonoticity case, respectively.
Fast extragradient (FEG) method for Lipschitz continuous and comonotone operators
This section considers an instance of (Class FEG) with , , and for all . The resulting method, named FEG, is illustrated in Algorithm 1, which has an fast rate with respect to the squared gradient norm, in Theorem 4.1. The proof of Theorem 4.1 is provided in Section 7.
For the -Lipschitz continuous and -comonotone operator with and for any , the sequence generated by FEG satisfies, for all ,
The following example shows that the bound (3) of the FEG is exact for and . The bound (3) is not known to be exact in general, and we leave finding the exact bound as future work.
We next compare the rate bound (3) with existing analyses for the three cases , , and .
Under the negative comonotonicity with , the (EG+) method with and has an rate on the squared gradient norm. To the best of our knowledge, this is the best known rate, and the FEG has a faster rate with a wider region of convergence .
2 Comparison to EAG under the monotonicity (ρ=0𝜌0\rho=0)
For an -Lipschitz continuous and monotone operator , proposed two EAG methods, named EAG-C and EAG-V, with same but with different choices of . EAG-C sets to be a constant for all in (EAG), and has a large constant in its convergence rate, for all . On the other hand, while EAG-V requires a complicated recursive update for , \alpha_{k+1}=\frac{\alpha_{k}}{1-\alpha_{k}^{2}L^{2}}\big{(}1-\frac{(k+2)^{2}}{(k+1)(k+3)}\alpha_{k}^{2}L^{2}\big{)} for all , with , its rate has a smaller constant .
The FEG takes a constant , unlike EAG-V, but has an even smaller constant in its convergence rate for . Therefore, the FEG with has about -times and -times faster convergence rate compared to those of EAG-C and EAG-V, respectively. Furthermore, the rate bound of FEG with is only about -times larger than the lower complexity bound of first-order methods under the considered setting , reducing the gap between the lower and upper complexity bounds from to .
3 Comparison to the Halpern iteration under the cocoercivity (ρ>0𝜌0\rho>0)
For a -cocoercive operator , an (explicit) version of Halpern iteration , studied in , has a fast rate, . Note that while the -cocoercivity implies the -Lipschitz continuity, there is case where the -cocoercive (and thus Lipschitz continuous) operator has a Lipschitz constant smaller than . Since , the FEG has a rate that is faster than that of Halpern iteration. However, if we take into account that the FEG requires computing the saddle gradient twice per iteration, unlike Halpern iteration studied in , the FEG method has a slower rate in terms of the number of gradient computations. If we narrow down to the case , the FEG has a faster rate, . For such case, the FEG has a rate faster than that of the Halpern iteration, even in terms of the number of gradient computations.
4 Toy example
We performed a toy experiment on a simple quadratic function, which has an -Lipschitz continuous and -comonotone saddle gradient. For the case and , Figure 2 illustrates that the FEG converges with an accelerated rate whereas EG+, EAG-C, EAG-V, and the (explicit) version of Halpern iteration diverge. This example presents that the existing guarantees on convergence and acceleration of the aforementioned methods under the convex-concave setting do not generalize to the nonconvex-nonconcave setting.
FEG with backtracking line-search
The FEG requires the knowledge of the two global parameters and for Lipschitz continuity and comonotonicity, respectively. Those global parameters are often difficult to compute in practice and can be locally conservative. To handle these two disadvantages, we employ the backtracking line-search technique in FEG. We adaptively decrease the two step size parameters, and , to satisfy the both conditions, the local -Lipschitz continuity and the -comonotonicity. In specific, and locally estimate and , respectively. One could have directly estimate , instead of , but this complicates the line-search process to handle both positive and negative values of , unlike our choice of in FEG-A. A pseudocode of the resulting method, named FEG-A, is illustrated in Algorithm 2. For a detailed description of the FEG-A, see Algorithm 4 in Appendix C.1.
The following lemma shows that each of the nonincreasing sequences and of the FEG-A has a positive lower bound, and thus FEG-A is well-defined This requires one to chooses strictly greater than the unknown value when ., under the condition . This condition for can be weaker than the condition of FEG, since the local Lipschitz parameter can be smaller than . This is another benefit of using a backtracking line-search in FEG, over the standard FEG.
For the -Lipschitz and -comonotone operator and a given constant , the step size of FEG-A is lower bounded by a positive value \underline{\tau}:=\min\big{\{}\tau_{-1},\frac{1-\delta}{L}\big{\}} for all , and if , the step size is lower bounded by a positive value \min\big{\{}\eta_{0},(1-\delta)\big{(}\tau_{k}+2\rho\big{)}\big{\}} for all .
The FEG-A method also has the following rate with respect to the squared gradient norm in Theorem 5.1, when . The proof is provided in Section 7 and Appendix C.3.
For the -Lipschitz and -comonotone operator and for any , the sequence generated by FEG-A satisfies
for all , if .
This rate bound of FEG-A reduces to that of FEG in Theorem 4.1, when we choose and for FEG-A.
FEG under stochastic setting
The following theorem provides an upper bound of the expected squared gradient norm for the S-FEG. (See Appendix D.3 for the proof.)
for all . Furthermore, if , and for all , then the bound (4) reduces to
Here, we needed the noise variance to decrease in the order of so that the stochastic error of the S-FEG does not accumulate. Otherwise, if is a constant for all , the error accumulates with rate . In short, the S-FEG will suffer from error accumulation, unless the stochastic error decreases with rate . Such error accumulation behavior also appears in a stochastic version of Nesterov’s fast gradient method for smooth convex minimization . Similar to , we believe that adjusting the step coefficients of the S-FEG can make the S-FEG become relatively stable even with a constant noise, which we leave as future work.
Convergence analysis with nonincreasing potential lemma
We analyze FEG and FEG-A by finding a nonincreasing potential function in a form in the lemma below. We provide a similar potential lemma for S-FEG in Appendix D.2. The convergence analyses of EAG and Halpern iteration are also based on such potential function .
for all . Assume that the following conditions are satisfied.
with , , ,
for all satisfies for all .
Based on the above potential lemma, we next provide a convergence analysis of FEG. The analyses for the convergence rate of FEG-A and S-FEG, i.e., the proofs of Theorem 5.1 and Theorem 6.1, are similar to that of FEG and are provided in Appendix C.3 and Appendix D.3.
Proof of Theorem 4.1. Recall that FEG is equivalent to (Class FEG) with , , and . It is straightforward to verify that the given and satisfy the conditions in Lemma 7.1 with for all . Since
The desired result follows directly by dividing both sides by \frac{k^{2}}{2}\big{(}\frac{1}{L}+2\rho\big{)}\|\bm{F}\bm{z}_{k}\|. ∎
Discussion: first-order methods for Lipschitz continuous operators
Throughout this paper, we studied and constructed efficient methods in a class of first-order methods:
denoted by , for smooth structured nonconvex-nonconcave problems. We observed that all existing first-order methods, including the FEG, required an additional condition, such as the negative comonoticity, on a Lipschitz continuous to guarantee convergence. One would then be curious whether or not there exists an (efficient) method in class that guarantees convergence without any additional condition on a Lipschitz continuous . Unfortunately, the following lemma states that there exists a worst-case also introduce worst-case minimax examples that existing methods cannot find a stationary point. A key difference from our example is that their saddle-gradient operators are not Lipschitz continuous. In addition, the considered classes of methods in exclude EG+ and FEG, unlike the class . smooth example that none of the methods in can find its stationary point. The corresponding smooth function is illustrated in Figure 3.
Its saddle-gradient operator is -Lipschitz continuous but not comonotone. Let \bm{z}=\big{(}\bm{x},\bm{x}+\sqrt{\frac{R}{L}}\big{)} and . Since and , we get and , which implies that in the comonotonicity condition as . Then, the sequence generated by any first-order method in class with satisfies for all .
The lemma implies that one should consider a class of methods, other than the class , to guarantee finding a stationary point of any smooth problem, which we leave as future work. We also leave finding additional conditions for a Lipschitz continuous , weaker than the weak MVI condition and the negative comonotonicity (with ), which guarantee convergence or its accelerated rate, respectively, as future work.
Conclusion
This paper proposed a two-time-scale and anchored extragradient method, named FEG, for smooth structured nonconvex-nonconcave problems. The proposed FEG has an accelerated rate, with respect to the squared gradient norm, for the Lipschitz continuous and negative comonotone operators for the first time. The FEG also has value for smooth convex-concave problems, compared to existing works. We further studied its backtracking line-search version, named FEG-A, for the smooth structured nonconvex-nonconcave problems and studied its stochastic version, named S-FEG, for smooth convex-concave problems. We leave extending this work to stochastic, composite, or more general nonconvex-nonconcave setting and applying to more realistic problems as future work.
Acknowledgments and Disclosure of Funding
This work was supported in part by the National Research Foundation of Korea (NRF) grant funded by the Korea government (MSIT) (No. 2019R1A5A1028324), the POSCO Science Fellowship of POSCO TJ Park Foundation, and the Samsung Science and Technology Foundation (No. SSTF-BA2101-02).
References
Appendix
Let be the saddle envelope of :
and be its saddle gradient operator. Proposition 2.10 in shows that satisfies
This implies that is -strongly monotone (and thus monotone).
It is enough to show that is monotone if and only if is -comonotone. By Lemma 2.5 in , we have the relationship , where \bm{R}:=\big{(}\bm{I}+\frac{1}{\eta}\bm{F}\big{)}^{-1} denotes the standard resolvent of . The resolvent is injective for . Let . Then, is monotone if and only if
which corresponds to the -comonotonicity of . ∎
B Proof for Section 4
Starting from , it is easy to verify that , , \bm{z}_{1+1/2}=\big{(}\frac{1}{2},1\big{)}, and . We next use the induction to show that \bm{z}_{k}=\big{(}0,\frac{2}{k}\big{)} for and for all . Assume that \bm{z}_{k}=\big{(}0,\frac{2}{k}\big{)} for some . Then, the next eight consecutive iterates are as follows:
so \bm{z}_{4l+6}=\big{(}0,\frac{2}{4l+6}\big{)}. Therefore, we get \bm{z}_{4l+2}=\big{(}0,\frac{1}{2l+1}\big{)} for all . ∎
C Algorithm and proofs for Section 5
A detailed description of the FEG-A, in Algorithm 2, is provided in Algorithm 4.
C.2 Proof of Lemma 5.1
We show that \tau_{k}\geq\underline{\tau}:=\min\big{\{}\tau_{-1},\frac{1-\delta}{L}\big{\}} for all , and for all by contradiction. Note that since and , both and are positive.
First, suppose that for some . (1) For the case , we get for all by the definition of , which contradicts to the assumption . (2) Consider the case , where the assumption reduces to . For , by the definition of , we get where , which contradicts to the -Lipschitz continuity of as . For , by the definition of , there exists such that the two corresponding iterates
satisfy for some . However, this inequality contradicts to the -Lipschitz continuity of as . Therefore, we have for all .
Similarly, suppose that for some . (1) For the case , we get for all by the definition of , which contradicts to the assumption . (2) Consider the case , where the assumption reduces to . Then by the definition of , there exists such that the two corresponding iterates
satisfy for some . However, this inequality contradicts to the -comonotonicity of as . Therefore, we have for all . ∎
C.3 Proof of Theorem 5.1
Note that FEG-A is equivalent to (Class FEG) with , , and . The given sequence in FEG-A satisfies the conditions in Lemma 7.1 with :
where the inequality follows from the fact that and are nonincreasing sequences. Since
Then by dividing both sides by and using Lemma 5.1, we get
D Proofs for Section 7
Hence, the sum of (6) and (7) with multiplying factor yields
Next, for , here we note the following relations for later use:
Hence, the sum of (8) and (D.1) with multiplying factor yields
Note that the given conditions imply that
Note that \{\alpha_{k}\}_{k\geq 1}\subseteq\big{(}0,\frac{1}{L_{k}}\big{]} and are the sufficient conditions for and for all . ∎
D.2 Convergence analysis for S-FEG
In this section, we consider the following class of stochastic methods:
Let be the sequence generated by (Class S-FEG) with and satisfying , \alpha_{k}\in\big{(}0,\frac{1}{L}\big{]}, , for all , and
We first prove the following lemma that is used in the proof of Lemma D.1.
Proof of Lemma D.1. First, for , note that
The sum of (10) and (11) with multiplying factor yields
Hence, the sum of (12) and (D.2) with multiplying factor yields
By the given conditions, we get and
where the last inequality follows from Lemma D.2. ∎
D.3 Proof of Theorem 6.1
Note that S-FEG is equivalent to (Class S-FEG) with and . It is straightforward to verify that the given and satisfy the conditions in Lemma D.1 for all . By noting that
In addition, if , and for all , then we have