Near-Optimal Algorithms for Minimax Optimization
Tianyi Lin, Chi Jin, Michael. I. Jordan
Introduction
The theoretical study of solutions of problem (1) has been an focus of several decades of research in mathematics, statistics, economics and computer science (Basar and Olsder, 1999; Nisan et al., 2007; Von Neumann and Morgenstern, 2007; Facchinei and Pang, 2007; Berger, 2013). Recently, this line of research has become increasingly relevant to algorithmic machine learning, with applications including robustness in adversarial learning (Goodfellow et al., 2014; Sinha et al., 2018), prediction and regression problems (Cesa-Bianchi and Lugosi, 2006; Xu et al., 2009) and distributed computing (Shamma, 2008; Mateos et al., 2010). Moreover, real-world machine-learning systems are increasingly embedded in multi-agent systems or matching markets and subject to game-theoretic constraints (Jordan, 2018).
Can we design first-order algorithms that achieve the lower bounds in these settings?
This paper presents an affirmative answer by resolving the above open problem up to logarithmic factors. More specifically, our contribution is as follows.
We provide a head-to-head comparison between our results and existing results in the literature in Table 1 for convex-concave settings, and Table 2 for nonconvex-concave settings.
Related work
To the best of our knowledge, the earliest algorithmic schemes for solving the bilinear minimax problem, , date back to Brown’s fictitious play (Brown, 1951) and Dantzig’s simplex method (Dantzig, 1998). This problem can also be solved by Korpelevich’s extragradient (EG) algorithm (Korpelevich, 1976), which can be shown to be linearly convergent when is square and full rank (Tseng, 1995). There are also several recent papers studying the convergence of EG and its variants; see Chambolle and Pock (2011); Malitsky (2015); Yadav et al. (2018) for reflected gradient descent ascent, Daskalakis et al. (2018); Mokhtari et al. (2019b, a) for optimistic gradient descent ascent (OGDA) and Rakhlin and Sridharan (2013a, b); Mertikopoulos et al. (2019); Chavdarova et al. (2019); Hsieh et al. (2019); Mishchenko et al. (2019) for other variants. In the bilinear setting, Daskalakis et al. (2018) established the convergence of the optimistic gradient descent ascent (OGDA) method to a neighborhood of the solution; Liang and Stokes (2019) proved the linear convergence of the OGDA algorithm using a dynamical system approach. Very recently, Mokhtari et al. (2019b) have proposed a unified framework for achieving the sharpest convergence rates of both EG and OGDA algorithms.
For the convex-concave minimax problem, Nemirovski (2004) proved that his mirror-prox algorithm returns an -saddle point within the gradient complexity of when and are bounded. This algorithm was subsequently generalized by Auslender and Teboulle (2005) to a class of distance-generating functions, and the complexity result was extended to unbounded sets and composite objectives (Monteiro and Svaiter, 2010, 2011) using the hybrid proximal extragradient algorithm with different error criteria. Nesterov (2007) developed a dual extrapolation algorithm which possesses the same complexity bound as in Nemirovski (2004). Later on, Tseng (2008) presented a unified treatment of these algorithms and a refined convergence analysis with same complexity result. Nedić and Ozdaglar (2009) analyzed the (sub)gradient descent ascent algorithm for convex-concave saddle point problems when the (sub)gradients are bounded over the constraint sets. Abernethy et al. (2019) presented a Hamiltonian gradient descent algorithm with last-iterate convergence under a “sufficiently bilinear” condition.
Several papers have studied special cases in the convex-concave setting. For the special case when the objective function is a composite bilinear form, , Chambolle and Pock (2011) introduced a primal-dual algorithm that converges to a saddle point with the rate of when the convex functions and are smooth. Nesterov (2005) proposed a smoothing technique and proved that the resulting algorithm achieves an improved rate with better dependence on Lipschitz constant of when is the convex and smooth function and are both bounded. He and Monteiro (2016) and Kolossoski and Monteiro (2017) proved that such result also hold when are unbounded or the space is non-Euclidean. Chen et al. (2014, 2017) generalized Nesterov’s technique to develop optimal algorithms for solving a class of stochastic saddle point problems and stochastic monotone variational inequalities. For a class of certain purely bilinear games where and are zero functions, Azizian et al. (2020) demonstrated that linear convergence is possible for several algorithms and their new algorithm achieved the tight bound. The second case is the so-called affinely constrained smooth convex problem, i.e., . Esser et al. (2010) proposed a primal-dual algorithm while Lan and Monteiro (2016) provided a first-order augmented Lagrangian method with the same rate. By exploiting the structure, Ouyang et al. (2015) proposed a near-optimal algorithm in this setting.
Preliminaries
In this section, we clarify the notation used in this paper, review some background and provide formal definitions for the class of functions and optimality measure considered in this paper.
1 Minimax optimization
Furthermore, is -strongly-concave if is -strongly-convex. If we set , then we recover the definitions of convexity and concavity for a continuous differentiable function.
we assume that is convex for each and is concave for each . Here and are both convex and bounded. Under these conditions, the Sion’s minimax theorem (Sion, 1958) guarantees that
Furthermore, there exists at least one saddle point (or Nash equilibrium) such that the following equality holds true:
Therefore, for any point , the duality gap forms the basis for a standard optimality criterion. Formally, we define
A point is an -saddle point of a convex-concave function if . If , then is a saddle point.
Nonconvex-concave setting:
If , then is a stationary point.
We note that this notion of stationarity of (Definition 3.5) is closely related to an optimality notion in terms of stationary points of the function for nonconvex-concave functions. We refer readers to Appendix A.1 for more discussion.
2 Nesterov’s accelerated gradient descent
The following theorem provides an upper bound on the gradient complexity of AGD; i.e., the total number of gradient evaluations to find an -optimal point in terms of function value.
Algorithm Components
In this section, we present two main algorithm components. Both of them are crucial for our final algorithms to achieve near-optimal convergence rates.
Our first component is the Accelerated Proximal Point Algorithm (APPA, Algorithm 2) for minimizing a function . Comparing APPA with classical AGD (Algorithm 1), we note that both of them have momentum steps which yield acceleration. The major difference is in Line 4 of Algorithm 2, where APPA solves a proximal subproblem
We present an inexact version in Algorithm 2 where we tolerate a small error in terms of the function value in solving the proximal subproblem (4). That is, the solution satisfies
We conclude that APPA has a unique advantage over AGD in settings where does not have a smoothness property but the proximal step (4) is easy to solve. These settings include LASSO (Beck and Teboulle, 2009), as well as minimax optimization problems (as we show in later sections).
2 Accelerated Solver for Minimax Proximal Steps
In minimax optimization problems of the form (1), we are interested in solving the following proximal subproblem as follows,
which is equivalent to solving the following minimax problem:
For a generic strongly-convex-strongly-concave function , solving a minimax problem is equivalent to solving a maximin problem, due to Sion’s minimax theorem:
A straightforward way of solving the maximin problem is to use a double-loop algorithm which solves the maximization and minimization problems on two different time scales. Specifically, the inner loop performs AGD on function to solve the inner minimization; i.e., to compute for each , and the outer loop performs Accelerated Gradient Ascent (AGA) on the function to solve the outer maximization. Since the algorithm aims to solve a maximin problem we use AGA-AGD, and we name the algorithm Maximin-AG2. See Algorithm 3 for the formal version of this algorithm. We also incorporate Lines 8-9 to check termination conditions, which ensures that the output achieves the desired optimality. The theoretical guarantee for Algorithm 3 is given in the following theorem.
Accelerating Convex-Concave Optimization
In this section, we present our main results for accelerating convex-concave optimization. We first present our new near-optimal algorithm and its theoretical guarantee for optimizing strongly-convex-strongly-concave functions. Then, we use simple reduction arguments to obtain results for strongly-convex-concave and convex-concave functions.
With the algorithm components from Section 4 in hand, we are now ready to state our near-optimal algorithm. Algorithm 4 is a simple combination of Algorithm 2 and Algorithm 3. Its outer loop performs an inexact APPA to minimize the function , while the inner loop uses Maximin-AG2 to solve the proximal subproblem (5), which is equivalent to solving (6). At the end, after finding a near-optimal , Algorithm 4 performs another AGD on the function to find a near-optimal . The theoretical guarantee for the algorithm is given in the following theorem.
2 Strongly-convex-concave setting
Our result in the strongly-convex-strongly-concave setting readily implies a near-optimal result in the strongly-convex-concave setting. Consider the following auxiliary function for an arbitrary which is defined by
By construction, it is clear that the difference between and is small in terms of function value:
This implies, according to Definition 3.4, that any -saddle point of function is also a -saddle point of function , and thus it is sufficient to only solve the problem . Finally, when is a -strongly-convex-concave function, becomes -strongly-convex--strongly-concave, which can be fed into Algorithm 4 to obtain the following result.
3 Convex-concave setting
Similar to the previous subsection, when is only convex-concave, we can construct following strongly-convex-strongly-concave function :
which can be fed into Algorithm 4 to obtain the following result.
where is defined as in (8).
Accelerating Nonconvex-Concave Optimization
In this section, we present methods for accelerating nonconvex-concave optimization. Similar to Section 5, we first present our algorithm and its theoretical guarantee for optimizing nonconvex-strongly-concave functions. We then use a simple reduction argument to obtain results for nonconvex-concave functions. This section present results using the stationarity of the function (Definition 3.5) as an optimality measure. Please see Appendix A for additional results using the stationarity of the function as the optimality measure (Definition A.1 and A.5).
Our algorithm for nonconvex-strongly-concave optimization is described in Algorithm 5. Similar to Algorithm 4, we still use our accelerated solver Maximin-AG2 for the same proximal subproblem in the inner loop. The only minor difference is that, in the outer loop, Algorithm 5 only uses the Proximal Point Algorithm (PPA) on function without acceleration (or momentum steps). This is due to fact that gradient descent is already optimal among all first-order algorithm for finding stationary points of smooth nonconvex functions (Carmon et al., 2019a). The standard acceleration technique will not help for smooth nonconvex functions. We presents the theoretical guarantees for Algorithm 5 in the following theorem.
2 Nonconvex-concave setting
Our result in the nonconvex-strongly-concave setting readily implies a fast result in the nonconvex-concave setting. Consider the following auxiliary function for an arbitrary :
Conclusions
This paper has provided the first set of near-optimal algorithms for strongly-convex-(strongly)-concave minimax optimization problems and the state-of-the-art algorithms for nonconvex-(strongly)-concave minimax optimization problems. For the former class of problems, our algorithms match the lower complexity bound for first-order algorithms (Ouyang and Xu, 2019; Ibrahim et al., 2019; Zhang et al., 2019) up to logarithmic factors. For the latter class of problems, our algorithms achieve the best known upper bound. In the future research, one important direction is to investigate the lower complexity bound of first-order algorithms for nonconvex-(strongly)-concave minimax problems. Despite several striking results on lower complexity bounds for nonconvex smooth problems (Carmon et al., 2019a, b), this problem remains challenging as solving it requires a new construction of “chain-style” functions and resisting oracles.
Acknowledgments
We would like to thank three anonymous referees for constructive suggestions that improve the quality of this paper. This work was supported in part by the Mathematical Data Science program of the Office of Naval Research under grant number N00014-18-1-2764.
References
Appendix A Additional Results for Nonconvex-Concave Optimization
In this section, we present our results for nonconvex-concave optimization using stationary of (Definition A.1 and Definition A.5) as the optimality measure.
One approach, inspired by nonconvex optimization, is to equivalently reformulate problem (1) as the following nonconvex minimization problem:
and define an optimality notion for the local surrogate of global optimum of . In robust learning, is the classifier while is the adversarial noise. Practitioners are often only interested in finding a robust classifier instead of an adversarial response to each data point. Such a stationary point precisely corresponds to a robust classifier that is stationary to the robust classification error.
We call an -stationary point of a smooth function if . If , then is called a stationary point.
In contrast, when is merely concave for each , is not necessarily smooth and even not differentiable. A weaker sufficient condition for the purpose of our paper is the weak convexity.
A.2 Nonconvex-strongly-concave setting
In the setting of nonconvex-strongly-concave function, we still use Algorithm 5. Similar to Theorem 6.1, we can obtain a guarantee, which finds a point satisfying in the same number of iterations as in Theorem 6.1.
A.3 Nonconvex-concave setting
We can similarly reduce the problem of optimizing a nonconvex-concave function to the problem of optimizing a nonconvex-strongly-concave function. The only caveat is that, in order to achieve the near-optimal point using Definition A.5 as optimality measure, we can only add a term as follows:
Appendix B Proofs for Algorithm Components
In this section, we present proofs for our algorithm components.
We divide the proof into three parts. In the first part, we show that the output satisfies . In the second part, we derive the sufficient condition for guaranteeing the stopping criteria in Algorithm 1. In the third part, we derive the gradient complexity of the algorithm using the condition derived in the second part.
Summing up the above two inequalities and rearranging yields that
Part II.
Putting these pieces together yields the desired sufficient condition as follows,
Part III.
We proceed to derive the gradient complexity of the algorithm using the condition in Eq. (14). Since Algorithm 1 is exactly Nesterov’s accelerated gradient descent, standard arguments based on estimate sequence [Nesterov, 2018] implies
Therefore, the gradient complexity of Algorithm 1 to guarantee Eq. (14) is bounded by
B.2 Proof of Theorem 4.1
Proof. Using the definition of in Algorithm 2, we have
Putting these pieces together yields that
Putting these pieces together with yields the desired inequality.
The remaining proof is based on Lemma B.1. Indeed, we have
Using the Young’s inequality again, we have
Putting Eq. (B.2)-Eq. (20) together with , we have
Combining Eq. (16) and Eq. (B.2) yields that
Repeating the above inequality yields that
Since the tolerance , we conclude that the iteration complexity of Algorithm 2 to guarantee that if there exists an absolute constant such that
B.3 Proof of Theorem 4.2
Before presenting the main proof, we define the following important functions:
All the above functions are well defined since is strongly convex-concave. We provide their complete characterization in the following structural lemma.
Under the assumptions imposed in Theorem 4.2, we have
A function is -Lipschitz.
A function is -Lipschitz.
In the second part, we get the sufficient condition for guaranteeing the stopping criteria in Algorithm 3. In the third part, we estimate an upper bound for the gradient complexity of the algorithm using the condition derived in the second part. For the ease of presentation, we denote as the unique solution to the minimax optimization .
By the definition of , the inequality in Eq. (22) can be rewritten as follows,
Using the Young’s inequality, we have . Putting these pieces together yields with yields that
In what follows, we prove that if the following stopping conditions hold true,
Indeed, we observe that . By definition, we have . Also, is -Lipschitz. Therefore, we have
First, we bound the term . Since is -strongly convex, we have
It remains to bound the term . Indeed, we have and
Putting these pieces together with Eq. (25) and Eq. (28) yields that
Summing up the above two inequalities and rearranging yields that
Plugging Eq. (28) and Eq. (31) into Eq. (26) yields that
Plugging Eq. (28) and Eq. (31) into Eq. (27) yields that
Putting these pieces together Eq. (23) yields the desired result.
Part II.
Furthermore, and
Also, Eq. (24) guarantees that Eq. (28) holds true. Then we have
Putting these pieces together yields the desired condition as follows,
Part III.
Putting these pieces together yields the desired inequality.
The remaining proof is based on the modification of Nesterov’s techniques [Nesterov, 2018, Section 2.2.5]. Indeed, we define the estimate sequence as follows,
We apply the inductive argument to prove,
It follows from the recursive rule for and its canonical form that
The recursive rule for can be achieved by solving . Then we have
Then we conclude the recursive rule for by plugging the recursive rule for into the above equality. By the induction, Eq. (33) holds true when which implies
Applying Lemma B.3 with and further implies that
Putting these pieces together yields that
On the other hand, Lemma B.3 and the update formula for implies that
Since , we have
Repeating the above inequality yields that
Now it suffices to establish the gradient complexity of the two AGD subroutines at each iteration. In particular, we use the gradient complexity of the AGD subroutine to guarantee that is bounded by
Appendix C Proofs for Convex-Concave Settings
In this section, we present proofs for all results in Section 5.
First, we note that Minimax-APPA in Algorithm 4 can be interpreted as an inexact accelerated proximal point algorithm Inexact-APPA with the inner loop solver Maximin-AG2 and AGD. Using Theorem 3.6 and Theorem 4.1, the point satisfies
We let and note that is -strongly convex function. Since is -strongly-convex--strongly-concave, the Nash equilibrium is unique and . Therefore, we have
Since is -strongly concave, Nesterov [2018, Theorem 2.1.5] implies that
Since is -Lipschitz (cf. Lemma B.2), . Thus, we have
Let . By the definition of , the following inequality holds true for any ,
Furthermore, we call the solver Maximin-AG2 at each iteration. Using Theorem 4.2 and , the number of gradient evaluations at each iteration is bounded by
Recalling , we conclude that the total number of gradient evaluations is bounded by
C.2 Proof of Corollary 5.2
Since the function is concave for each , we have
Putting these pieces together yields that .
C.3 Proof of Corollary 5.3
Since the function is concave for each , we have
Putting these pieces together yields that .
Appendix D Proofs for Nonconvex-Concave Settings
In this section, we present proofs for all results in Section 6 and Section A
Putting Eq. (34), Eq. (35) and Eq. (D.1) together with the Cauchy-Schwarz inequality yields
Summing up the above inequality over and dividing it by yields that
Putting these pieces together yields that
Therefore, we conclude that the total number of gradient evaluations is bounded by
D.2 Proof of Corollary 6.2
This implies that the following statement holds for all that
D.3 Proof of Theorem A.7
Using the same argument as in Theorem 6.1, we have
Putting Eq. (37), Eq. (38) and Eq. (39) together with the Cauchy-Schwarz inequality yields
Summing up the above inequality over and dividing it by yields that
Therefore, we conclude that the total number of gradient evaluations is bounded by
D.4 Proof of Corollary A.8
Recall that the function is defined by
This implies that the following statement holds for all that
Using Theorem A.7 and letting , we have
Putting these pieces together yields that
Since a point satisfies that
Appendix E Proof of Technical Lemmas
In this section, we provide complete proofs for the lemmas in the paper.
We provide a proof for an expanded version of Lemma A.4.
E.2 Proof of Lemma A.6
E.3 Proof of Lemma B.2
Summing up Eq. (40) with and Eq. (41) with yields
Since is -strongly concave, we have
Summing up the above two inequalities yields that
Therefore, we conclude that the function is -Lipschitz.
Part (b):
Since is -strongly convex for each , we have
Therefore, the function is -strongly convex.
Part (c):
Summing up Eq. (42) with and Eq. (43) with yields
Since is -strongly convex, we have
Summing up the above two inequalities yields that
Therefore, we conclude that the function is -Lipschitz.
Part (d):
Since is -strongly concave for each , we have
Therefore, the function is -strongly concave.