Performance of first-order methods for smooth convex minimization: a novel approach
Yoel Drori, Marc Teboulle
Introduction
First-order convex optimization methods have recently gained in popularity both in theoretical optimization and in many scientific applications, such as signal and image processing, communications, machine learning, and many more. These problems are very large scale, and first-order methods, which in general involve very cheap and simple computational iterations, are often the best option to tackle such problems in a reasonable time, when moderate accuracy solutions are sufficient. For convex optimization problems, there exists an extensive literature on the development and analysis of first-order methods, and in recent years, this has been revitalized at a quick pace due to the emergence of many fundamental new applications alluded above, see e.g., the recent collections and references therein.
where stands for the gradient of . Note that the algorithm has another implicit knowledge, i.e., that the distance from its initial point to a minimizer of is bounded by some constant , see more precise definitions in the next section.
Given a desired accuracy , applying the given algorithm on the function in the class , the algorithm stops when it produces an approximate solution which is -optimal, that is such that
The performance (or complexity) of a first order black-box optimization algorithm is then measured by the number of oracle calls the algorithm needs to find such an approximate solution. Equivalently, we can measure the performance of an algorithm by looking at the absolute inaccuracy
where is the result of the algorithm after making calls to the oracle. Throughout this paper we will use the latter form to measure the performance of a given algorithm.
Building on this model, in this work we introduce a novel approach for analyzing the performance of a given first order scheme. Our approach relies on the observation that by definition, the worst case behavior of a first-order black-box optimization algorithm is by itself an optimization problem which consists of finding the maximal absolute inaccuracy over all possible inputs to the algorithm. Thus, with being the output of the algorithm after making calls to the oracle, we look at the solution of the following Performance Estimation Problem (PEP):
At first glance this problem seems very hard or impossible to solve. We overcome this difficulty through an analysis that relies on various types of relaxations, including duality and semi-definite relaxation techniques. The problem and setting, and an outline of the underlying idea of the proposed approach for analyzing (P) are described in Section 2. In order to develop the basic idea and tools underlying our proposed approach, we first focus on the fundamental gradient method (GM) for smooth convex minimization, and then extend it to a broader class of first order black box minimization methods. Obviously, the gradient method is a particular case of this broader class that will be analyzed below. However, it is quite important to start with the gradient method for two reasons. First, it allows to acquaint the reader in a more transparent way with the techniques and methodolgy we need to develop in order to analyze (PEP), thus paving the way to tackle more general schemes. Secondly, for the gradient method, we are able to prove a new and tight bound on its performance which is given analytically, see Section 3. Capitalizing on the methodology and tools developed in the past section, in Section 4, we consider a broader class of first-order black-box methods, which among others, is shown to include the so-called heavy-ball and fast gradient schemes . Although an analytical solution is not available for this general case, we show that for this broader class of methods, it is always possible to compute numerical bounds for an adequate relaxation of the corresponding PEP, allowing to derive new bounds on the performance of these methods. We then derive in Section 5 an efficient procedure for finding optimal step sizes which results in a first-order method that achieves best performance. Our approach and analysis give rise to some interesting problems leading us to suggest some conjectures. Finally, an appendix includes the proof of a technical result.
The Problem and the Main Approach
Let be a first-order algorithm for solving the optimization problem
Throughout the paper we make the following assumptions:
We assume that (M) is solvable, i.e., the optimal set is nonempty, and for we set
There exists , such that the distance from to an optimal solution is bounded by .In general, the terms and are unknown or difficult to compute, in which case some upper bound estimates can be used in place. Note that all currently known complexity results for first order methods depend on and .
2 Basic Idea and Main Approach
We are interested in measuring the worst-case behavior of a given algorithm in terms of the absolute inaccuracy , by solving problem (P) defined in the introduction, namely
To tackle this problem, we suggest to perform a series of relaxations thereby reaching a tractable optimization problem.
An un-formal description of the underlying idea consists of two main steps as follows:
To define constraints that relate the new variables, we use relevant/useful properties characterizing the family of convex functions in , as well as the rule(s) describing the given algorithm .
This approach can, in principle, be applied to any optimization algorithm. Note that any relaxation performed on the maximization problem (P) may increase its optimal value, however, the optimal value of the relaxed problem still remains a valid upper bound on .
A formal description on how this approach can be applied to the gradient method is described in the next section, which as we shall see, allows us to derive a new tight bound on the performance of the gradient method.
An Analytical Bound for the Gradient Method
To develop the basic idea and tools underlying the proposed approach for analyzing the performance of iterative optimization algorithms, in this section we focus on the simplest fundamental method for smooth convex minimization, the Gradient Method (GM). It will also pave the way to tackle more general first-order schemes as developed in the forthcoming sections.
Consider the gradient algorithm with constant step size, as applied to problem , which generates a sequence of points as follows:
Here is fixed. At this point, we recall that for , the convergence rate of the GM algorithm can be shown to be (see for example ):
To begin our analysis, we first recall a fundamental well-known property for the class of convex functions, see e.g., [14, Theorem 2.1.5].
and note that we always have and .
In terms of , , condition (3.3) becomes
The obtained problem remains nontrivial to tackle. We will now perform some simplifications on this problem that will be useful for the forthcoming analysis.
Secondly, we consider (3.4) for the four cases , , and , and use the equality constraints
to eliminate the variables . After some algebra, we reach the following form for the PEP:
where is a shorthand notation for .
Finally, we note that the optimal solution for this problem is attained when , and hence we can also eliminate the variables and . This produces the following PEP for the gradient method, a nonconvex quadratic minimization problem:
Therefore, by defining the following symmetric matrices
Problem (G) is a nonhomogeneous Quadratic Matrix Program, a class of problems recently introduced and studied by Beck .
2 A Tight Performance Estimate for the Gradient Method
We now proceed to establish the two main results of this section. First, we derive an upper bound on the performance of the gradient method, this is accomplished via using duality arguments. Then, we show that this bound can actually be attained by applying the gradient method on a specific convex function in the class .
In order to simplify the following analysis, we will remove some constraints from (G) and consider the bound produced by the following relaxed problem:
As we shall show below, it turns out that this additional relaxation has no damaging effects and produces the desired performance bound when .
To establish our dual result, the next lemma shows that a dimension reduction is possible when minimizing a quadratic matrix function sharing the special form as the one that appears in problem (G′).
and using (3.7) it follows . Now, using (3.6)-(3.7), one obtains and , and hence it follows that
Equipped with Lemma 3.1, we now derive a Lagrangian dual for problem (G′).
For convenience, we recast (G′) as a minimization problem, and we also omit the fixed term from the objective. That is, we consider the equivalent problem (G′′) defined by
The dual objective function is then defined by
and the dual problem of (G′′) is then given by
Since is linear in , we have whenever
and otherwise. Invoking Lemma 3.1, we get
Therefore for any satisfying (3.10), we have obtained that the dual objective reduces to
Now, recalling the definition of the matrices (see (3.5)), we obtain
Finally, using the relations (3.10) to eliminate , and recalling that was defined as , the desired form of the stated dual problem follows. ∎
The next lemma will be crucial in invoking duality in the forthcoming theorem. The proof for this lemma is quite technical and appears in the appendix.
We are now ready to establish a new upper bound on the complexity of the gradient method for values of between 0 and 1. To the best of our knowledge, the tightest bound thus far is given by (3.1).
First note that both (G) and (G′) are clearly feasible and . Invoking Lemma 3.2, by weak duality for the pair of primal-dual problems (G′) and (DG′), we thus obtain that and hence:
Now consider the following point for the dual problem (DG′):
Assuming that this point is (DG′)-feasible, it follows from (3.12) that
which proves the desired result. Thus, it remains to show that the above given choice is feasible for (DG′). First, it is easy to see that all the linear constraints of (DG′) on the variables described through the set hold true. Now we prove that the matrix is positive semidefinite. From Lemma 3.3, with , we get that is positive definite, as a convex combination of positive definite matrices. Next, we argue that the determinant of is zero. Indeed, take , then from the definition of and the choice of and it follows by elementary algebra that . To complete the argument, we note that the determinant of can also be found via the identity (see, e.g., [7, Section A.5.5]):
Since we have just shown that , then and we get from the above identity that the value of , which is the Schur complement of the matrix , is equal to 0. By a well known lemma on the Schur complement [6, Lemma 4.2.1], we conclude that is positive semidefinite. ∎
The next theorem gives a lower bound on the complexity of Algorithm GM for all values of . In particular, it shows that the bound (3.11) is tight and that it can be attained by a specific convex function in .
We will describe two functions that attain the two parts of the expression in the above claimed statement. For the sake of simplicity we will assume that and . Generalizing this proof to general values of and can be done by an appropriate scaling.
To show the first part of the expression, consider the function
To show the second part of the expression, we apply the gradient method on
with . We then get that for :
Numerical experiments we have performed on problem (G) suggest that the lower complexity bound given by Theorem 3.2 is in fact the exact complexity bound on the gradient method with .
Suppose the sequence is generated by the gradient method GM with , then
We now turn our attention to the problem of choosing the step size, . Assuming the above conjecture holds true, the optimal step size (i.e., the step size that produces the best complexity bound) for the gradient method with constant step size is given by the unique positive solution to the equation
The solution of this equation approaches as grows. Therefore, assuming the conjecture holds true and is large enough, the complexity of the gradient method with the optimal step size approaches . This represents a significant improvement, by the factor of 4, upon the best known bound on the gradient method (3.1), and also supports the observation that the gradient method often performs better in practice than in theory.
A Class of First-Order Methods: Numerical Bounds
The interest in the analysis of first-order algorithms of this type is motivated by the fact that it covers some fundamental first-order schemes beyond the gradient method. In particular, to motivate (FO), let us consider the following two algorithms which are of particular interest, and as we shall see below can be seen as special cases of (FO).
We start with the so-called Heavy Ball Method (HBM). For earlier work on this method see Polyak , and for some interesting modern developments and applications, we refer the reader to Attouch et al. and references therein.
Here the step sizes and are chosen such that and , see .
By recursively eliminating the term in step 2 of HBM, we can rewrite this step as
Therefore, the heavy ball method is a special case of Algorithm FO with the choice
The next algorithm is Nesterov’s celebrated Fast Gradient Method .
A major breakthrough was achieved by Nesterov in , where he proved that the FGM, which requires almost no increase in computational effort when compared to the basic gradient scheme, achieves the improved rate of convergence for function values. More precisely, one hasThe bound given here, which improves the original bound derived in , was recently obtained in Beck-Teboulle .
This fundamental algorithm discovered about 30 years ago by Nesterov has been recently revived and is currently subject of intensive research activities. For some of its extensions and many applications, see e.g., the recent survey paper Beck-Teboulle and references therein.
At first glance, Algorithm FGM seems different than the scheme FO defined above. Here two sequences are defined: the main sequence and an auxiliary sequence . Observing that the gradient of the function is only evaluated on the auxiliary sequence of points , we show in the next proposition that FGM fits in the scheme FO through the following algorithm:
The points generated by Algorithm FGM′ are identical to the respective points generated by Algorithm FGM.
We will show by induction that the sequence generated by Algorithm FGM′ is identical to the sequence generated by Algorithm FGM, and that the value of generated by Algorithm FGM′ is equal to the value of generated by Algorithm FGM.
First note that the sequence is defined by the two algorithms in the same way. Now let , be the sequences generated by FGM and denote by , the sequence generated by FGM′. Obviously, and since we get using the relations 4.2:
Assuming for , we then have
2 A Numerical Bound for FO
To build the performance estimation problem for Algorithm FO, from which a complexity bound can be derived, we follow the approach used to derive problem (G) for the gradient method. The only difference being that here, of course, the relation between the variables is derived from the main iteration of Algorithm FO. After some algebra, the resulting PEP for the class of algorithms (FO) reads
In view of the difficulties in the analysis required to find the solution of (G), an analytical solution to this more general case seems unlikely. However, as we now proceed to show, we can derive a numerical bound on this problem that can be efficiently computed.
Following the analysis of the gradient method, (cf.(G′) in §3.2) we consider the following simpler relaxed problem:
With the same proof as given in Lemma 3.2 for problem (Q′), we obtain that a dual problem for (Q′) is given by the following convex semidefinite optimization problem:
Note that the data matrices of both primal-dual problems (Q′) and (DQ′) depend on the step-sizes . To avoid a trivial bound for problem , here we need the following assumption on the dual problem (DQ′):
Assumption 1 Problem (DQ′) is solvable, i.e., the minimum is finite and attained for the given step-sizes .
Actually, the attainment requirement can be avoided if we can exhibit a feasible point for the problem (DQ′). As noted earlier, given the difficulties already encountered for the simpler gradient method, finding explicitly such a point for the general class of algorithms (FO) is unlikely. However, the structure of problem (DQ′) will be very helpful in the analysis of the next section which further addresses the role of the step-sizes.
The promised numerical bound now easily follows showing that a complexity bound for (FO) is determined by the optimal value of the dual problem (DQ′) which can be computed efficiently by any numerical algorithm for SDP (see e.g., ).
Follows from weak duality for the pair of primal-dual problems (Q′)-(DQ′) ∎
3 Numerical Illustrations
We apply Proposition 4.2 to find bounds on the complexity of the heavy ball method (HBM) withAccording to our simulations, this choice for the values of produce results that are typical of the behavior of the algorithm. and and on the fast gradient method (FGM) with as given in (4.2), which as shown earlier, can both be viewed as particular realizations of (FO).
The resulting SDP programs were solved for different values of using CVX . These results, together with the classical bound on the convergence rate of the main sequence of the fast gradient method (4.1), are summarized in Figures 1 and 2.
Note that as far as the authors are aware, there is no known convergence rate result for the HBM on the class of convex functions in . As can be seen from the above results, the numerical bound for HBM behaves slightly better than the gradient method (compare with the explicit bound given in Theorem 3.1), but remains much slower than the fast gradient scheme (FGM).
Considering the results on the FGM, note that the numerical bounds for the main sequence of point and the corresponding values at the auxiliary sequence of the fast gradient method are very similar and perform slightly better than predicted by the classical bound (4.1). To the best of our knowledge, the complexity of the auxiliary sequence is yet unknown, thus these results encourage us to raise the following conjecture.
Let and be the main and auxiliary sequences defined by FGM (respectively), then and converge to the optimal value of the problem with the same rate of convergence.
A Best Performing Algorithm: Optimal Step Sizes for The Algorithm Class FO
We now consider the problem of finding the “best” performing algorithm of the form FO with respect to the new bounds. Namely, we consider the problem of minimizing , the optimal value of (Q′), with respect to the step sizes defining the algorithm FO, and which are now considered as unknown variables in FO.
Note that the feasibility of (BIL) follows from the proof of Theorem 3.2, where an explicit feasible point is given to (DG′), which is a special instance of (BIL) when the steps are chosen as in the gradient method.
and denoting , we obtain the following linear SDP relaxation of (BIL):
This convex SDP can now be efficiently solved by numerical methods. As the following theorem shows, its solution can be used to construct a solution for (BIL) with optimal step sizes .
Suppose is an optimal solution for (LIN), then is an optimal solution for (BIL), where is defined by the following recursive rule
As (LIN) is a relaxation of (BIL), it is enough to show that (BIL) can achieve the same objective value. Let be an optimal solution for (LIN). If for all , then (5.2) satisfies all the equations in (5.1) and therefore is feasible for (BIL).
Suppose for some and that is the maximal index with this property. Then by the equality and non-negativity constraints in (LIN), we get that and . Let , then by the positive semidefinite constraint in (LIN), we have . From the linear equalities connecting and it follows that
and we get that . By the properties of positive semidefinite matrices we now get that for and , hence the set of equations (5.1) with the chosen values of is consistent. ∎
The optimal value of (LIN) for various values of is summarized in Figure 3. The resulting new algorithm with the computed optimal step sizes is illustrated for and given in Figure 4. As can be seen from these results, (compare with Figure 2) the performance of the new algorithm is almost exactly two times better than the performance of the fast gradient method.
Acknowledgements
This work was initiated during our participation to the “Modern Trends in Optimization and Its Application” program at IPAM, (UCLA), September-December 2010. We would like to thank IPAM for their support and for the very pleasant and stimulating environment provided to us during our stay. We would also like to thank Simi Haber, Ido Ben-Eliezer and Rani Hod for their help in the proof of Lemma 3.3.
Appendix A Proof of Lemma 3.3
We now establish the positive definiteness of the matrices and given in (3.8) and (3.9), respectively.
We begin by showing that is positive definite. Recall that
Let us look at for any :
which is always positive for . We conclude that is positive definite.
We will show that is positive definite using Sylvester’s criterionDespite the interesting structure of the matrix , this proof is quite involved. A simpler proof would be most welcome!.
We begin by deriving a recursion rule for the determinant of matrices of the following form:
To find the determinant of , subtract the one before last row multiplied by from the last row: the last row becomes
Expanding the determinant along the last row we get
where denotes the minor:
If we multiply the last column of by we get a matrix that is different from by only the corner element. Thus by basic determinant properties we get that
Combining these two results, we have found the following recursion rule for , :
Obviously, the recursion base cases are given by
Closed form expressions for the determinants
Going back to our matrix, , by choosing
we get that is the ’th leading principal minor of the matrix . The recursion rule (A.1) can now be solved for this choice of and . The solution is given by:
Verification
We now proceed to verify the expressions (A.2) and (A.3) given above. We will show that these expressions satisfy the recursion rule (A.1) and the base cases of the problem. We begin by verifying the base cases:
then the recursion rule (A.1) can be written as
Substituting (A.2) in the RHS of (A.1) we get that for
It is straightforward (although somewhat involved) to verify that for
and thus (A.2) satisfies (A.1). It is also possible to show that
and the expression (A.3) is also verified.
To complete the proof, note that the closed form expressions for consist of sums and products of positive values, hence is positive, and thus follows from Sylvester’s criterion that is positive definite.