Last-Iterate Convergence: Zero-Sum Games and Constrained Min-Max Optimization
Constantinos Daskalakis, Ioannis Panageas
Introduction
A central problem in Game Theory and Optimization is computing a pair of probability vectors , solving
and that all solutions to the LHS are solutions to the RHS, and vice versa. This result was a founding stone in the development of Game Theory. Indeed, interpreting as the payment of the “min player” to the “max player” when the former selects a distribution over columns and the latter selects a distribution over rows of matrix , a solution to (1) constitutes an equilibrium of the game defined by matrix , called a “minimax equilibrium”, a pair of randomized strategies such that neither player can improve their payoff by unilaterally changing their distribution.
Besides their fundamental value for Game Theory, it is known that (1) and (2) are also intimately related to Linear Programming. It was shown by von Neumann that (2) follows from strong linear programming duality. Moreover, it was suggested by Dantzig and recently proven by Adler that any linear program can be solved by solving some min-max problem of the form (1). In particular, min-max problems of form (1) are exactly as expressive as min-max problems of the following form, which capture any linear program (by Lagrangifying the constraints):
Soon after the minimax theorem was proven and its connection to linear programming was forged, researchers proposed dynamics for solving min-max optimization problems by having the min and max players of (1) run a simple learning procedure in tandem. An early method, proposed by Brown and analyzed by Robinson , was fictitious play. Soon after, Blackwell’s approachability theorem propelled the field of online learning, which lead to the discovery of several learning algorithms converging to minimax equilibrium at faster rates, while also being robust to adversarial environments, situations where one of the players of the game deviates from the prescribed dynamics; see e.g. . These learning methods, called “no-regret”, include the celebrated multiplicative-weights-update method, follow-the-regularized-leader, and follow-the-perturbed-leader. Compared to centralized linear programming procedures the advantage of these methods is the simplicity of executing their steps, and their robustness to adversarial environments, as we just discussed.
Despite the extensive literature on no-regret learning, an unsatisfactory feature of known results is that min-max equilibrium is shown to be attained only in an average sense. To be precise, if is the trajectory of a no-regret learning method, it is usually shown that the average converges to the optimal value of (1), as . Moreover, if the solution to (1) is unique, then converges to the optimal solution. Unfortunately that does not mean that the last iterate converges to an optimal solution, and indeed it commonly diverges or enters a limit cycle. Furthermore, in the optimization literature, Nesterov provides a method that can give pointwise convergence (i.e., convergence of the last iterate) to problem (1)Nesterov showed that by optimizing for yields an approximation to the problem problem (1)., however his algorithm is not a no-regret learning algorithm. Recent work by Daskalakis et al and Liang and Stokes studies whether last iterate convergence can be established for no-regret learning methods in the simple unconstrained min-max problem of the form:
Motivated by the afore-described lines of work, and the importance of last iterate convergence for Game Theory and the modern applications of GDA-style methods in Optimization, our goal in this work is to generalize the results of to the general min-max problem (3), or equivalently (1); indeed, we will focus on the latter, but our algorithms are readily applicable to the former as the two problems are equivalent . With the constraint that should remain in , GDA and OGDA are not applicable. Indeed, the natural GDA-style method for min-max problems in this case is the celebrated Multiplicative-Weights-Update (MWU) method, which is tantamount to FTRL with entropy-regularization. Unsurprisingly, in the same way that GDA suffers in the unconstrained problem (4), MWU exhibits cycling in the constrained problem (1) (a recent work is and was also shown empirically in ). So it is natural for us to study instead its optimistic variant, “Optimistic Multiplicative-Weights-Update (OMWU),” (called Optimistic Hedge in ) which corresponds to Optimistic FTRL with entropy-regularization, the equations of which are given in Section 2.2. Our main result is the following (restated as Theorem 2.7 after Section 2.2) and answers an open question asked in as applicable to two player zero sum games:
Whenever (1) has a unique optimal solution , OMWU with appropriate choice of learning rate and initialized at the pair of uniform distributions exhibits last-iterate convergence to the optimal solution. That is, if are the vectors maintained by OMWU at step , then .
We note that the assumption about uniqueness of the optimal solution for problem (1) is generic in the following sense: Within the set of all zero-sum games, the set of zero-sum games with non-unique equilibrium has Lebesgue measure zero . This implies that if ’s entries are sampled independently from some continuous distribution, then with probability one the min-max problem (1) will have a unique solution.
Our paper provides two important messages:
It strengthens the intuition that optimism helps the trajectories of learning dynamics stabilize (e.g., Optimistic MWU vs MWU or Optimistic GDA vs GDA; as the papers of Syrgkanis et al and Daskalakis et al also do).
The techniques we use (typically appear in dynamical systems literature) to prove convergence for the last iterate, are fundamentally different from those commonly used to prove convergence of the time average of a learning algorithm.
Preliminaries
A recurrence relation of the form is a discrete time dynamical system, with update rule where for our purposes. The point is called a fixed point or equilibrium of if . We will be interested in the following well known fact that will be used in our proofs.
If the Jacobian of the update rule We assume is a continuously differential function. at a fixed point has spectral radius less than one, then there exists a neighborhood around such that for all , the dynamics converges to , i.e., . We call an asymptotic stable mapping in .
2 OMWU Method
Our main contribution is that the last iterate of OMWU converges to the optimal solution. The OMWU dynamics is defined as follows ():
Points are the initial conditions and are given as input. We call the stepsize of the dynamics. It is more convenient to interpret OMWU dynamics as mapping a quadruple to quadruple (, see Section 3.2 for the construction of the dynamical system).
Let be the optimal solution. We see that is a fixed point of the mapping. Furthermore, is invariant under OMWU dynamics. For , if then remains zero for all times greater than , and if it is positive, it remains positive (both numerator and denominator are positive) Same holds for vector .. In words, at all times the OMWU satisfies the non-negativity constraints and the renormalization factor (denominator) makes both ’s coordinates sum up to one. A last observation is that every fixed point of OMWU dynamics (mapping a quadruple to quadruple) has the form (two same copies). Equation (8) shows how to express OMWU dynamics as a dynamical system.
3 Linear Variant of OMWU
We provide the linear variant of OMWU dynamics (5) because we use it in some intermediate lemmas (appear in appendix).
This dynamics is derived by considering the first order approximation of the exponential function. Stepsize in this case should be chosen sufficiently small so that both numerator and denominator are positive.
4 More definitions and statement of our result
Assume . We call a point -close if for each we have that or and for each it holds or .
Think of -close points as -approximate optimal solutions for min-max problems that are induced by submatrices of (-approximate stationary points). Moreover, if is -close point does not necessarily imply is the optimal solution of problem (1)!
Think of -approximate points as approximate optimal solutions to the min-max problem (1). Moreover, if is -approximate then is the optimal solution of problem (1).
We finish the preliminary section by stating formally the main result.
Let be a matrix and assume that
has a unique solution . It holds that for sufficiently small (depends on ), starting from the uniform distribution, i.e., , it holds
under OMWU dynamics. The stepsize is constant, i.e., does not scale with timeOur proof also works if the starting points are both in the interior of and not necessarily uniform, however the choice of depends on the initial distributions as well and not only on ..
We need to note that it is not clear from our theorem how small is and its dependence on the size of . Moreover, OMWU has two phases (the phase where KL divergence decreases and the local asymptotic stability phase, see theorems below) where the stepsize is constant but it might change when we move from phase one to phase two. Nevertheless, our convergence result holds for constant stepsizes as opposed to the classic no-regret learning literature where scales like after iterations. Another result we know of this flavor is about MWU algorithm on congestion games .
Last iterate convergence of OMWU
In this section we show our main result (Theorem 2.7), by breaking the proof into three key theorems. The first theorem says that KL divergence from the -th iterate to the optimal solution , i.e., (sum of KL divergences to be exact)
decreases with time by at least a factor of per iteration, unless the iterate is -close (see Definition 2.3). Moreover, provided that the stepsize is small enough, we can show the structural result that lies in a neighborhood of that becomes smaller and smaller as . Finally, as long as OMWU dynamics has reached a small neighborhood around , we show that the update rule of the dynamical system induced by OMWU is locally (asymptotically) stable (for maybe different choice of learning rate), and the last iterate convergence result follows. Formally we show:
Let be the unique optimal solution of problem (1) and sufficiently small. Then
is decreasing with time by (at least) unless is -close.
Assume that is unique optimal solution of the problem (1). Let (depends on ) be the first time KL divergence does not decrease by . It follows that as , the -close point has distance from that goes to zero, i.e., .
Let be the unique optimal solution to the min-max problem (1). There exists a neighborhood of Since might be on the boundary of , is the intersection of an open ball around with . so that for all we have that under OMWU dynamics as defined in (5) and (8) (Section 3.2).
Assuming these three theorems, our main result is straightforward.
In the next subsections we will provide the proofs to all three key theorems.
In this subsection we argue about the proofs of Theorems 3.1 and 3.2. The inequality we managed to prove (see in the appendix the proof of Theorem 3.1) is the following:
The proof of the inequality is quite long, we choose to provide intuition and skip the details. We refer to the appendix for a proof. The inequality says that OMWU dynamics has a good progress (KL divergence decreases by at least a factor of ) as long as the current and previous iterate are not -close for chosen to be . This situation appears a lot in gradient methods when the dynamics is close to a stationary point, the gradient of is small and the progress is small as opposed to the case where the gradient of is big and there is satisfying progress. The RHS of inequality (7) captures the “distance” from stationarity. Thus, as long as we are not close to a stationary point (i.e., -close) in a time window between 1,2,…,k, KL divergence from current iterate (-th) to the optimum has decreased by (at least) compared to KL divergence from first iterate to the optimum.
Moreover, suppose that at some point of OMWU dynamics, KL divergence from current iterate to the optimum did not decrease by at least a factor of and let be the iteration this happened. As we have already argued, is a -close point. We can show that as long as is sufficiently small, then for all in the support of , are (at least) i.e., coordinates in the support of the optimum will have non negligible probability in . Formally:
Let and . It holds that and as long as
By definition of , the KL divergence is decreasing for , thus
Therefore . It follows that for (). Since is (Lemma B.1) the result follows. Similarly, the argument works for . ∎
Lemma 3.4 indicates that the stepsize might have to be exponentially small in the dimension (OMWU dynamics is slow when is very small). We can now prove Theorem 3.2.
From Lemma 3.4 and definition of , we get that is for all in the support of and is for all in the support of . We consider to be the projection of the point by removing all the coordinates that have probability mass less than and rescale so that the coordinates sum up to one.
We restrict ourselves to the corresponding subproblem (submatrix). It is clear that is a -approximate solution By -approximate optimal solution we mean the -approximate Nash equilibrium notion (additive), see Definition 2.5. for the subproblem. Let be the optimal value. By uniqueness of the optimal solution, we get that for all and otherwise (check Lemma C.3 in paper for a proof, where they use Farkas’ lemma to show it, we use this fact later in Section 3.2). Similarly for the min player if lies in the support of and otherwise. We choose so small that every -approximate solution has the property that , for all and respectively (this is possible by continuity of the bilinear function and Claim 3.5 below). Hence we conclude that if is small enough, the coordinates in the vector that are not in the support of the optimal solution (since ), should have probability mass at time .
Let be the unique optimal solution to the problem (1). For every , there exists an so that for every -approximate solution we get that for all . Analogously holds for player .
We will prove this by contradiction. Assume there is an that violates this statement. We choose a sequence so that and also there is a sequence of -approximate Nash equilibrium with for some strategy . Since is compact and the sequence above is bounded, there is a convergent subsequence. The limit of the convergent subsequence is a Nash equilibrium by definition of -approximate (Definition 2.5). By uniqueness it follows that the -th coordinate of the convergent sequence must converge to , hence we reached a contradiction. ∎
Therefore, if we restrict to the subproblem induced by the strategies in the support of , the projected vector is a -approximate solution of the subgame.
2 Proving local convergence
The purpose of this section is to prove Theorem 3.3. First of all, we assume that the stepsize is some fixed constant (sufficiently small, not necessarily the same stepsize as in the first phase where KL divergence decreases). To show asymptotic stability of OMWU dynamics in a neighborhood of the optimal solution , we first construct a dynamical system that captures OMWU. Moreover, we prove that the Jacobian of the update rule of that particular dynamical system computed at the optimal solution, has spectral radius less than one. This suffices to prove asymptotic stability (see Proposition 2.1). As a result, as long as OMWU reaches a small neighborhood of , it converges pointwise (last iterate convergence) to itSince the dynamical system is from a quadruple to a quadruple, it is a neighborhood of .. Below we provide the update rule of the dynamical system, which consists of 4 components:
so captures exactly the dynamics of OMWU (5). The equations of the Jacobian of can be found in the appendix (see Section A).
The rest of the section constitutes the proof of Theorem 3.3. Assume , i.e., is the value of the bilinear function at the optimal solution. We will analyze the Jacobian computed at See also Equations (14) of the Jacobian computed at ..
Assume , then
and all other partial derivatives of are zero, thus is an eigenvalue of the Jacobian computed at . Moreover because of uniqueness of the optimal solution, it holds that because (check Lemma C.3 in for a proof, where they use Farkas’ Lemma to show it). Similarly, it holds for that (again by C.3 in it holds that ) and all other partial derivatives of are zero, hence is an eigenvalue of the Jacobian computed at the optimal solution.
Let be the characteristic polynomial of the matrix (10). After row/column operations it boils down to
where is the characteristic polynomial of
Experiments
For the latter case, we fix and we consider the error to be . Starting from uniform distribution, we count the number of iterations to reach error . The stepsize is fixed at at all times. The results can be found in the figure below (Figure 4). If we had to guess, it seems that the relation between dimension and iterations is between linear and quadratic (i.e., OMWU dynamics has roughly cubic-quartic running time in if we count the cost of each iteration as quadratic) and the dependence between error and iterations seems like is inverse polynomial in .
We note the importance of stepsize . must be sufficiently small for our proofs to work. If is chosen to be big, then OMWU might not converge (might cycle, we observed such behavior in experiments). On the other hand, the smaller is chosen, the smaller the progress of OMWU dynamics (see the inequality claim for KL divergence) and hence the slower the dynamics.
Conclusion
In this paper we showed that a no-regret algorithm called Optimistic Multiplicative Weights Update (OMWU) converges pointwise to a Nash equilibrium in two player zero sum games (See also a concurrent work to ours , in which the authors provide a pointwise result about other dynamics, using different techniques). Our analysis is novel and does not follow the standard approaches of the literature of no-regret learning. We believe that our techniques can be useful in the analysis of other learning algorithms with no provable guarantees of pointwise convergence.
One interesting open question is to show that OMWU algorithm converges in polynomial time in (for proper choice of stepsize ) and find exact rates of convergence. Another possible future direction is to generalize our results about OMWU beyond the bilinear setting.
References
Appendix A Equations of the Jacobian of OMWU dynamics
Set , and let be arbitrary indexes ( captures the -th coordinate of function etc),
Set , and let be arbitrary indexes ( captures the -th coordinate of function etc). Assume , it is not hard to see that for all and . We get that:
Appendix B Missing claims and proofs
Lemma B.1 shows that the change between next and current iterate in both OMWU algorithms (classic and linear variant) is of order and that the difference between the next iterate of both algorithms is .
Let be the vector of the max player, and suppose are the next iterates of OMWU and its linear variant with current vector and vectors of the min player. It holds that
Analogously, it holds for vector of the min player and its next iterates.
Let be sufficiently small (smaller than maximum in absolute value entry of ).
and hence is . Moreover we have that
By triangle inequality and the two above proofs we get the third part of the lemma. ∎
Lemmas B.2, B.3 and B.5 will be used in the proof of Theorem 3.1.
Let , and suppose are the next iterates of OMWU and its linear variant with current vector and inputs , i.e., has coordinates and has coordinates . It holds that (for sufficiently small)
It suffices to prove the second equality. The rest follow from Lemma B.1. Set . We have that (from definition of linear variant of OMWU dynamics). It follows that
Using same arguments as in proof of Lemma B.2 we have the following lemma:
Let , and suppose is the next iterate of OMWU with current vector and inputs , i.e., has coordinates . It holds that (for sufficiently small)
Let be the -th iterate of OMWU dynamics (5). For each time step it holds that
The second equality comes from Lemmas B.2 and B.3. By canceling out the common terms and bring to the LHS the appropriate remaining terms, the claim follows. ∎
Let denote the -th iterate of OMWU dynamics. It holds for that
where is the optimal solution of the min-max problem.
It is true that , hence for sufficiently small. Therefore lies in the simplex . Hence since is the optimum (Nash equilibrium) we get that ( is the max player). Similarly the second inequality can be proved. ∎
We compute the difference between and
We use Lemma B.5 and we get that , therefore the LHS (difference in the KL divergence) is at most
We furthermore use second order Taylor approximation ( is sufficiently small) to the function and we get that previous expression is at most
Finally, using Taylor approximation on and Lemma B.4 (last equality) we get the following system:
It is clear that as long as (and thus by Lemma B.1) is not -close, from above inequalities/equalities we get
meaning that KL divergence decreases by at least a factor of and the claim follows. ∎
Let be a real diagonal matrix with positive diagonal entries and be a real skew-symmetric matrix (). It holds that has eigenvalues with real part zero (i.e., it has only imaginary eigenvalues).
Let be the conjugate transpose of and be a left eigenvector of with complex eigenvalue . It holds that
Since has positive diagonal entries, we conclude that (since ), thus and the claim follows. ∎