A Unified Single-loop Alternating Gradient Projection Algorithm for Nonconvex-Concave and Convex-Nonconcave Minimax Problems
Zi Xu, Huiling Zhang, Yang Xu, Guanghui Lan
Introduction
We consider the following minimax optimization problem:
Minimax optimization problems have been studied for many years, but most previous works focused on convex-concave minimax problems, i.e., is convex with respect to and concave with respect to Nedic ; Boyd ; Chen . Under this setting, Nemirovski Nemi2004 proposed a mirror-prox algorithm which returns an -saddle point within the complexity of when and are bounded sets. Nesterov Nes2007 developed a dual extrapolation algorithm which owns the same complexity bound as in Nemi2004 . Monteiro and Svaiter Mon2010 ; Mon2011 extended the complexity result to unbounded sets and composite objectives by using the hybrid proximal extragradient algorithm with a different termination criterion. Tseng Tseng2008 proved the same result using a refined convergence analysis. Abernethy et al. Abernethy presented a Hamiltonian gradient descent algorithm with last-iterate convergence under a “sufficient bilinear” condition. A few other papers have studied special cases in the convex-concave setting, for more details, we refer to Chen ; Chen2017 ; Dang ; He2016 ; Lan2016 ; Lin2020 ; Ouyang2015 ; Ouyang2019 and the references therein.
As mentioned earlier, another interesting class of minimax problems is the (strongly) convex-nonconcave setting of (P), i.e., is (strongly) convex w.r.t. and nonconcave w.r.t. . However, for any given , to solve the inner maximization subproblem, i.e., , is already NP-hard. Due to this reason, almost all the existing nested-loop algorithms will lose their theoretical guarantees since they need to solve the inner subproblem exactly, or approximately with an error proportional to the accuracy . Most existing single-loop algorithms, e.g., HiBSA or GDmax, will also get stuck, since they require the solution of the inner maximization problem. One possible alternative approach would be to switch the order of the “” and “” operators. However, in general, when is nonconvex w.r.t. or nonconcave w.r.t. . The set of stationary points for these problems obtained by switching the order of “” and “” could also be different under some criterions, e.g., with which is meaningful when is strongly concave with respect to as defined in Lin2020 . Whereas the set of stationary points for the above two problems might the same (e.g., in terms of the stationarity of ), the algorithms applied to these problems may have drastically different trajectories and would converge to quite different solutions.
Contributions. In this paper, we focus on single-loop algorithms for solving nonconvex-concave and convex-nonconcave minimax problems. Our main contributions are as follows.
We propose a simple and unified single-loop Alternating Gradient Projection (AGP) algorithm for solving both nonconvex-(strongly) concave and (strongly) convex-nonconcave smooth minimax problems. At each iteration, only simple gradient projection steps are employed for updating and alternatively.
We analyze the gradient complexity of the proposed unified AGP algorithm under four different settings. For the nonconvex-concave setting, we show that an -stationary point of can be obtained in (resp. ) iterations for nonconvex-strongly concave (resp. nonconvex concave) minimax problems. To the best of our knowledge, these represent the state-of-the-art single loop algorithms under nonconvex-concave setting. Secondly, we show that the gradient complexity to obtain an -stationary point of is (resp., ) under the strongly convex-nonconcave (resp., convex-nonconcave ) setting. To the best of our knowledge, these are the first two theoretically guaranteed convergence results reported in the literature under these settings. Existing single-loop algorithms under both nonconvex-concave and convex-nonconcave settings are summarized in Table 1.
As shown in Table 1, AGP matches the best-known complexity for the nonconvex-strongly concave setting, whereas for the general nonconvex-concave setting, it improves the best-known complexity for single-loop algorithms from to . A key step in our development is to construct a suitable potential function which involves function value plus some distance between two adjacent iterates for the nonconvex-strongly concave and strongly convex-nonconcave case, whereas an additional quadratic regularization term of or is needed for the general nonconvex-concave and convex-nonconcave case (see Lemma 3.4 and Lemma 4.4). To the best of our knowledge, this is the first time that such potential functions have been constructed for solving minimax problems. Moreover, the functional descent results in Lemmas 3.1, 3.3, 4.1, and 4.3 have not been reported before and appear to be novel from our point of view. It should be noted that the complexity of AGP does not match the best-known complexity possessed by nested-loop algorithms for the general nonconvex-concave case. One possible reason is that only one gradient projection step is employed for solving the inner problem in single loop algorithms, and hence the error associated with the gradient for the outer problem will be larger than that for nested loop algorithms and such errors will accumulate as the algorithms proceeds. This also explains why their iteration complexity analysis might be more difficult than that of nested-looped algorithms. It is not evident to us whether or not this complexity bound obtained for single-loop algorithms can be further improved. Nevertheless, due to its simplicity and the fact that it does not require the input and fine-tuning of too many algorithmic parameters, the proposed AGP algorithm can numerically outperform the state-of-the-art nested loop algorithms as shown in Section 6.
Furthermore, AGP provides a flexible algorithmic framework that can be easily generalized to solve more complicated minimax problems. Based on the basic idea of the AGP algorithm, we propose a block alternating proximal gradient (BAPG) algorithm for solving more general nonsmooth multi-block nonconvex-(strongly) concave and (strongly) convex-nonconcave minimax problems. Each BAPG iteration requires only simple proximal gradient steps to update each block of the multi-block variables alternatively. We prove the gradient complexity of the proposed BAPG algorithm under these four different settings. To the best of our knowledge, existing algorithms (especially the ones with nested loop) can hardly be extended to these more complicated multi-block settings and these complexity results have not been obtained before.
The rest of this paper is organized as follows. In Section 2, we propose a unified alternating gradient projection (AGP) algorithm for nonconvex-(strongly) concave and (strongly) convex-concave minimax problems, and we then analyze the corresponding gradient complexity for four different settings in Section 3 and Section 4. We propose a block alternating proximal gradient (BAPG) algorithm for solving more general nonsmooth multi-block nonconvex-(strongly) concave and (strongly) convex-nonconcave minimax problems, and also establish the corresponding gradient complexity for four different settings in Section 5. We report some numerical results in Section 6 and make some concluding remarks in the last section.
A continuously differentiable function is called -strongly convex if there exists a constant such that for any ,
If satisfies (1.1), is called -strongly concave. A pair is a Nash equilibrium (or equivalently a saddle point) of function , if ,
A pair is a local Nash equilibrium (or equivalently a local saddle point) of , if there exists such that for any in and , (1.2) is satisfied. A pair is an -first-order Nash equilibrium of function , if and , where
An Alternating Gradient Projection Algorithm for (P)
In this section, we propose a unified alternating gradient projection (AGP) algorithm that will be used later for solving a few different classes of (P). Each iteration of the proposed AGP algorithm consists of two gradient projection steps for updating both and . Instead of the original function , at the -th iteration, AGP uses the gradient of a regularized version of the original function, i.e.,
where and are two regularization parameters. More specifically, for a given pair , AGP minimizes a linearized approximation of plus some regularized terms to update as follows:
where is the projection operator onto and denotes a stepsize parameter. Similarly, it updates by maximizing a linearized approximation of minus some regularized terms, i.e.,
where is the projection operator onto and is another stepsize parameter. The proposed AGP method is formally stated in Algorithm 1, where sequences , , , and the stopping rule in Step 4 will be specified later in each of the different problem settings to be studied.
Observe that if we set and , the AGP algorithm is exactly the alternating version of the GDA algorithm, which is rather natural and has been widely used by practitioners for solving minimax problems, e.g., in generative adversarial networks. However, to the best of our knowledge, even for the alternating GDA algorithm, the convergence for solving (P) has never been established before in the literature. Moreover, if or is not equal to , AGP algorithm is completely new. It turns out that or plays a very crucial role to guarantee the convergence of AGP when solving general nonconvex-concave or convex-nonconcave minimax problems.
Before we prove the iteration complexity of AGP algorithm for solving (P), we define the stationarity gap as the termination criterion as follows.
At each iteration of Algorithm 1, the stationarity gap for problem (P) w.r.t. is defined as:
We denote , , and .
Note that the widely used metrics for convex-concave minimax problems such as the min-max value or the distance to the set of optimal solutions of the min-max problem are not applicable in nonconvex cases. For the latter cases we call an -stationary point of if with being defined as in Definition 2.1, which has been widely used as the optimality measure, e.g., Lin2020 ; Lu . In the absence of constraints, reduces to the standard condition and for unconstrained problems. The vector also refers to gradient mapping at , see Nes2013 for the details.
At each iteration of Algorithm 1, the stationarity gap for problem (P) w.r.t. is defined as:
We also need to make the following assumption about the smoothness of .
has Lipschitz continuous gradients, i.e., there exist positive scalars , , , such that for any , ,
We denote .
Under this assumption, we can first prove the following lemma for estimating bounds on changes in the function value when or is updated at each iteration in Algorithm 1.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 1. Then we have
Firstly, by the optimality condition for (2.4), we have
By Assumption 2.1, the gradient of is Lipschitz continuous, implying that
Adding (2.8) and (2.9), we can easily show that
This completes the proof of (2.6). By the optimality condition for (2.5), we have
By Assumption 2.1, the gradient of is Lipschitz continuous, implying that
The result (2.7) then follows by adding (2.10) and (2.11).
In the following two sections, we will establish the iteration complexity of the AGP algorithm under four different problem settings. Although there are some different technical details under different problem settings, the main process used in these proofs is similar. Firstly, we estimate a bound on the change of function values between two adjacent iterates, shown as in Lemmas 3.1, 3.3, 4.1 and 4.3 respectively. Then, we estimate an upper bound on the weighted distance between two adjacent pairs of iterates by constructing a suitable potential function according to different properties of the objective function, shown as in Lemmas 3.2, 3.4, 4.2 and 4.4 respectively. Finally, by using those upper bounds, we prove the complexity of the algorithm through some careful parameter selection, shown as in Theorems 3.1, 3.2, 4.1 and 4.2.
Complexity Analysis for Nonconvex-Concave Minimax Problems
In this subsection, we analyze the iteration complexity of Algorithm 1 for solving nonconvex-strongly concave minimax optimization problems, i.e., is nonconvex w.r.t. for any fixed , and -strongly concave w.r.t. for any given . Under this setting, , we set
in Algorithm 1, and simplify the update for and as follows:
which is exactly the alternating version of GDA algorithm. Our goal in the remaining part of this subsection is to establish the iteration complexity of Algorithm 1 under the nonconvex-strongly concave setting.
We now establish an important recursion for the AGP algorithm.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 1 with parameter settings in (3.1). If , then we have
The optimality condition for in (3.3) implies that and ,
On the other hand, by replacing with and choosing in (3.5), we obtain
which, in view of the fact that is -strongly concave w.r.t. for any given , then implies that
Denoting , we can write the first inner product term in the r.h.s. of (3.1) as
Next, we estimate the three terms in the right hand side of (3.1) respectively. By Assumption 2.1 and the Cauchy-Schwarz inequality, we can bound the first two terms according to
For the third term, by -strongly-concavity of with respect to ,
Plugging (3.1)-(3.13) into (3.1) and rearranging the terms, we conclude that
By setting , in (2.6) of Lemma 2.1 and the assumption , we have
The proof is completed by combining (3.1) with (3.15).
One may want to take the telescoping sum of (3.1) in order to provide a bound on . However, since , the coefficient of the third term in the r.h.s. of (3.1) is always positive. As a result, we need to further refine this relation as shown below.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 1 with parameter settings in (3.1). Also let us denote
which together with (3.1) then imply that
By plugging (3.11),(3.12),(3.19) into (3.1), and using the identity , we conclude that
Multiplying on both sides of (3.1) and using the definition of , we obtain
It then follows from (3.1) in Lemma 3.1 and the definition of that
We are now ready to establish the iteration complexity for the AGP algorithm in the nonconvex-strongly concave setting. In particular, letting be defined as in Definition 2.1 and be a given target accuracy, we provide a bound on , the first iteration index to achieve an -stationary point, i.e., , which is equivalent to
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 1 with parameter settings in (3.1). If the relations are satisfied, then it holds that
where and with and .
It follows immediately from (3.1) and (3.2) that
On the other hand, by (3.3) and the triangle inequality, we obtain that
By combining (3.23) and (3.1), and using the Cauchy-Schwarz inequality, we obtain
Observing that . Multiplying both sides of (3.25) by , and using (3.2) in Lemma 3.2, we have
Summing up the above inequalities from to , we obtain
Note that by the definition of in Lemma 3.2, we have
where the inequality follows from the definitions of and , and the facts that () and due to the selection of . We then conclude from (3.27) that which, in view of the definition of , implies that or equivalently, .
A few remarks are in place for the results obtained in Theorem 3.1. First, in view of Theorem 3.1, the gradient complexity of Algorithm 1 to obtain a stationary point that satisfies is given by under the nonconvex-strongly concave setting. Second, under this setting, very few single-loop algorithms have been investigated although a few other existing algorithms can achieve similar complexity bound. In particular, it seems that even the complexity bound for the GDA algorithm remains unknown under this setting when and are both convex compact sets. Compared to the state-of-the-art algorithm in Lin2020 , we improve the complexity for the nonconvex-strongly concave setting by a logarithmic factor, since we do not need to solve the inner problem at each iteration. Third, as mentioned earlier, Algorithm 1 is a single-loop alternating gradient projection method with constant stepsizes, which is very easy to implement in practice.
2 Complexity Analysis for General Nonconvex-Concave Setting
In this subsection, we analyze the iteration complexity of Algorithm 1 for solving (P) under the general nonconvex-concave setting. Under this setting, , we set
where , are two constants, and are stepsize parameters to be defined later. We need to make the following assumption on the parameters .
is a nonnegative monotonically decreasing sequence.
By Assumption 2.1 and , we have
Denoting , by Assumption 3.1 and (3.2), we have
It then follows from the above inequality and the strong concavity of w.r.t. (Theorem 2.1.12 in Nestrov ) that
This is a key inequality that we will use to establish some important recursions for the AGP method under the nonconvex-concave setting in the following two results. This is also one of the key differences between the proof for the nonconvex-strongly concave setting and the one for the nonconvex-concave setting.
Suppose that Assumptions 2.1 and 3.1 hold. Let be a sequence generated by Algorithm 1 with parameter settings in (3.28). If and , then
The optimality condition for in (2.5) implies that, , ,
On the other hand, by replacing with and choosing in (3.32), we obtain
The concavity of w.r.t. together with (3.34) then imply that
where . We now provide bounds on the inner product terms of (3.2). Firstly, by the definition of and , Assumptions 2.1 and 3.1, and the Cauchy-Schwarz inequality, we have
Secondly, by the Cauchy-Schwarz inequality,
Plugging (3.2)-(3.39) into (3.2), and using the definition of and and the assumption , we obtain
By setting , in (2.6) of Lemma 2.1 and the assumption , we have
The result in (3.3) then follows by adding (3.2) and (3.41).
It turns out that from Lemma 3.3 we can not obtain an upper bound on the positively weighted sum of and to provide an upper bound for . We need to further refine this result in (3.3) to overcome this difficulty. In particular, we obtain below a new inequality as in (3.2) to further investigate the relation between and . Then by using this new inequality, we construct a new potential function as shown in the following important result for Algorithm 1. Note that in Lemma 3.2 we have constructed a potential function which involves the function value plus some distance between two adjacent iterates for the nonconvex-strongly concave cases, whereas in the following lemma an additional quadratic regularization term, i.e., , is added to construct another potential function for the general nonconvex-concave case.
Suppose that Assumptions 2.1 and 3.1 hold. Let be a sequence generated by Algorithm 1 with parameter settings in (3.28). Also let us denote
Similar to (3.2) in Lemma 3.3, (3.44) can be rewritten as
Using an argument similar to the proof of (3.2)-(3.39), the relation in (3.2), and the Cauchy-Schwarz inequality, we conclude from the above inequality that
for any . Observing by , and Assumption 3.1, we have
Combining and rearranging the terms in (3.2), we obtain
By multiplying on both sides of the above inequality, we then obtain
Setting in the above inequality, and using the definition of and (3.42), we have
Combining (3.2) and (3.3) in Lemma 3.3, and using the definition of , we conclude
We are now ready to establish the iteration complexity for the AGP algorithm to achieve an -stationary point in the general nonconvex-concave setting.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 1 with parameter settings in (3.28). If , , , then for any given ,
where , , , , with .
By and , let us denote , , we can easily see that the relations in (3.42) are satisfied. It follows from the selection of and that
This observation, in view of Lemma 3.4, then immediately implies that
We can easily check from the definition of that
By replacing with , with , similar to (3.23) and (3.1), we immediately obtain that
Combining (3.49) and (3.50), and using the Cauchy-Schwarz inequality, we have
Since both and are in the same order when becomes large enough, it then follows from the definition of that ,
Combining the previous two inequalities in (3.2) and (3.51), we obtain
Denote . By multiplying on the both sides of (3.53), and using (3.2), we have
where the last inequality follows since is a decreasing sequence. Denoting
Note that by the definition of in Lemma 3.4, we have
where . We then conclude from (3.2) that
On the other hand, if , then . This inequality together with the definition of then imply that . Therefore, there exists a
According to Theorem 3.2, we can show that, by specifying and as in the order of and , the gradient complexity of Algorithm 1 to obtain an -stationarity point of in the nonconvex-concave setting can be bounded by . In this setting the stepsize for updating is in the order of , while the one for updating is a constant at iteration .
Note that for solving unconstrained bilinear minimax problem, the alternating GDA algorithm with any fixed step size will cause recurrence Bailey2020 . In the classic convex optimization literature, such a divergence issue was usually handled by incorporating averaging, smoothing, or direct acceleration techniques (see, e.g., Sections 3.5-3.8 and Sections 4.3-4.5 of Lan2020book ). However, the proposed AGP algorithm for nonconvex minimax problems is not equivalent to the alternating GDA algorithm, since a regularized version of the original function is incorporated, and a variable stepsize policy has been used for the nonconvex-concave setting. Hence, the results of our paper do not contradict with existing ones.
Complexity Analysis for Convex-Nonconcave Minimax Problems
In this section, we establish the convergence of AGP algorithm for the cases where is convex w.r.t. , but possibly nonconcave w.r.t. . These are important minimax problems but the studies on their solution methods are still quite limited. Although there exists some symmetry between nonconvex-concave and convex-nonconcave minimax problems, the complexity analysis of the same AGP algorithm for nonconvex-concave setting cannot be trivially extended to that for convex-nonconcave setting.
In this subsection, we analyze the iteration complexity of Algorithm 1 for solving strongly convex-nonconcave minimax optimization problems (P), i.e., is -strongly convex w.r.t. for any fixed , and nonconcave w.r.t. for any given . Under this setting, , we set
in Algorithm 1, and simplify the update for and as follows:
Our goal in the remaining part of this subsection is to establish the iteration complexity of Algorithm 1 under the strongly convex-nonconcave setting. The convergence analysis for this setting is different from that of the nonconvex-strongly concave setting in Section 3.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 1 with parameter settings in (4.1). If , then we have
Similar to (3.5)-(3.7) in the proof of Lemma 3.1, by the optimality condition for in (4.2) implies that and ,
which, in view of the fact that is -strongly convex w.r.t. for any given , then implies that
Denoting , by Assumption 2.1, the Cauchy-Schwarz inequality and the -strongly convexity of w.r.t. , we can estimate the first inner product term in the r.h.s. of (4.1) as
Plugging (4.10) and (4.11) into (4.1) and rearranging the terms, we conclude that
By setting , in (2.7) of Lemma 2.1 and the assumption , we have
The proof is completed by combining (4.1) with (4.13).
Next, we further refine this relation in (4.1) as shown below. Note that in Lemma 3.2 we have constructed a potential function for the nonconvex-strongly concave case, whereas in the following lemma, an additional term involving the distance between two adjacent iterates, i.e., , is added to construct another potential function for the strongly convex-nonconcave case.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 1 with parameter settings in (4.1). Denote
which together with (4.9) then imply that
where the second inequality is similar to the proof of (4.10) except that for the first term in the r.h.s., we use . By using the identity , we conclude from (4.17) that
Multiplying on both sides of (4.1) and using the definition of , we obtain
The proof is completed by (4.1) in Lemma 4.1 and the definition of .
We are now ready to establish the iteration complexity for the AGP algorithm in the strongly convex-nonconcave setting.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 1 with parameter settings in (4.1). If the relations are satisfied, then , it holds that
where and with and .
On the other hand, similar to the proof of (3.1), by (4.3) and the triangle inequality, and the nonexpansiveness of the projection operator , we conclude that
By combining (4.20) and (4.21), and using the Cauchy-Schwarz inequality, we obtain
Observing that . Multiplying both sides of (4.22) by , and using (4.2) in Lemma 4.2, we have
Summing up the above inequalities from to , we obtain
Note that by the definition of in Lemma 4.2, we have
where the last inequality follows from the definitions of and , and due to the selection of . We then conclude from (4.24) that which, in view of the definition of , implies that or equivalently, .
Theorem 4.1 shows that the number of gradient evaluations performed by Algorithm 1 to obtain an -stationary point of is bounded by under the strongly convex-nonconcave setting. To the best of our knowledge, this is the first theoretical guarantee that has been obtained in the literature for solving this class of minimax problems.
2 Complexity Analysis for Convex-Nonconcave Setting
In this subsection, we analyze the iteration complexity of Algorithm 1 applied to the general convex-nonconcave setting, for which is convex w.r.t. for any fixed , and nonconcave w.r.t. for any given . Under this setting, , we set
where and are stepsize parameters to be defined later. We need to make the following assumption on the parameters .
is a nonnegative monotonically decreasing sequence.
By Assumption 2.1 and , we have
Denoting , by Assumption 4.1 and (4.26), we have
It then follows from the above inequality and the strong convexity of w.r.t. (Theorem 2.1.12 in Nestrov ) that
By using the strong convexity of w.r.t. instead of the strong concavity of w.r.t. , this inequality provides a lower bound for the inner product instead of an upper bound shown as in (3.2). This is a key inequality that we will use to establish some important recursions for the AGP algorithm under the convex-nonconcave setting in the following two results.
Similar to Lemma 4.1, we first provide an estimate on the increase of the function values from to .
Suppose that Assumption 2.1 and 4.1 hold. Let be a sequence generated by Algorithm 1 with parameter settings in (4.25). If and , then
Similar to (4.5)-(4.9) in the proof of Lemma 4.1, by replacing with , with respectively, and setting , we obtain that
Next, we prove lower bound for the four terms in the r.h.s. of (4.2) which are different from that in Lemma 4.1. By using the convexity of w.r.t instead of the concavity of w.r.t , the opposite side Cauchy-Schwarz inequality, and replacing , by , respectively, Assumptions 2.1 and 4.1, similar to the proof of (3.2)-(3.2) we conclude that
Plugging (4.2)-(4.33) into (4.2), and by the definition of and and , we conclude that
By setting , in (2.7) of Lemma 2.1 and the assumption , we have
The proof is completed by combing (4.34) and (4.35).
We need to further refine the relation in (4.3) in order to establish the convergence of the AGP algorithm as shown below. Note that in Lemma 4.2 we have constructed a potential function which involves the function value plus some distance between two adjacent iterates for the strongly convex-nonconcave case, whereas in the following lemma an additional quadratic regularization term, i.e., , is added to construct another potential function for the general convex-nonconcave case.
Suppose that Assumptions 2.1 and 4.1 hold. Let be a sequence generated by Algorithm 1 with parameter settings in (4.25). Denote
Similar to the proof of (4.16), by replacing with , we have
Using an argument similar to the proof of (4.2)-(4.33), and the Cauchy-Schwarz inequality, we conclude from the above inequality that
where . Observing that and by Assumption 4.1, we have
By and Assumption 4.1, rearranging the terms in (4.2), we obtain
By multiplying on both sides of the above inequality, we then obtain
Setting in the above inequality, and using the definition of and (4.36), we have
The proof is completed by combining (4.2) and (4.3) in Lemma 4.3, and using the definition of .
Note that the proof of Lemma 4.4 is different from that of Lemma 3.4, since different potential functions, i.e., and respectively, are used to establish the convergence of the proposed algorithms. The construction of these potential functions is the key step for our convergence analysis of AGP algorithms. We are now ready to establish the iteration complexity for the AGP algorithm to achieve an -stationary point for solving (P) under general convex-nonconcave setting.
Suppose that Assumptions 2.1 holds. Let be a sequence generated by Algorithm 1 with parameter settings in (4.25). If , , , then for any given ,
where ,
with , with .
By ,, let us denote , , it can be easily checked that the relations in (4.36) are satisfied. It follows from the selection of and that
This observation, in view of Lemma 4.4, then immediately implies that
We can easily check from the definition of that
Since both and are in the same order when becomes large enough, it then follows from the definition of that ,
Combining the previous two inequalities in (4.42) and (4.41), we obtain
Denote . By multiplying on the both sides of (4.43), and using (4.2), we get
Note that by the definition of in Lemma 4.4, we have
where . We then conclude from (4.2) that
Using the assumptions and , (4.47) and the fact , we conclude that or equivalently,
On the other hand, if , then , this inequality together with the definition of then imply that . Therefore, there exists a
Theorem 4.2 shows that the number of gradient evaluations performed by Algorithm 1 to obtain an -stationary point of which satisfies (4.36) is bounded by under the general convex-nonconcave setting. To the best of our knowledge, this is the first theoretical guarantee that has been obtained in the literature for solving this class of minimax problems. As mentioned in Section 1, it is difficult to extend existing algorithms, especially those nested-loop methods, for minimax optimization to the general convex-nonconcave setting. One possible approach would be to switch the order of the “” and “” operators. However, in general, when is nonconvex w.r.t. or nonconcave w.r.t. . Even if the set of stationary points for the above two problems are the same, e.g., by using the same stationarity criterion as in this paper, the algorithm applied to these problems will converge to different solutions. This can be seen by applying the unified AGP algorithm to solve and . The AGP algorithm will likely converge to different stationary points because different potential functions (i.e., and in Lemma 3.6 and Lemma 4.6) have to be applied for analyzing the convergence of AGP for solving these two problems.
For nonconvex-strongly concave or strongly convex-nonconcave setting, the AGP algorithm with properly chosen parameters is actually equivalent to the alternating GDA algorithm. By constructing suitable potential function, i.e., , we were able to show the iteration complexity of this method. Whereas for nonconvex-concave or convex-nonconcave setting, where or is not equal to , AGP algorithm is completely new and the selection of these parameters or plays a very crucial role to guarantee the convergence of the APG method. In these cases, the key step in our proof is also to construct a suitable potential function, i.e., and with or playing a very crucial role (see Lemma 3.6 and Lemma 4.6 respectively). To our best knowledge, this is the first time that such potential functions have been constructed for solving minimax problems.
Block-wise Nonsmooth Nonconvex Minimax Problems
In this section, we consider a more general block-wise nonsmooth minimax problem as follows.
Note that (BP) reduces to (P) if we set , and , which means that (P) is a special case of (BP). There exist very few known existing nested-loop algorithms that are proposed to solve (BP) with multi-block structure. However, the minimax problems with block structure are important in machine learning and signal processing, e.g., distributed training Lu .
The difficulty to solve (BP) comes from two aspects. One is that for any given , is nonconcave w.r.t. for any given and other blocks of , i.e., , , and . For any given , to solve the inner maximization subproblems with respect to is already NP-hard. Due to this reason, almost all the existing nested-loop algorithms will lose their theoretic guarantees since they need to solve the inner subproblem exactly, or approximately with an error proportional to the accuracy , unless we can exchange the order of “” and “” operators. However, in general, when is nonconvex w.r.t. or nonconcave w.r.t. . Although and share the same stationary point set, when we use the same algorithm to solve them, it may converge to different stationary points. Another difficulty comes from the multi-block structure. To the best of our knowledge, there are very few nested-loop algorithms (and complexity results) for solving the aforementioned multi-block structure nonconvex minimax problems before our work. On the other hand, single loop algorithms do not need to solve the inner subproblem, and can be easily generalized to handle the multi-block structure.
Similar to the idea of AGP algorithm, we propose a block alternating proximal gradient algorithm (BAPG) to solve (BP). Instead of the original function , the BAPG algorithm uses the gradient of a regularized version of the original function, i.e.,
where with regularization parameters and at the th iteration. Before presenting the detailed algorithm, we give some notations as follows. We denote two proximity operators for and as follows,
Let be the number of iteration. Denote , and define
Each iteration of the proposed BAPG algorithm conducts two proximal gradient steps for updating both and . More specifically, at the th iteration, it updates by minimizing a linearized approximation of with the gradient at point , i.e., for each ,
where is the proximal operator which is defined in (5.2) and . Similarly, it updates by maximizing a linearized approximation of minus some regularized terms, i.e.,
where is the proximal operator which is defined in (5.3) and . The proposed BAPG algorithm is formally stated in Algorithm 2, where sequences , , , and the stopping rule in Step 4 will be specified later in each of the different problem settings to be studied. Note that it reduces to the AGP algorithm when , and .
Before analyzing the convergence of Algorithm 2, we define the stationarity gap as the termination criterion as follows.
At each iteration of Algorithm 2, the stationarity gap for problem (BP) w.r.t. is defined as:
For simplicity, we denote , and .
At each iteration of Algorithm 2, the stationarity gap for problem (BP) w.r.t. , denoted by , is defined almost the same as in Definition 5.1 except by replacing with . For simplicity, we denote , and .
In this subsection, we analyze the iteration complexity of Algorithm 2 for solving the nonsmooth one-sided block-wise nonconvex-strongly concave minimax optimization problem (BP) with , i.e., is nonconvex w.r.t. for any fixed , and -strongly concave w.r.t. for any given . Under this setting, , we set
Lemma 5.1 below shows a descent result for the update.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.6). If , we have
By the optimality condition for (5.4), we have
By Assumption 2.1 and the convexity of , we obtain
Adding (5.8) and (5.1.1), and summing up the inequality from to , we have
By using the assumption that , we complete the proof.
The rest of the proof is almost the same as that of the AGP algorithm shown in Section 3 since the ascent step of ’s update can be similarly estimated. By replacing with , and using , and the fact that and , we can show similar results to those in Lemmas 3.1 and 3.2. Then, similar to the proof of Theorem 3.1, we obtain the following convergence result by some parameters replacement, e.g., with in (3.23) and with in (3.25). We omit the proof details here for simplicity.
In particular, letting be defined as in Definition 5.1 and be a given target accuracy, we provide a bound on , the first iteration index to achieve an -stationary point, i.e., , which is equivalent to
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.6). If
where , and with and .
Theorem 5.1 implies that the iteration complexity of Algorithm 2 to obtain an -stationary point for solving general block-wise nonsmooth nonconvex-strongly concave minimax problems (BP) is bounded by .
1.2 Nonconvex-Concave Setting
We analyze the iteration complexity of Algorithm 2 for solving (BP) in the nonconvex-concave setting. Under this setting, let , and , we set
where is stepsize parameter to be defined later.
is a nonnegative monotonically decreasing sequence.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.11). If , we have
The proof is the same with that of Lemma 5.1 except replacing by . We omit the details here.
The rest of the proof is almost the same as that of the AGP algorithm shown in Section 3 since the ascent step of ’s update can be similarly estimated. By replacing with , and using , and the fact that and , we can prove similar results to those in Lemmas 3.3 and 3.4. Then, similar to the proof of Theorem 3.2, we obtain the following result only through some parameters replacement, e.g., with in (3.49) and with in (3.51).
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.11). If , , , then for any given ,
where , , , , , with .
By setting , from Theorem 5.2, we conclude that for any given ,
This implies that the iteration complexity of the proposed BAPG algorithm to obtain a point that satisfies with being defined in Definition 5.1 for nonsmooth block-wise nonconvex-concave minimax problems (BP) is bounded by .
2 Complexity Analysis for (Strongly) Convex-Nonconcave Setting
In this subsection we analyze the iteration complexity of Algorithm 2 for solving (BP) with under the convex-nonconcave setting.
We first consider the strongly convex-nonconcave setting. Under this setting, , we set
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.13). If , then we have
By the optimality condition for (5.5), , we have
By the convexity of and Assumption 2.1, the gradient of is Lipschitz continuous, implying that
Adding (5.15) and (5.2), and then summing it up from to , we have
The result then follows by the assumption that .
By Lemma 5.3, similar to the proof of Lemma 4.1-4.2 and Theorem 4.1 in Subsection 4.1, we can prove the following theorem and we omit the details here.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.13). If
then , it holds that
where and , with and .
Theorem 5.3 implies that the iteration complexity of the proposed BAPG algorithm to obtain an -stationary point for smooth strongly convex-nonconcave minimax problems (BP) is bounded by .
We can also analyze the iteration complexity of Algorithm 2 applied to the general convex-nonconcave setting. Under this setting, , we set
where is stepsize parameter to be defined later.
is a nonnegative monotonically decreasing sequence.
Similar to the proof of Lemma 4.3-4.4 and Theorem 4.2 in Subsection 4.2, we can prove the following theorem.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.17). If ,, , then for any given ,
where ,
with , with , .
Theorem 5.4 implies that the iteration complexity of Algorithm 2 to obtain an -stationary point for general block-wise nonsmooth convex-nonconcave minimax problems (BP) is bounded by .
Numerical results
In this section, we compare the numerical performance of the proposed AGP algorithm with the gradient descent ascent algorithm (GDA), the alternating gradient descent ascent algorithm (AGDA) and the state-of-the-art nested-looped algorithm, i.e., MINIMAX-PPA in Lin2020 through two representative test problems. The first numerical test is implemented in Python 3.9 and run with an Apple M1 processor, while the second one is carried out on an NVIDIA Tesla P100 GPU.
The Dirac-GAN problem Mescheder2018 can be formulated as the following nonconvex-concave minimax problem :
where is the unique stationary point.
Experimental setup. Let and be the stepsize of x and y, respectively, and denote the outer-loop iteration number for AGP, GDA, AGDA and MINIMAX-PPA algorithms. The initial point of all algorithms is chosen as
2 Robust learning over multiple domains
In this subsection, we perform some numerical tests for solving a robust learning problem over multiple domains Qian , formulated as a nonconvex-linear minimax problem:
Data augmentation. Since the data in MNIST are gray images while those in CIFAR10 are RGB images, we first repeat gray channel to RGB channels and resize the examples in MNIST from to using bilinear interpolation. We adopt AlexNet Alexnet as our base model which is the same as the one used in Lu . Then, we convert to with the same method to fit the input of AlexNet.
Experiment setup. We set , , and to be , , respectively for the AGP algorithm. In the GDA algorithm, we set and . In the Heuristic Algorithm, we set the stepsize of to be 0.02, but fix as . We set the batch size in all tasks to be and run epochs for all algorithms. Moreover, we sample our results every epoch and calculate the average accuracy on these two testing sets to evaluate the performance of different algorithms.
Results. Figure 2 shows the testing accuracy on MNIST dataset. All three algorithms achieve high precision, while AGP algorithm takes less time than that of GDA. Figure 2 shows the testing accuracy on CIFAR10 dataset. AGP algorithm still performs slightly better than other two algorithms.
Table 3 shows the accuracies of all algorithms. MNIST and CIFAR10 indicate training only on MNIST and CIFAR10 dataset respectively. GDA and Heuristic Algorithm can achieve good performance while AGP algorithm can further improve the performance and provide a more reliable model for multi-task learning.
Remark. Due to hardware (especially memory) limitations, we use mini-batch randomized gradient instead of the exact gradient at each iteration in our numerical experiment, which is the same as in Lu . Since at each iteration solving the inner subproblem will take huge amounts of time in the nested loop algorithms, e.g., MINIMAX-PPA algorithm, as the gradient calculation needs to go through the whole dataset and they need to calculate gradient several times per iteration. Moreover, high requirements for precision in solving subproblems will occupy a lot of memory. Hence, we were not able to compare AGP algorithm with nested-loop algorithm for this test problem.
Conclusions and Discussion
In this paper, we propose a unified single-loop algorithm for general smooth one block nonconvex-concave or convex-nonconcave minimax problems. At each iteration, only one gradient projection step is employed for updating and respectively. The gradient complexity of the proposed unified AGP algorithm under four different settings are established. We prove that an -first order stationary point of can be obtained in (resp. ) iterations for nonconvex-concave (resp. nonconvex-strongly concave) setting. To the best of our knowledge, they are the best complexity bounds among single-loop algorithms in general nonconvex-(strongly) concave settings, and very simple to be implemented. Numerical results show the efficiency of the proposed AGP algorithm. Nonetheless, there is still a certain gap between the iteration complexity of the proposed AGP algorithm and the best complexity of among nested-looped algorithms in nonconvex-concave setting, which is also an open problem worth studying in the future.
Moreover, we consider the (strongly) convex-nonconcave setting of (P). Under this setting, the whole problem is convex. However, for any given , to solve the inner max subproblem, i.e., , is already NP-hard. Due to this reason, almost all the existing nested-loop algorithms will lose the theoretic guarantee since they all need to solve the inner subproblem exactly, or inexactly but only an error proportional to the accuracy is allowed. We show that the proposed unified single-loop AGP algorithm can deal with general convex-nonconcave setting. More specifically, the gradient complexity to obtain an -first-order stationary point of is (resp., ) under the strongly convex-nonconcave (resp., convex-nonconcave ) setting. To the best of our knowledge, these theoretical performance guarantees under this setting have not been obtained before in the literature.
Furthermore, we consider more general nonsmooth multi-block nonconvex-(strongly) concave and (strongly) convex-nonconcave minimax problems, which include smooth one block setting as a special case. We propose a block alternating proximal gradient algorithm (BAPG) to solve it. At each iteration, only simple proximal gradient steps are employed for updating one block variable, and for each block of multi-block variables alternatively. We prove the gradient complexity of the proposed BAPG algorithm under four different settings. To the best of our knowledge, these are the state-of-the-art single loop algorithms under nonconvex-concave setting and the first two theoretically guaranteed convergence results reported in the literature under (non)smooth multi-blocks (strongly) convex-nonconcave minimax problems. Our development shows that single loop algorithms do not need to solve the inner subproblem, and thus can be easily generalized. This is also the reason that the iteration complexity analysis will be more difficult than that of nested-looped algorithms. It will also be interesting to study whether the iteration complexity of the proposed AGP algorithm is tightest or not among single loop algorithms.
References
Appendix A Proof of Theorem 5.1
We prove the following two lemmas before giving the proof of Theorem 5.1.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.6). If , then we have
The optimality condition for in (5.5) implies that and ,
On the other hand, by replacing with and choosing in (A.2), we obtain
which, in view of the fact that is -strongly concave w.r.t. for any given and is convex, then implies that
Denoting , we can write the first inner product term in the r.h.s. of (A) as
Next, we estimate the three terms in the right hand side of (A) respectively. By Assumption 2.1 and the Cauchy-Schwarz inequality, we can bound the first two terms according to
For the third term, by -strongly-concavity of with respect to ,
Plugging (A)-(A.10) into (A) and rearranging the terms, we conclude that
The proof is completed by combining (A) with (5.7) in Lemma 5.1.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.6). Also let us denote
By the , and together with (A) then imply that
By plugging (A.8),(A.9),(A.15) into (A), and using the identity , we conclude that
Multiplying on both sides of (A) and using the definition of , we obtain
It then follows from (A.1) in Lemma A.1 and the definition of that
Noting that (5.4) is equivalent to , we immediately obtain
where the second inequality holds by the nonexpansiveness of the proximal operator , the last inequality hold since , by the definition of and . On the other hand, since (5.5) is equivalent to , we conclude from the triangle inequality that
By combining (A)-(A), and using Cauchy-Schwarz inequality, we obtain
Observing that . Multiplying both sides of (A.20) by , and using (A.12) in Lemma A.2, we have
Summing up the above inequalities from to and using the definition of , we obtain
Note that by the definition of in Lemma A.2, we have
where the inequality follows from the definitions of and , and the facts that () and due to the selection of . We then conclude from (A.22) that
which, in view of the definition of , implies that or equivalently, .
Appendix B Proof of Theorem 5.2
We prove the following two lemmas before giving the proof of Theorem 5.2.
Suppose that Assumptions 2.1 and 5.1 hold. Let be a sequence generated by Algorithm 2 with parameter settings in (5.11). If and , then
The optimality condition for in (5.5) implies that, , ,
On the other hand, by replacing with and choosing in (B.2), we obtain
The concavity of w.r.t. together with (B.4) and is convex, then imply that
where . We now provide bounds on the inner product terms of (B). Firstly, by the definition of and , Assumptions 2.1 and 5.1, and the Cauchy-Schwarz inequality, we have
Secondly, by the Cauchy-Schwarz inequality,
Thirdly, it follows from the concavity of w.r.t. that
Plugging (B)-(B.9) into (B), and using the definition of and and the assumption , we obtain
The result in (B.1) the follows by adding (B) and (5.12) in Lemma 5.2.
Suppose that Assumptions 2.1 and 5.1 hold. Let be a sequence generated by Algorithm 2 with parameter settings in (5.11). Also let us denote
By the , similar to (B) in Lemma B.1, (B.14) can be rewritten as
Using an argument similar to the proof of (B)-(B.9), the concavity of w.r.t. , and the Cauchy-Schwarz inequality, we conclude from the above inequality that
for any . Observing by , and Assumption 5.1, we have
Combining and rearranging the terms in (B), we obtain
By multiplying on both sides of the above inequality, we then obtain
Setting in the above inequality, and using the definition of and (B.11), we have
Combining (B) and (B.1) in Lemma B.1, and using the definition of , we conclude
By and , let us denote , , we can easily see that the relations in (B.11) are satisfied. It follows from the selection of and that
This observation, in view of Lemma B.2, then immediately implies that
We can easily check from the definition of that
By replacing with , with , similar to (A) and (A), we immediately obtain that
Combining (B.19) and (B.20), and using the Cauchy-Schwarz inequality, we have
Since both and are in the same order when becomes large enough, it then follows from the definition of that ,
Combining the previous two inequalities in (B) and (B.21), we obtain
Denote . By multiplying on the both sides of (B.23), and using (B), we have
where the last inequality follows since is a decreasing sequence. Denoting
Summing both sides of (B.24) from to , we then obtain
Note that by the definition of in Lemma B.2, we have
where . By the definition of , we then conclude from (B) that
We can see from the selection of that . Observe that is an increasing sequence, when , we have that , which implies that , by multiplying on the both sides of (B), and combining the definition of , we have , which, by the definition of , implies that
Note that when , . By using the fact and (B.27), we conclude or equivalently,
On the other hand, if , then . This inequality together with the definition of then imply that . Therefore, there exists a
such that .
Appendix C Proof of Theorem 5.3
We prove the following two lemmas before giving the proof of Theorem 5.3.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.13). If , then we have
Similar to (A.2)-(A.4) in the proof of Lemma A.1, by the optimality condition for in (5.4) implies that and ,
which, in view of the fact that is -strongly convex w.r.t. for any given and is convex, then implies that
The rest proof is the same with that of Lemma 4.1 except replacing by and by . We omit the details here.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.13). Denote
If , then ,
where , together with then imply that
The rest proof is the same with that of Lemma 4.2 except replacing by and by . We omit the details here.
Noting that (5.5) is equivalent to , we immediately obtain
where the second inequality holds by the nonexpansiveness of the proximal operator , the last inequality hold since , by the definition of and . On the other hand, since (5.4) is equivalent to , we conclude from the triangle inequality that
By combining (C)-(C.10), and using Cauchy-Schwarz inequality, we obtain
Observing that . Multiplying both sides of (C.11) by , and using (A.12) in Lemma C.2, we have
Summing up the above inequalities from to and using the definition of , we obtain
Note that by the definition of in Lemma C.2, we have
which, in view of the definition of , implies that or equivalently, .
Appendix D Proof of Theorem 5.4
We prove the following three lemmas before giving the proof of Theorem 5.4.
Suppose that Assumption 2.1 holds. Let be a sequence generated by Algorithm 2 with parameter settings in (5.17), If , we have
The proof is the same with that of Lemma 5.3 except replacing by . We omit the details here.
Suppose that Assumption 2.1 and 5.2 hold. Let be a sequence generated by Algorithm 2 with parameter settings in (5.17). If and , then
Similar to (C.2)-(C.4) in the proof of Lemma C.1, by replacing with , with , with respectively, and setting , we obtain that
The rest proof is the same with Lemma 4.3 except replacing by and by . We omit the details here.
Suppose that Assumptions 2.1 and 5.2 hold. Let be a sequence generated by Algorithm 2 with parameter settings in (5.17). Denote
Similar to (C.7) in the proof of Lemma C.2, by replacing with , with respectively, we have
The rest proof is the same with Lemma 4.4 except replacing by and by . We omit the details here.
By ,, let us denote , , it can be easily checked that the relations in (D.4) are satisfied. It follows from the selection of and that
This observation, in view of Lemma D.3, then immediately implies that
We can easily check from the definition of that
Similar to (C) and (C.10) in the proof of Theorem 5.3, by replacing with , with , with , with respectively, we conclude that
Since both and are in the same order when becomes large enough, it then follows from the definition of that ,
Combining the previous two inequalities in (D.6) and (D), we obtain
Denote . By multiplying on the both sides of (D.8), and using (D), we get
Summing both sides of (D.9) from to , we then obtain
Note that by the definition of in Lemma D.3, we have
where . By the definition of , we then conclude from (D) that
Note that . Since is an increasing sequence when , we have that , which implies that . By multiplying on the both sides of (D), and using the definition of , we have , which, by the definition of , implies that
Using the assumptions and , (D.12) and the fact , we conclude that or equivalently,
On the other hand, if , then , this inequality together with the definition of then imply that . Therefore, there exists a
such that .