Convex Optimization: Algorithms and Complexity
Sébastien Bubeck
Chapter 1 Introduction
We are interested in algorithms that take as input a convex set and a convex function and output an approximate minimum of over . We write compactly the problem of finding the minimum of over as
In the following we will make more precise how the set of constraints and the objective function are specified to the algorithm. Before that we proceed to give a few important examples of convex optimization problems in machine learning.
Many fundamental convex optimization problems in machine learning take the following form:
In classification one has . Taking (the so-called hinge loss) and one obtains the SVM problem. On the other hand taking (the logistic loss) and again one obtains the (regularized) logistic regression problem.
Our last two examples are of a slightly different flavor. In particular the design variable is now best viewed as a matrix, and thus we denote it by a capital letter . The sparse inverse covariance estimation problem can be written as follows, given some empirical covariance matrix ,
Intuitively the above problem is simply a regularized maximum likelihood estimator (under a Gaussian assumption).
Finally we introduce the convex version of the matrix completion problem. Here our data set consists of observations of some of the entries of an unknown matrix , and we want to “complete" the unobserved entries of in such a way that the resulting matrix is “simple" (in the sense that it has low rank). After some massaging (see Candès and Recht (2009)) the (convex) matrix completion problem can be formulated as follows:
where and are given.
2 Basic properties of convexity
A basic result about convex sets that we shall use extensively is the Separation Theorem.
Note that if is not closed then one can only guarantee that (and ). This immediately implies the Supporting Hyperplane Theorem ( denotes the boundary of , that is the closure without the interior):
We introduce now the key notion of subgradients.
The set of subgradients of at is denoted .
To put it differently, for any and , is above the linear function . The next result shows (essentially) that a convex functions always admit subgradients.
It is obvious that a function is convex if and only if its epigraph is a convex set.
The first claim is almost trivial: let , then by definition one has
which clearly shows that is convex by adding the two (appropriately rescaled) inequalities.
Clearly, by letting tend to infinity, one can see that . Now let us assume that is in the interior of . Then for small enough, , which implies that cannot be equal to (recall that if then necessarily which allows to conclude by contradiction). Thus rewriting (1.2) for one obtains
Thus which concludes the proof of the second claim.
Finally let be a convex and differentiable function. Then by definition:
which shows that .
3 Why convexity?
The key to the algorithmic success in minimizing convex functions is that these functions exhibit a local to global phenomenon. We have already seen one instance of this in Proposition 1.2.4, where we showed that : the gradient contains a priori only local information about the function around while the subdifferential gives a global information in the form of a linear lower bound on the entire function. Another instance of this local to global phenomenon is that local minima of convex functions are in fact global minima:
Let be convex. If is a local minimum of then is a global minimum of . Furthermore this happens if and only if .
Clearly if and only if is a global minimum of . Now assume that is local minimum of . Then for small enough one has for any ,
which implies and thus is a global minimum of .
The nice behavior of convex functions will allow for very fast algorithms to optimize them. This alone would not be sufficient to justify the importance of this class of functions (after all constant functions are pretty easy to optimize). However it turns out that surprisingly many optimization problems admit a convex (re)formulation. The excellent book Boyd and Vandenberghe (2004) describes in great details the various methods that one can employ to uncover the convex aspects of an optimization problem. We will not repeat these arguments here, but we have already seen that many famous machine learning problems (SVM, ridge regression, logistic regression, LASSO, sparse covariance estimation, and matrix completion) are formulated as convex problems.
We conclude this section with a simple extension of the optimality condition “” to the case of constrained optimization. We state this result in the case of a differentiable function for sake of simplicity.
Let be convex and a closed convex set on which is differentiable. Then
The “if" direction is trivial by using that a gradient is also a subgradient. For the “only if" direction it suffices to note that if , then is locally decreasing around on the line to (simply consider and note that ).
4 Black-box model
A zeroth order oracle takes as input a point and outputs the value of at .
A first order oracle takes as input a point and outputs a subgradient of at .
In this context we are interested in understanding the oracle complexity of convex optimization, that is how many queries to the oracles are necessary and sufficient to find an -approximate minima of a convex function. To show an upper bound on the sample complexity we need to propose an algorithm, while lower bounds are obtained by information theoretic reasoning (we need to argue that if the number of queries is “too small" then we don’t have enough information about the function to identify an -approximate solution).
The black-box model was essentially developed in the early days of convex optimization (in the Seventies) with Nemirovski and Yudin (1983) being still an important reference for this theory (see also Nemirovski (1995)). In the recent years this model and the corresponding algorithms have regained a lot of popularity, essentially for two reasons:
It is possible to develop algorithms with dimension-free oracle complexity which is quite attractive for optimization problems in very high dimension.
Many algorithms developed in this model are robust to noise in the output of the oracles. This is especially interesting for stochastic optimization, and very relevant to machine learning applications. We will explore this in details in Chapter 6.
Chapter 2, Chapter 3 and Chapter 4 are dedicated to the study of the black-box model (noisy oracles are discussed in Chapter 6). We do not cover the setting where only a zeroth order oracle is available, also called derivative free optimization, and we refer to Conn et al. (2009); Audibert et al. (2011) for further references on this.
5 Structured optimization
The black-box model described in the previous section seems extremely wasteful for the applications we discussed in Section 1.1. Consider for instance the LASSO objective: . We know this function globally, and assuming that we can only make local queries through oracles seem like an artificial constraint for the design of algorithms. Structured optimization tries to address this observation. Ultimately one would like to take into account the global structure of both and in order to propose the most efficient optimization procedure. An extremely powerful hammer for this task are the Interior Point Methods. We will describe this technique in Chapter 5 alongside with other more recent techniques such as FISTA or Mirror Prox.
We briefly describe now two classes of optimization problems for which we will be able to exploit the structure very efficiently, these are the LPs (Linear Programs) and SDPs (Semi-Definite Programs). Ben-Tal and Nemirovski (2001) describe a more general class of Conic Programs but we will not go in that direction here.
6 Overview of the results and disclaimer
The overarching aim of this monograph is to present the main complexity theorems in convex optimization and the corresponding algorithms. We focus on five major results in convex optimization which give the overall structure of the text: the existence of efficient cutting-plane methods with optimal oracle complexity (Chapter 2), a complete characterization of the relation between first order oracle complexity and curvature in the objective function (Chapter 3), first order methods beyond Euclidean spaces (Chapter 4), non-black box methods (such as interior point methods) can give a quadratic improvement in the number of iterations with respect to optimal black-box methods (Chapter 5), and finally noise robustness of first order methods (Chapter 6). Table 1.1 can be used as a quick reference to the results proved in Chapter 2 to Chapter 5, as well as some of the results of Chapter 6 (this last chapter is the most relevant to machine learning but the results are also slightly more specific which make them harder to summarize).
An important disclaimer is that the above selection leaves out methods derived from duality arguments, as well as the two most popular research avenues in convex optimization: (i) using convex optimization in non-convex settings, and (ii) practical large-scale algorithms. Entire books have been written on these topics, and new books have yet to be written on the impressive collection of new results obtained for both (i) and (ii) in the past five years.
A few of the blatant omissions regarding (i) include (a) the theory of submodular optimization (see Bach (2013)), (b) convex relaxations of combinatorial problems (a short example is given in Section 6.6), and (c) methods inspired from convex optimization for non-convex problems such as low-rank matrix factorization (see e.g. Jain et al. (2013) and references therein), neural networks optimization, etc.
With respect to (ii) the most glaring omissions include (a) heuristics (the only heuristic briefly discussed here is the non-linear conjugate gradient in Section 2.4), (b) methods for distributed systems, and (c) adaptivity to unknown parameters. Regarding (a) we refer to Nocedal and Wright (2006) where the most practical algorithms are discussed in great details (e.g., quasi-newton methods such as BFGS and L-BFGS, primal-dual interior point methods, etc.). The recent survey Boyd et al. (2011) discusses the alternating direction method of multipliers (ADMM) which is a popular method to address (b). Finally (c) is a subtle and important issue. In the entire monograph the emphasis is on presenting the algorithms and proofs in the simplest way, and thus for sake of convenience we assume that the relevant parameters describing the regularity and curvature of the objective function (Lipschitz constant, smoothness constant, strong convexity parameter) are known and can be used to tune the algorithm’s own parameters. Line search is a powerful technique to replace the knowledge of these parameters and it is heavily used in practice, see again Nocedal and Wright (2006). We observe however that from a theoretical point of view (c) is only a matter of logarithmic factors as one can always run in parallel several copies of the algorithm with different guesses for the values of the parametersNote that this trick does not work in the context of Chapter 6.. Overall the attitude of this text with respect to (ii) is best summarized by a quote of Thomas Cover: “theory is the first term in the Taylor series of practice”, Cover (1992).
Chapter 2 Convex optimization in finite dimension
As we will see these algorithms have an oracle complexity which is linear (or quadratic) in the dimension, hence the title of the chapter (in the next chapter the oracle complexity will be independent of the dimension). An interesting feature of the methods discussed here is that they only need a separation oracle for the constraint set . In the literature such algorithms are often referred to as cutting plane methods. In particular these methods can be used to find a point given only a separating oracle for (this is also known as the feasibility problem).
We consider the following simple iterative algorithmAs a warm-up we assume in this section that is known. It should be clear from the arguments in the next section that in fact the same algorithm would work if initialized with .: let , and for do the following:
Query the first order oracle at and obtain . Let
If stopped after queries to the first order oracle then we use queries to a zeroth order oracle to output
This procedure is known as the center of gravity method, it was discovered independently on both sides of the Wall by Levin (1965) and Newman (1965).
Before proving this result a few comments are in order.
To attain an -optimal point the center of gravity method requires queries to both the first and zeroth order oracles. It can be shown that this is the best one can hope for, in the sense that for small enough one needs calls to the oracle in order to find an -optimal point, see Nemirovski and Yudin (1983) for a formal proof.
The rate of convergence given by Theorem 2.1.1 is exponentially fast. In the optimization literature this is called a linear rate as the (estimated) error at iteration is linearly related to the error at iteration .
The last and most important comment concerns the computational complexity of the method. It turns out that finding the center of gravity is a very difficult problem by itself, and we do not have computationally efficient procedure to carry out this computation in general. In Section 6.7 we will discuss a relatively recent (compared to the 50 years old center of gravity method!) randomized algorithm to approximately compute the center of gravity. This will in turn give a randomized center of gravity method which we will describe in detail.
We now turn to the proof of Theorem 2.1.1. We will use the following elementary result from convex geometry:
Let be such that . Since one has
which clearly implies that one can never remove the optimal point from our sets in consideration, that is for any . Without loss of generality we can assume that we always have , for otherwise one would have which immediately conludes the proof. Now using that for any and Lemma 2.1.2 one clearly obtains
2 The ellipsoid method
Recall that an ellipsoid is a convex set of the form
We give now a simple geometric lemma, which is at the heart of the ellipsoid method.
By doing a quick picture, one can see that it makes sense to look for an ellipsoid that would be centered at , with (presumably will be small), and such that one principal direction is (with inverse squared semi-axis ), and the other principal directions are all orthogonal to (with the same inverse squared semi-axes ). In other words we are looking for with
Now we have to express our constraints on the fact that should contain the half Euclidean ball . Since we are also looking for to be as small as possible, it makes sense to ask for to "touch" the Euclidean ball, both at , and at the equator . The former condition can be written as:
As one can see from the above two equations, we are still free to choose any value for (the fact that we need comes from ). Quite naturally we take the value that minimizes the volume of the resulting ellipsoid. Note that
where . Elementary computations show that the maximum of (on $h=1+\frac{1}{n}t=\frac{1}{n+1}$), and the value is
where the lower bound follows again from elementary computations. Thus we showed that, for , (2.3) and (2.4) are satisfied with the ellipsoid given by the set of points satisfying:
Applying Sherman-Morrison formula to (2.8) one can recover (2.6) which concludes the proof.
Let be the ellipsoid given in Lemma 2.2.1 that contains , that is
If stopped after iterations and if , then we use the zeroth order oracle to output
The following rate of convergence can be proved with the exact same argument than for Theorem 2.1.1 (observe that at step one can remove a point in from the current ellipsoid only if ).
For the ellipsoid method satisfies and
We observe that the oracle complexity of the ellipsoid method is much worse than the one of the center gravity method, indeed the former needs calls to the oracles while the latter requires only calls. However from a computational point of view the situation is much better: in many cases one can derive an efficient separation oracle, while the center of gravity method is basically always intractable. This is for instance the case in the context of LPs and SDPs: with the notation of Section 1.5 the computational complexity of the separation oracle for LPs is while for SDPs it is (we use the fact that the spectral decomposition of a matrix can be done in operations). This gives an overall complexity of for LPs and for SDPs. We note however that the ellipsoid method is almost never used in practice, essentially because the method is too rigid to exploit the potential easiness of real problems (e.g., the volume decrease given by (2.4) is essentially always tight).
3 Vaidya’s cutting plane method
We focus here on the feasibility problem (it should be clear from the previous sections how to adapt the argument for optimization). We have seen that for the feasibility problem the center of gravity has a oracle complexity and unclear computational complexity (see Section 6.7 for more on this), while the ellipsoid method has oracle complexity and computational complexity . We describe here the beautiful algorithm of Vaidya (1989, 1996) which has oracle complexity and computational complexity , thus getting the best of both the center of gravity and the ellipsoid method. In fact the computational complexity can even be improved further, and the recent breakthrough Lee et al. (2015) shows that it can essentially (up to logarithmic factors) be brought down to .
This section, while giving a fundamental algorithm, should probably be skipped on a first reading. In particular we use several concepts from the theory of interior point methods which are described in Section 5.3.
We also consider the volumetric barrier defined by
The intuition is clear: is equal to the logarithm of the inverse volume of the Dikin ellipsoid (for the logarithmic barrier) at . It will be useful to spell out the hessian of the logarithmic barrier:
3.2 Vaidya’s algorithm
If for some one has , then is defined by removing the row in (in particular ).
Then we define by adding to the row given by (in particular ).
It can be shown that the volumetric barrier is a self-concordant barrier, and thus it can be efficiently minimized with Newton’s method. In fact it is enough to do one step of Newton’s method on initialized at , see Vaidya (1989, 1996) for more details on this.
3.3 Analysis of Vaidya’s method
The construction of Vaidya’s method is based on a precise understanding of how the volumetric barrier changes when one adds or removes a constraint to the polytope. This understanding is derived in Section 2.3.4. In particular we obtain the following two key inequalities: If case 1 happens at iteration then
We show now how these inequalities imply that Vaidya’s method stops after steps. First we claim that after iterations, case 2 must have happened at least times. Indeed suppose that at iteration , case 2 has happened times; then is singular and the leverage scores are infinite, so case 2 must happen at iteration . Combining this claim with the two inequalities above we obtain:
Since is always included in the polytope we have that is at most the logarithm of the volume of the initial polytope which is . This clearly concludes the proof as the procedure will necessarily stop when the volume is below (we also used the trivial bound ).
3.4 Constraints and the volumetric barrier
The leverage scores and are closely related:
First we observe that by Sherman-Morrison’s formula one has
This immediately proves . It also implies the inequality thanks the following fact: . For the last inequality we use that together with
We now assume the following key result, which was first proven by Vaidya. To put the statement in context recall that for a self-concordant barrier the suboptimality gap is intimately related to the Newton decrement . Vaidya’s inequality gives a similar claim for the volumetric barrier. We use the version given in [Theorem 2.6, Anstreicher (1998)] which has slightly better numerical constants than the original bound. Recall also the definition of from (2.10).
Let be an approximate Newton decrement, , and assume that . Then
We also denote for the approximate Newton decrement of . The goal for the rest of the section is to prove the following theorem which gives the precise understanding of the volumetric barrier we were looking for.
Let , and assume that . Then one has
On the other hand assuming that and that , one has
Before going into the proof let us see briefly how Theorem 2.3.4 give the two inequalities stated at the beginning of Section 2.3.3. To prove (2.12) we use (2.14) with and , and we observe that in this case the right hand side of (2.14) is lower bounded by . On the other hand to prove (2.11) we use (2.15), and we observe that for the right hand side of (2.15) is upper bounded by .
We start with the proof of (2.14). First observe that by factoring on the left and on the right of one obtains
To bound the suboptimality gap of in we will invoke Theorem 2.3.3 and thus we have to upper bound the approximate Newton decrement . Using [(2.16), Lemma 2.3.6] below one has
We now turn to the proof of (2.15). Following the same steps as above we immediately obtain
To invoke Theorem 2.3.3 it remains to upper bound . Using [(2.17), Lemma 2.3.6] below one has
We can apply Theorem 2.3.3 since the assumption implies that . This concludes the proof of (2.15).
Furthermore if then one also has
We start with the proof of (2.16). First observe that by Lemma 2.3.1 one has and thus by definition of the Newton decrement
We now use that to obtain
By Lemma 2.3.1 one has and thus we see that it only remains to prove
The above inequality follows from a beautiful calculation of Vaidya (see [Lemma 12, Vaidya (1996)]), starting from the identity
We now turn to the proof of (2.17). Following the same steps as above we immediately obtain
Using Lemma 2.3.1 together with the assumption yields (2.17), thus concluding the proof.
4 Conjugate gradient
Equation (2.20) is true by construction for , and for it follows by induction, assuming (2.20) at and using the following formula:
which concludes the proof of .
In order to have a proper black-box method it remains to describe how to build iteratively the orthogonal set based only on gradient evaluations of . A natural guess to obtain a set of orthogonal directions (w.r.t. ) is to take and for ,
Furthermore using the definition (2.22) and one also has
Thus we arrive at the following rewriting of the (linear) conjugate gradient algorithm, where we recall that is some fixed starting point and ,
Observe that the algorithm defined by (2.23) and (2.24) makes sense for an arbitary convex function, in which case it is called the non-linear conjugate gradient. There are many variants of the non-linear conjugate gradient, and the above form is known as the Fletcher-–Reeves method. Another popular version in practice is the Polak-Ribière method which is based on the fact that for the general non-quadratic case one does not necessarily have , and thus one replaces (2.24) by
We refer to Nocedal and Wright (2006) for more details about these algorithms, as well as for advices on how to deal with the line search in (2.23).
Finally we also note that the linear conjugate gradient method can often attain an approximate solution in much fewer than steps. More precisely, denoting for the condition number of (that is the ratio of the largest eigenvalue to the smallest eigenvalue of ), one can show that linear conjugate gradient attains an optimal point in a number of iterations of order . The next chapter will demistify this convergence rate, and in particular we will see that (i) this is the optimal rate among first order methods, and (ii) there is a way to generalize this rate to non-quadratic convex functions (though the algorithm will have to be modified).
Chapter 3 Dimension-free convex optimization
where is a fixed step-size parameter. The rationale behind (3.1) is to make a small step in the direction that minimizes the local first order Taylor approximation of (also known as the steepest descent direction).
As we shall see, methods of the type (3.1) can obtain an oracle complexity independent of the dimensionOf course the computational complexity remains at least linear in the dimension since one needs to manipulate gradients.. This feature makes them particularly attractive for optimization in very high dimension.
The following lemma will prove to be useful in our study. It is an easy corollary of Proposition 1.3.3, see also Figure 3.1.
which also implies .
Unless specified otherwise all the proofs in this chapter are taken from Nesterov (2004a) (with slight simplification in some cases).
In this section we assume that is contained in an Euclidean ball centered at and of radius . Furthermore we assume that is such that for any and any (we assume ), one has . Note that by the subgradient inequality and Cauchy-Schwarz this implies that is -Lipschitz on , that is .
In this context we make two modifications to the basic gradient descent (3.1). First, obviously, we replace the gradient (which may not exist) by a subgradient . Secondly, and more importantly, we make sure that the updated point lies in by projecting back (if necessary) onto it. This gives the projected subgradient descent algorithmIn the optimization literature the term “descent” is reserved for methods such that . In that sense the projected subgradient descent is not a descent method. which iterates the following equations for :
This procedure is illustrated in Figure 3.2. We prove now a rate of convergence for this method under the above assumptions.
The projected subgradient descent method with satisfies
Using the definition of subgradients, the definition of the method, and the elementary identity , one obtains
Now note that , and furthermore by Lemma 3.0.1
Summing the resulting inequality over , and using that yield
Plugging in the value of directly gives the statement (recall that by convexity ).
We will show in Section 3.5 that the rate given in Theorem 3.1.1 is unimprovable from a black-box perspective. Thus to reach an -optimal point one needs calls to the oracle. In some sense this is an astonishing result as this complexity is independentObserve however that the quantities and may dependent on the dimension, see Chapter 4 for more on this. of the ambient dimension . On the other hand this is also quite disappointing compared to the scaling in of the center of gravity and ellipsoid method of Chapter 2. To put it differently with gradient descent one could hope to reach a reasonable accuracy in very high dimension, while with the ellipsoid method one can reach very high accuracy in reasonably small dimension. A major task in the following sections will be to explore more restrictive assumptions on the function to be optimized in order to have the best of both worlds, that is an oracle complexity independent of the dimension and with a scaling in .
Finally we observe that the step-size recommended by Theorem 3.1.1 depends on the number of iterations to be performed. In practice this may be an undesirable feature. However using a time-varying step size of the form one can prove the same rate up to a factor. In any case these step sizes are very small, which is the reason for the slow convergence. In the next section we will see that by assuming smoothness in the function one can afford to be much more aggressive. Indeed in this case, as one approaches the optimum the size of the gradients themselves will go to , resulting in a sort of “auto-tuning" of the step sizes which does not happen for an arbitrary convex function.
2 Gradient descent for smooth functions
We say that a continuously differentiable function is -smooth if the gradient is -Lipschitz, that is
Before embarking on the proof we state a few properties of smooth convex functions.
We represent as an integral, apply Cauchy-Schwarz and then -smoothness:
This gives in particular the following important inequality to evaluate the improvement in one step of gradient descent:
The next lemma, which improves the basic inequality for subgradients under the smoothness assumption, shows that in fact is convex and -smooth if and only if (3.4) holds true. In the literature (3.4) is often used as a definition of smooth convex functions.
Let . Then one has
Using (3.5) and the definition of the method one has
In particular, denoting , this shows:
We will prove that is decreasing with , which with the two above displays will imply
Let us see how to use this last inequality to conclude the proof. Let , thenThe last step in the sequence of implications can be improved by taking into account. Indeed one can easily show with (3.4) that . This improves the rate of Theorem 3.2.1 from to .
Thus it only remains to show that is decreasing with . Using Lemma 3.2.4 one immediately gets
We use this as follows (together with )
We now come back to the constrained problem
Similarly to what we did in Section 3.1 we consider the projected gradient descent algorithm, which iterates .
The key point in the analysis of gradient descent for unconstrained smooth optimization is that a step of gradient descent started at will decrease the function value by at least , see (3.5). In the constrained case we cannot expect that this would still hold true as a step may be cut short by the projection. The next lemma defines the “right" quantity to measure progress in the constrained case.
Let , , and . Then the following holds true:
Indeed the above inequality is equivalent to
which follows from Lemma 3.0.1. Now we use (3.7) as follows to prove the lemma (we also use (3.4) which still holds true in the constrained case)
Let be convex and -smooth on . Then projected gradient descent with satisfies
We will prove that is decreasing with , which with the two above displays will imply
Thus it only remains to show that is decreasing with . Using Lemma 3.2.7 one can see that which implies
3 Conditional gradient descent, aka Frank-Wolfe
We describe now an alternative algorithm to minimize a smooth convex function over a compact convex set . The conditional gradient descent, introduced in Frank and Wolfe (1956), performs the following update for , where is a fixed sequence,
In words conditional gradient descent makes a step in the steepest descent direction given the constraint set , see Figure 3.3 for an illustration. From a computational perspective, a key property of this scheme is that it replaces the projection step of projected gradient descent by a linear optimization over , which in some cases can be a much simpler problem.
Let be a convex and -smooth function w.r.t. some norm , , and for . Then for any , one has
The following inequalities hold true, using respectively -smoothness (it can easily be seen that (3.4) holds true for smoothness in an arbitrary norm), the definition of , the definition of , and the convexity of :
Rewriting this inequality in terms of one obtains
A simple induction using that finishes the proof (note that the initialization is done at step with the above inequality yielding ).
Next we describe an application where the three properties of conditional gradient descent (projection-free, norm-free, and sparse iterates) are critical to develop a computationally efficient procedure.
This assumption is met for many combinatorial dictionaries. For instance the dictionary elements could be vector of incidence of spanning trees in some fixed graph, in which case the linear optimization problem can be solved with a greedy algorithm.
First let us study the computational complexity of the step of conditional gradient descent. Observe that
To derive a rate of convergence it remains to study the smoothness of . This can be done as follows:
Putting together (3.11) and (3.12) we proved that one can get an -optimal solution to (3.10) with a computational effort of using the conditional gradient descent.
4 Strong convexity
Of course this definition does not require differentiability of the function , and one can replace in the inequality above by . It is immediate to verify that a function is -strongly convex if and only if is convex (in particular if is twice differentiable then the eigenvalues of the Hessians of have to be larger than ). The strong convexity parameter is a measure of the curvature of . For instance a linear function has no curvature and hence . On the other hand one can clearly see why a large value of would lead to a faster rate: in this case a point far from the optimum will have a large gradient, and thus gradient descent will make very big steps when far from the optimum. Of course if the function is non-smooth one still has to be careful and tune the step-sizes to be relatively small, but nonetheless we will be able to improve the oracle complexity from to . On the other hand with the additional assumption of -smoothness we will prove that gradient descent with a constant step-size achieves a linear rate of convergence, precisely the oracle complexity will be . This achieves the objective we had set after Theorem 3.1.1: strongly-convex and smooth functions can be optimized in very large dimension and up to very high accuracy.
Before going into the proofs let us discuss another interpretation of strong-convexity and its relation to smoothness. Equation (3.13) can be read as follows: at any point one can find a (convex) quadratic lower bound to the function , i.e. (and ). On the other hand for -smoothness (3.4) implies that at any point one can find a (convex) quadratic upper bound to the function , i.e. (and ). Thus in some sense strong convexity is a dual assumption to smoothness, and in fact this can be made precise within the framework of Fenchel duality. Also remark that clearly one always has .
We consider here the projected subgradient descent algorithm with time-varying step size , that is
The following result is extracted from Lacoste-Julien et al. (2012).
Let be -strongly convex and -Lipschitz on . Then projected subgradient descent with satisfies
Coming back to our original analysis of projected subgradient descent in Section 3.1 and using the strong convexity assumption one immediately obtains
Multiplying this inequality by yields
Now sum the resulting inequality over to , and apply Jensen’s inequality to obtain the claimed statement.
4.2 Strongly convex and smooth functions
As we will see now, having both strong convexity and smoothness allows for a drastic improvement in the convergence rate. We denote for the condition number of . The key observation is that Lemma 3.2.7 can be improved to (with the notation of the lemma):
Let be -strongly convex and -smooth on . Then projected gradient descent with satisfies for ,
Using (3.14) with one directly obtains
We now show that in the unconstrained case one can improve the rate by a constant factor, precisely one can replace by in the oracle complexity bound by using a larger step size. This is not a spectacular gain but the reasoning is based on an improvement of (3.6) which can be of interest by itself. Note that (3.6) and the lemma to follow are sometimes referred to as coercivity of the gradient.
Let . By definition of -strong convexity one has that is convex. Furthermore one can show that is -smooth by proving (3.4) (and using that it implies smoothness). Thus using (3.6) one gets
which gives the claimed result with straightforward computations. (Note that if the smoothness of directly implies that which proves the lemma in this case.)
First note that by -smoothness (since ) one has
5 Lower bounds
We prove here various oracle complexity lower bounds. These results first appeared in Nemirovski and Yudin (1983) but we follow here the simplified presentation of Nesterov (2004a). In general a black-box procedure is a mapping from “history" to the next query point, that is it maps (with ) to . In order to simplify the notation and the argument, throughout the section we make the following assumption on the black-box procedure: and for any , is in the linear span of , that is
Let , . There exists a convex and -Lipschitz function such that for any black-box procedure satisfying (3.15),
There also exists an -strongly convex and -lipschitz function such that for any black-box procedure satisfying (3.15),
Note that the above result is restricted to a number of iterations smaller than the dimension, that is . This restriction is of course necessary to obtain lower bounds polynomial in : as we saw in Chapter 2 one can always obtain an exponential rate of convergence when the number of calls to the oracle is larger than the dimension.
We consider the following -strongly convex function:
Next we describe the first order oracle for this function: when asked for a subgradient at , it returns where is the first coordinate that satisfies . In particular when asked for a subgradient at it returns . Thus must lie on the line generated by . It is easy to see by induction that in fact must lie in the linear span of . In particular for we necessarily have and thus .
It remains to compute the minimal value of . Let be such that for and for . It is clear that and thus the minimal value of is
Wrapping up, we proved that for any one must have
Taking and we proved the lower bound for -strongly convex functions (note in particular that with these parameters). On the other taking and concludes the proof for convex functions (note in particular that with these parameters).
We proceed now to the smooth case. As we will see in the following proofs we restrict our attention to quadratic functions, and it might be useful to recall that in this case one can attain the exact optimum in calls to the oracle (see Section 2.4). We also recall that for a twice differentiable function , -smoothness is equivalent to the largest eigenvalue of the Hessians of being smaller than at any point, which we write
Furthermore -strong convexity is equivalent to
Let , . There exists a -smooth convex function such that for any black-box procedure satisfying (3.15),
We consider now the following -smooth convex function:
Similarly to what happened in the proof Theorem 3.5.1, one can see here too that must lie in the linear span of (because of our assumption on the black-box procedure). In particular for we necessarily have for , which implies . In other words, if we denote
Thus it simply remains to compute the minimizer of , its norm, and the corresponding function value .
The point is the unique solution in the span of of . It is easy to verify that it is defined by for . Thus we immediately have:
Note that for large values of the condition number one has
Furthermore since is -strongly convex, one has
Thus it only remains to compute . This can be done by differentiating and setting the gradient to , which gives the following infinite set of equations
It is easy to verify that defined by satisfy this infinite set of equations, and the conclusion of the theorem then follows by straightforward computations.
6 Geometric descent
So far our results leave a gap in the case of smooth optimization: gradient descent achieves an oracle complexity of (respectively in the strongly convex case) while we proved a lower bound of (respectively ). In this section we close these gaps with the geometric descent method which was recently introduced in Bubeck et al. (2015b). Historically the first method with optimal oracle complexity was proposed in Nemirovski and Yudin (1983). This method, inspired by the conjugate gradient (see Section 2.4), assumes an oracle to compute plane searches. In Nemirovski (1982) this assumption was relaxed to a line search oracle (the geometric descent method also requires a line search oracle). Finally in Nesterov (1983) an optimal method requiring only a first order oracle was introduced. The latter algorithm, called Nesterov’s accelerated gradient descent, has been the most influential optimal method for smooth optimization up to this day. We describe and analyze this method in Section 3.7. As we shall see the intuition behind Nesterov’s accelerated gradient descent (both for the derivation of the algorithm and its analysis) is not quite transparent, which motivates the present section as geometric descent has a simple geometric interpretation loosely inspired from the ellipsoid method (see Section 2.2).
We focus here on the unconstrained optimization of a smooth and strongly convex function, and we prove that geometric descent achieves the oracle complexity of , thus reducing the complexity of the basic gradient descent by a factor . We note that this improvement is quite relevant for machine learning applications. Consider for example the logistic regression problem described in Section 1.1: this is a smooth and strongly convex problem, with a smoothness of order of a numerical constant, but with strong convexity equal to the regularization parameter whose inverse can be as large as the sample size. Thus in this case can be of order of the sample size, and a faster rate by a factor of is quite significant. We also observe that this improved rate for smooth and strongly convex objectives also implies an almost optimal rate of for the smooth case, as one can simply run geometric descent on the function .
In Section 3.6.1 we describe the basic idea of geometric descent, and we show how to obtain effortlessly a geometric method with an oracle complexity of (i.e., similar to gradient descent). Then we explain why one should expect to be able to accelerate this method in Section 3.6.2. The geometric descent method is described precisely and analyzed in Section 3.6.3.
Rewriting the definition of strong convexity (3.13) as
one obtains an enclosing ball for the minimizer of with the and order information at :
Furthermore recall that by smoothness (see (3.5)) one has which allows to shrink the above ball by a factor of and obtain the following:
Thus we see that in the strategy described above, the radius squared of the enclosing ball for shrinks by a factor at each iteration, thus matching the rate of convergence of gradient descent (see Theorem 3.4.3).
6.2 Acceleration
Thus it only remains to deal with the caveat noted above, which we do via a line search. In turns this line search might shift the new ball (3.16), and to deal with this we shall need the following strengthening of the above set inclusion (we refer to Bubeck et al. (2015b) for a simple proof of this result):
6.3 The geometric descent method
and (respectively ) be the center (respectively the squared radius) of the ball given by (the proof of) Lemma 3.6.1 which contains
Formulas for and are given at the end of this section.
We will prove a stronger claim by induction that for each , one has
The case follows immediately by (3.16). Let us assume that the above display is true for some . Then using one gets
Thus it only remains to observe that the squared radius of the ball given by Lemma 3.6.1 which encloses the intersection of the two above balls is smaller than . We apply Lemma 3.6.1 after moving to the origin and scaling distances by . We set , , and . The line search step of the algorithm implies that and therefore, and Lemma 3.6.1 applies to give the result.
One can use the following formulas for and (they are derived from the proof of Lemma 3.6.1). If then one can tate and . On the other hand if then one can tate
7 Nesterov’s accelerated gradient descent
We describe here the original Nesterov’s method which attains the optimal oracle complexity for smooth convex optimization. We give the details of the method both for the strongly convex and non-strongly convex case. We refer to Su et al. (2014) for a recent interpretation of the method in terms of differential equations, and to Allen-Zhu and Orecchia (2014) for its relation to mirror descent (see Chapter 4).
Nesterov’s accelerated gradient descent, illustrated in Figure 3.6, can be described as follows: Start at an arbitrary initial point and then iterate the following equations for ,
Let be -strongly convex and -smooth, then Nesterov’s accelerated gradient descent satisfies
We define -strongly convex quadratic functions by induction as follows:
Intuitively becomes a finer and finer approximation (from below) to in the following sense:
The above inequality can be proved immediately by induction, using the fact that by -strong convexity one has
Equation (3.18) by itself does not say much, for it to be useful one needs to understand how “far" below is . The following inequality answers this question:
The rest of the proof is devoted to showing that (3.19) holds true, but first let us see how to combine (3.18) and (3.19) to obtain the rate given by the theorem (we use that by -smoothness one has ):
In particular is by definition minimized at which can now be defined by induction using the above identity, precisely:
Using the form of and , as well as the original definition (3.17) one gets the following identity by evaluating at :
Finally we show by induction that , which concludes the proof of (3.20) and thus also concludes the proof of the theorem:
where the first equality comes from (3.21), the second from the induction hypothesis, the third from the definition of and the last one from the definition of .
7.2 The smooth case
In this section we show how to adapt Nesterov’s accelerated gradient descent for the case , using a time-varying combination of the elements in the primary sequence . First we define the following sequences:
(Note that .) Now the algorithm is simply defined by the following equations, with an arbitrary initial point,
Let be a convex and -smooth function, then Nesterov’s accelerated gradient descent satisfies
We follow here the proof of Beck and Teboulle (2009). We also refer to Tseng (2008) for a proof with simpler step-sizes.
Using the unconstrained version of Lemma 3.2.7 one obtains
Now multiplying (3.23) by and adding the result to (3.24), one obtains with ,
Multiplying this inequality by and using that by definition , as well as the elementary identity , one obtains
Putting together (3.25) and (3.26) one gets with ,
Summing these inequalities from to one obtains:
By induction it is easy to see that which concludes the proof.
Chapter 4 Almost dimension-free convex optimization in non-Euclidean spaces
We also define the Bregman divergence associated to as
The following identity will be useful several times:
is strictly convex and differentiable.
The gradient of diverges on the boundary of , that is
Property (i) and (iii) ensures the existence and uniqueness of this projection (in particular since is locally increasing on the boundary of ). The following lemma shows that the Bregman divergence essentially behaves as the Euclidean norm squared in terms of projections (recall Lemma 3.0.1).
Let and , then
The proof is an immediate corollary of Proposition 1.3.3 together with the fact that .
2 Mirror descent
See Figure 4.1 for an illustration of this procedure.
Let be a mirror map -strongly convex on w.r.t. . Let , and be convex and -Lipschitz w.r.t. . Then mirror descent with satisfies
Let . The claimed bound will be obtained by taking a limit . Now by convexity of , the definition of mirror descent, equation (4.1), and Lemma 4.1.1, one has
which concludes the proof up to trivial computation.
We observe that one can rewrite mirror descent as follows:
This last expression is often taken as the definition of mirror descent (see Beck and Teboulle (2003)). It gives a proximal point of view on mirror descent: the method is trying to minimize the local linearization of the function while not moving too far away from the previous point, with distances measured via the Bregman divergence of the mirror map.
3 Standard setups for mirror descent
“Simplex setup". A more interesting choice of a mirror map is given by the negative entropy
“Spectrahedron setup". We consider here functions defined on matrices, and we are interested in minimizing a function on the spectrahedron defined as:
where are the eigenvalues of . It can be shown that the gradient update can be written equivalently as
where the matrix exponential and matrix logarithm are defined as usual. Furthermore the projection on is a simple trace renormalization.
With highly non-trivial computation one can show that is -strongly convex with respect to the Schatten -norm defined as
4 Lazy mirror descent, aka Nesterov’s dual averaging
In this section we consider a slightly more efficient version of mirror descent for which we can prove that Theorem 4.2.1 still holds true. This alternative algorithm can be advantageous in some situations (such as distributed settings), but the basic mirror descent scheme remains important for extensions considered later in this text (saddle points, stochastic oracles, …).
In lazy mirror descent, also commonly known as Nesterov’s dual averaging or simply dual averaging, one replaces (4.2) by
and also is such that . In other words instead of going back and forth between the primal and the dual, dual averaging simply averages the gradients in the dual, and if asked for a point in the primal it simply maps the current dual point to the primal using the same methodology as mirror descent. In particular using (4.2) one immediately sees that dual averaging is defined by:
Let be a mirror map -strongly convex on w.r.t. . Let , and be convex and -Lipschitz w.r.t. . Then dual averaging with satisfies
where the second inequality comes from the first order optimality condition for (see Proposition 1.3.3). Next observe that
Putting together the two above displays and using Cauchy-Schwarz (with the assumption ) one obtains
In particular this shows that and thus with the above display
Now we claim that for any ,
which would clearly conclude the proof thanks to (4.7) and straightforward computations. Equation (4.8) is equivalent to
5 Mirror prox
It can be shown that mirror descent accelerates for smooth functions to the rate . We will prove this result in Chapter 6 (see Theorem 6.2.1). We describe here a variant of mirror descent which also attains the rate for smooth functions. This method is called mirror prox and it was introduced in Nemirovski (2004a). The true power of mirror prox will reveal itself later in the text when we deal with smooth representations of non-smooth functions as well as stochastic oraclesBasically mirror prox allows for a smooth vector field point of view (see Section 4.6), while mirror descent does not..
Mirror prox is described by the following equations:
In words the algorithm first makes a step of mirror descent to go from to , and then it makes a similar step to obtain , starting again from but this time using the gradient of evaluated at (instead of ), see Figure 4.2 for an illustration. The following result justifies the procedure.
Let be a mirror map -strongly convex on w.r.t. . Let , and be convex and -smooth w.r.t. . Then mirror prox with satisfies
Let . We write
We will now bound separately these three terms. For the first one, using the definition of the method, Lemma 4.1.1, and equation (4.1), one gets
For the second term using the same properties than above and the strong-convexity of the mirror map one obtains
Finally for the last term, using Cauchy-Schwarz, -smoothness, and one gets
Thus summing up these three terms and using that one gets
The proof is concluded with straightforward computations.
6 The vector field point of view on MD, DA, and MP
In this section we consider a mirror map that satisfies the assumptions from Theorem 4.2.1.
The observation that the sequence of vectors does not have to come from the subgradients of a fixed function is the starting point for the theory of online learning, see Bubeck (2011) for more details. In this monograph we will use this observation to generalize mirror descent to saddle point calculations as well as stochastic settings. We note that we could also use dual averaging (defined by (4.6)) which satisfies
Under the assumption that the vector field is -Lipschitz w.r.t. , i.e., one obtains with
Chapter 5 Beyond the black-box model
We consider here the following problemWe restrict to unconstrained minimization for sake of simplicity. One can extend the discussion to constrained minimization by using ideas from Section 3.2.:
where is convex and -smooth, and is convex. We assume that can be accessed through a first order oracle, and that is known and “simple". What we mean by simplicity will be clear from the description of the algorithm. For instance a separable function, that is , will be considered as simple. The prime example being . This section is inspired from Beck and Teboulle (2009) (see also Nesterov (2007); Wright et al. (2009)).
ISTA (Iterative Shrinkage-Thresholding Algorithm)
Recall that gradient descent on the smooth function can be written as (see (4.5))
Here one wants to minimize , and is assumed to be known and “simple". Thus it seems quite natural to consider the following update rule, where only is locally approximated with a first order oracle:
The algorithm described by the above iteration is known as ISTA (Iterative Shrinkage-Thresholding Algorithm). In terms of convergence rate it is easy to show that ISTA has the same convergence rate on as gradient descent on . More precisely with one has
This improved convergence rate over a subgradient descent directly on comes at a price: in general (5.1) may be a difficult optimization problem by itself, and this is why one needs to assume that is simple. For instance if can be written as then one can compute by solving convex problems in dimension . In the case where this one-dimensional problem is given by:
Elementary computations shows that this problem has an analytical solution given by , where is the shrinkage operator (hence the name ISTA), defined by
Much more is known about (5.1) (which is called the proximal operator of ), and in fact entire monographs have been written about this equation, see e.g. Parikh and Boyd (2013); Bach et al. (2012).
FISTA (Fast ISTA)
An obvious idea is to combine Nesterov’s accelerated gradient descent (which results in a rate to optimize ) with ISTA. This results in FISTA (Fast ISTA) which is described as follows. Let
Let an arbitrary initial point, and
Again it is easy show that the rate of convergence of FISTA on is similar to the one of Nesterov’s accelerated gradient descent on , more precisely:
CMD and RDA
ISTA and FISTA assume smoothness in the Euclidean metric. Quite naturally one can also use these ideas in a non-Euclidean setting. Starting from (4.5) one obtains the CMD (Composite Mirror Descent) algorithm of Duchi et al. (2010), while with (4.6) one obtains the RDA (Regularized Dual Averaging) of Xiao (2010). We refer to these papers for more details.
2 Smooth saddle-point representation of a non-smooth function
Quite often the non-smoothness of a function comes from a operation. More precisely non-smooth functions can often be represented as
where the functions are smooth. This was the case for instance with the function we used to prove the black-box lower bound for non-smooth optimization in Theorem 3.5.1. We will see now that by using this structural representation one can in fact attain a rate of . This was first observed in Nesterov (2004b) who proposed the Nesterov’s smoothing technique. Here we will present the alternative method of Nemirovski (2004a) which we find more transparent (yet another version is the Chambolle-Pock algorithm, see Chambolle and Pock (2011)). Most of what is described in this section can be found in Juditsky and Nemirovski (2011a, b).
In the next subsection we introduce the more general problem of saddle point computation. We then proceed to apply a modified version of mirror descent to this problem, which will be useful both in Chapter 6 and also as a warm-up for the more powerful modified mirror prox that we introduce next.
By Sion’s minimax theorem there exists a pair such that
We will explore algorithms that produce a candidate pair of solutions . The quality of is evaluated through the so-called duality gapObserve that the duality gap is the sum of the primal gap and the dual gap .
The key observation is that the duality gap can be controlled similarly to the suboptimality gap in a simple convex optimization problem. Indeed for any ,
In particular, using the notation and we just proved
We will assume in the next subsections that is equipped with a mirror map (defined on ) which is -strongly convex w.r.t. a norm on . We denote . We define similar quantities for the space .
2.2 Saddle Point Mirror Descent (SP-MD)
where with and .
Assume that is -Lipschitz w.r.t. , that is . Similarly assume that is -Lipschitz w.r.t. . Then SP-MD with , , and satisfies
First we endow with the norm defined by
It is immediate that is -strongly convex with respect to on . Furthermore one can easily check that
and thus the vector field used in the SP-MD satisfies:
Using (4.10) together with (5.3) and the values of and concludes the proof.
2.3 Saddle Point Mirror Prox (SP-MP)
We now consider the most interesting situation in the context of this chapter, where the function is smooth. Precisely we say that is -smooth if for any ,
Assume that is -smooth. Then SP-MP with , , and satisfies
In light of the proof of Theorem 5.2.1 and (4.11) it clearly suffices to show that the vector field is -Lipschitz w.r.t. with . In other words one needs to show that
which can be done with straightforward calculations (by introducing and using the definition of smoothness for ).
2.4 Applications
We investigate briefly three applications for SP-MD and SP-MP.
The problem (5.2) (when has to minimized over ) can be rewritten as
that is . On the other hand , and thus
that is and . Thus using SP-MP with some mirror map on and the negentropy on (see the “simplex setup" in Section 4.3), one obtains an -optimal point of in iterations. Furthermore an iteration of SP-MP has a computational complexity of order of a step of mirror descent in on the function (plus for the update in the -space).
Thus by using the structure of we were able to obtain a much better rate than black-box procedures (which would have required iterations as is potentially non-smooth).
Here we equip both and with . Let . Using that and one immediately obtains . Furthermore since
3 Interior point methods
We describe here interior point methods (IPM), a class of algorithms fundamentally different from what we have seen so far. The first algorithm of this type was described in Karmarkar (1984), but the theory we shall present was developed in Nesterov and Nemirovski (1994). We follow closely the presentation given in [Chapter 4, Nesterov (2004a)]. Other useful references (in particular for the primal-dual IPM, which are the ones used in practice) include Renegar (2001); Nemirovski (2004b); Nocedal and Wright (2006).
IPM are designed to solve convex optimization problems of the form
The idea of the barrier method is to move along the central path by “boosting" a fast locally convergent algorithm, which we denote for the moment by , using the following scheme: Assume that one has computed , then one uses initialized at to compute for some . There is a clear tension for the choice of , on the one hand should be large in order to make as much progress as possible on the central path, but on the other hand needs to be close enough to so that it is in the basin of fast convergence for when run on .
IPM follows the above methodology with being Newton’s method. Indeed as we will see in the next subsection, Newton’s method has a quadratic convergence rate, in the sense that if initialized close enough to the optimum it attains an -optimal point in iterations! Thus we now have a clear plan to make these ideas formal and analyze the iteration complexity of IPM:
First we need to describe precisely the region of fast convergence for Newton’s method. This will lead us to define self-concordant functions, which are “natural" functions for Newton’s method.
Then we need to evaluate precisely how much larger can be compared to , so that is still in the region of fast convergence of Newton’s method when optimizing the function with . This will lead us to define -self concordant barriers.
3.2 Traditional analysis of Newton’s method
Thus, starting at , in order to minimize it seems natural to move in the direction that minimizes
While this method can have an arbitrarily bad behavior in general, if started close enough to a strict local minimum of , it can have a very fast convergence:
Then Newton’s method is well-defined and converges to at a quadratic rate:
Now note that , and thus with the above formula one obtains
Using the Lipschitz property of the Hessian one immediately obtains that
3.3 Self-concordant functions
Before giving the definition of self-concordant functions let us try to get some insight into the “geometry" of Newton’s method. Let be a non-singular matrix. We look at a Newton step on the functions and , starting respectively from and , that is:
In other words Newton’s method will follow the same trajectory in the “-space" and in the “-space" (the image through of the -space), that is Newton’s method is affine invariant. Observe that this property is not shared by the methods described in Chapter 3 (except for the conditional gradient descent).
The issue is that this inequality depends on the choice of an inner product. More importantly it is easy to see that a convex function which goes to infinity on a compact set simply cannot satisfy the above inequality. A natural idea to try fix these issues is to replace the Euclidean metric on the right hand side by the metric given by the function itself at , that is:
Observe that to be clear one should rather use the notation , but since will always be clear from the context we stick to .
We say that is standard self-concordant if is self-concordant with constant .
An easy consequence of the definition is that a self-concordant function is a barrier for the set , see [Theorem 4.1.4, Nesterov (2004a)]. The main example to keep in mind of a standard self-concordant function is for . The next definition will be key in order to describe the region of quadratic convergence for Newton’s method on self-concordant functions.
see [Equation 4.1.18, Nesterov (2004a)]. We state the next theorem without a proof, see also [Theorem 4.1.14, Nesterov (2004a)].
In other words the above theorem states that, if initialized at a point such that , then Newton’s iterates satisfy . Thus, Newton’s region of quadratic convergence for self-concordant functions can be described as a “Newton decrement ball" . In particular by taking the barrier to be a self-concordant function we have now resolved Step (1) of the plan described in Section 5.3.1.
3.4 ν𝜈\nu-self-concordant barriers
We deal here with Step (2) of the plan described in Section 5.3.1. Given Theorem 5.3.5 we want to be as large as possible and such that
Since the Hessian of is the Hessian of , one has
Observe that, by first order optimality, one has which yields
immediately yields (5.6). In particular with the value of given in (5.8) the Newton’s method on initialized at will converge quadratically fast to .
It remains to verify that by iterating (5.8) one obtains a sequence diverging to infinity, and to estimate the rate of growth. Thus one needs to control . Luckily there is a natural class of functions for which one can control uniformly over . This is the set of functions such that
Thus a safe choice to increase the penalization parameter is . Note that the condition (5.9) can also be written as the fact that the function is -exp-concave, that is is concave. We arrive at the following definition.
is a -self-concordant barrier if it is a standard self-concordant function, and it is -exp-concave.
A key property of -self-concordant barriers is the following inequality:
see [Equation (4.2.17), Nesterov (2004a)]. More generally using (5.10) together with (5.5) one obtains
In the next section we describe a precise algorithm based on the ideas we developed above. As we will see one cannot ensure to be exactly on the central path, and thus it is useful to generalize the identity (5.7) for a point close to the central path. We do this as follows:
3.5 Path-following scheme
We can now formally describe and analyze the most basic IPM called the path-following scheme. Let be -self-concordant barrier for . Assume that one can find such that for some small value (we describe a method to find at the end of this subsection). Then for , let
The next theorem shows that after iterations of the path-following scheme one obtains an -optimal point.
The path-following scheme described above satisfies
We show that the iterates remain close to the central path . Precisely one can easily prove by induction that
Indeed using Theorem 5.3.5 and equation (5.12) one immediately obtains
where we used in the last inequality that and .
Observe that , which finally yields
At this point we still need to explain how one can get close to an intial point of the central path. This can be done with the following rather clever trick. Assume that one has some point . The observation is that is on the central path at for the problem where is replaced by . Now instead of following this central path as , one follows it as . Indeed for small enough the central paths for and for will be very close. Thus we iterate the following equations, starting with ,
A straightforward analysis shows that for , which corresponds to , one obtains a point such that . In other words one can initialize the path-following scheme with and .
3.6 IPMs for LPs and SDPs
There is one important issue that we overlooked so far. In most interesting cases LPs and SDPs come with equality constraints, resulting in a set of constraints with empty interior. From a theoretical point of view there is an easy fix, which is to reparametrize the problem as to enforce the variables to live in the subspace spanned by . This modification also has algorithmic consequences, as the evaluation of the Newton direction will now be different. In fact, rather than doing a reparametrization, one can simply search for Newton directions such that the updated point will stay in . In other words one has now to solve a convex quadratic optimization problem under linear equality constraints. Luckily using Lagrange multipliers one can find a closed form solution to this problem, and we refer to previous references for more details.
Chapter 6 Convex optimization and randomness
In this chapter we explore the interplay between optimization and randomness. A key insight, going back to Robbins and Monro (1951), is that first order methods are quite robust: the gradients do not have to be computed exactly to ensure progress towards the optimum. Indeed since these methods usually do many small steps, as long as the gradients are correct on average, the error introduced by the gradient approximations will eventually vanish. As we will see below this intuition is correct for non-smooth optimization (since the steps are indeed small) but the picture is more subtle in the case of smooth optimization (recall from Chapter 3 that in this case we take long steps).
We also note that the situation with a biased oracle is quite different, and we refer to d’Aspremont (2008); Schmidt et al. (2011) for some works in this direction.
The two canonical examples of a stochastic oracle in machine learning are as follows.
The second example is the one described in Section 1.1, where one wants to minimize . In this situation a stochastic oracle can be obtained by selecting uniformly at random and reporting .
This immediately yields a rate of convergence thanks to the following simple observation based on the tower rule:
Similarly, in the Euclidean and strongly convex case, one can directly generalize Theorem 3.4.1. Precisely we consider stochastic gradient descent (SGD), that is S-MD with , with time-varying step size , that is
2 Smooth stochastic optimization and mini-batch SGD
In the previous section we showed that, for non-smooth optimization, there is basically no cost for having a stochastic oracle instead of an exact oracle. Unfortunately one can show (see e.g. Tsybakov (2003)) that smoothness does not bring any acceleration for a general stochastic oracleWhile being true in general this statement does not say anything about specific functions/oracles. For example it was shown in Bach and Moulines (2013) that acceleration can be obtained for the square loss and the logistic loss.. This is in sharp contrast with the exact oracle case where we showed that gradient descent attains a rate (instead of for non-smooth), and this could even be improved to thanks to Nesterov’s accelerated gradient descent.
The next result interpolates between the for stochastic smooth optimization, and the for deterministic smooth optimization. We will use it to propose a useful modification of SGD in the smooth case. The proof is extracted from Dekel et al. (2012).
Using -smoothness, Cauchy-Schwarz (with for any ), and the 1-strong convexity of , one obtains
Observe that, using the same argument as to derive (4.9), one has
By summing this inequality from to one can easily conclude with the standard argument.
where are independent random variables (conditionally on ) obtained from repeated queries to the stochastic oracle. Assuming that is -smooth and that the stochastic oracle is such that , one can obtain a rate of convergence for mini-batch SGD with Theorem 6.2.1. Indeed one can apply this result with the modified stochastic oracle that returns , it satisfies
Thus one obtains that with calls to the (original) stochastic oracle, that is iterations of the mini-batch SGD, one has a suboptimality gap bounded by
Thus as long as one obtains, with mini-batch SGD and calls to the oracle, a point which is -optimal.
Mini-batch SGD can be a better option than basic SGD in at least two situations: (i) When the computation for an iteration of mini-batch SGD can be distributed between multiple processors. Indeed a central unit can send the message to the processors that estimates of the gradient at point have to be computed, then each processor can work independently and send back the estimate they obtained. (ii) Even in a serial setting mini-batch SGD can sometimes be advantageous, in particular if some calculations can be re-used to compute several estimated gradients at the same point.
3 Sum of smooth and strongly convex functions
Let us examine in more details the main example from Section 1.1. That is one is interested in the unconstrained minimization of
where are -smooth and convex functions, and is -strongly convex. Typically in machine learning can be as small as , while is of order of a constant. In other words the condition number can be as large as . Let us now compare the basic gradient descent, that is
where is drawn uniformly at random in (independently of everything else). Theorem 3.4.3 shows that gradient descent requires gradient computations (which can be improved to with Nesterov’s accelerated gradient descent), while Theorem 6.1.2 shows that SGD (with appropriate averaging) requires gradient computations. Thus one can obtain a low accuracy solution reasonably fast with SGD, but for high accuracy the basic gradient descent is more suitable. Can we get the best of both worlds? This question was answered positively in Le Roux et al. (2012) with SAG (Stochastic Averaged Gradient) and in Shalev-Shwartz and Zhang (2013a) with SDCA (Stochastic Dual Coordinate Ascent). These methods require only gradient computations. We describe below the SVRG (Stochastic Variance Reduced Gradient descent) algorithm from Johnson and Zhang (2013) which makes the main ideas of SAG and SDCA more transparent (see also Defazio et al. (2014) for more on the relation between these different methods). We also observe that a natural question is whether one can obtain a Nesterov’s accelerated version of these algorithms that would need only , see Shalev-Shwartz and Zhang (2013b); Zhang and Xiao (2014); Agarwal and Bottou (2014) for recent works on this question.
To obtain a linear rate of convergence one needs to make “big steps", that is the step-size should be of order of a constant. In SGD the step-size is typically of order because of the variance introduced by the stochastic oracle. The idea of SVRG is to “center" the output of the stochastic oracle in order to reduce the variance. Precisely instead of feeding into the gradient descent one would use where is a centering sequence. This is a sensible idea since, when and are close to the optimum, one should have that will have a small variance, and of course will also be small (note that by itself is not necessarily small). This intuition is made formal with the following lemma.
Let . By convexity of one has for any and in particular using (3.5) this yields which can be equivalently written as
On the other hand the computation of is expensive (it requires gradient computations), and thus the centering sequence should be updated more rarely than the main sequence. These ideas lead to the following epoch-based algorithm.
where is drawn uniformly at random (and independently of everything else) in . Also let
which clearly implies the theorem. To simplify the notation in the following we drop the dependency on , that is we want to show that
We start as for the proof of Theorem 3.4.3 (analysis of gradient descent for smooth and strongly convex functions) with
and thus plugging this into (6.2) together with (6.3) one obtains
Summing the above inequality over yields
Noting that and that by -strong convexity one has , one can rearrange the above display to obtain
Using that and finally yields (6.1) which itself concludes the proof.
4 Random coordinate descent
where is drawn uniformly at random from (and independently of everything else).
Thus using Theorem 6.1.1 (with , that is S-MD being SGD) one immediately obtains the following result.
Somewhat unsurprisingly RCD requires times more iterations than gradient descent to obtain the same accuracy. In the next section, we will see that this statement can be greatly improved by taking into account directional smoothness.
If is twice differentiable then this is equivalent to . In particular, since the maximal eigenvalue of a matrix is upper bounded by its trace, one can see that the directional smoothness implies that is -smooth with . We now study the following “aggressive" RCD, where the step-sizes are of order of the inverse smoothness:
Furthermore we study a more general sampling distribution than uniform, precisely for we assume that is drawn (independently) from the distribution defined by
This algorithm was proposed in Nesterov (2012), and we denote it by RCD(). Observe that, up to a preprocessing step of complexity , one can sample from in time .
The following rate of convergence is derived in Nesterov (2012), using the dual norms defined by
Recall from Theorem 3.2.1 that in this context the basic gradient descent attains a rate of where (see the discussion above). Thus we see that RCD() greatly improves upon gradient descent for functions where is of order of . Indeed in this case both methods attain the same accuracy after a fixed number of iterations, but the iterations of coordinate descent are potentially much cheaper than the iterations of gradient descent.
Thus putting together the above calculations one obtains
The proof can be concluded with similar computations than for Theorem 3.2.1.
We discussed above the specific case of . Both and also have an interesting behavior, and we refer to Nesterov (2012) for more details. The latter paper also contains a discussion of high probability results and potential acceleration à la Nesterov. We also refer to Richtárik and Takác (2012) for a discussion of RCD in a distributed setting.
4.2 RCD for smooth and strongly convex optimization
If in addition to directional smoothness one also assumes strong convexity, then RCD attains in fact a linear rate.
By strong convexity, Hölder’s inequality, and an elementary calculation,
which concludes the proof by taking .
In the proof of Theorem 6.4.2 we showed that
The proof is concluded with straightforward calculations.
5 Acceleration by randomization for saddle points
Using S-SP-MD we revisit the examples of Section 5.2.4 and Section 5.2.4. In both cases one has (with being the column of ), and thus and .
Matrix games. Here and . Thus there is a quite natural stochastic oracle:
Unfortunately this last term can be . However it turns out that one can do a more careful analysis of mirror descent in terms of local norms, which allows to prove that the “local variance" is dimension-free. We refer to Bubeck and Cesa-Bianchi (2012) for more details on these local norms, and to Clarkson et al. (2012) for the specific details in the linear classification situation.
6 Convex relaxation and randomized rounding
In this section we briefly discuss the concept of convex relaxation, and the use of randomization to find approximate solutions. By now there is an enormous literature on these topics, and we refer to Barak (2014) for further pointers.
Viewing as the (weighted) adjacency matrix of a graph, one can rewrite (6.6) as follows, using the graph Laplacian where is the diagonal matrix with entries ,
It turns out that this optimization problem is -hard, that is the existence of a polynomial time algorithm to solve (6.7) would prove that . The combinatorial difficulty of this problem stems from the hypercube constraint. Indeed if one replaces by the Euclidean sphere, then one obtains an efficiently solvable problem (it is the problem of computing the maximal eigenvalue of ).
We show now that, while (6.7) is a difficult optimization problem, it is in fact possible to find relatively good approximate solutions by using the power of randomization. Let be uniformly drawn on the hypercube , then clearly
Next we show that one can obtain an even better approximation ratio by combining the power of convex optimization and randomization. This approach was pioneered by Goemans and Williamson (1995). The Goemans-Williamson algorithm is based on the following inequality
The proof of this result is based on the following elementary geometric lemma.
We can now get to the proof of Theorem 6.6.1.
and in particular for , . Thus, using Lemma 6.6.2, and the facts that and (see the proof of Lemma 6.6.2), one has
Theorem 6.6.1 depends on the form of the Laplacian (insofar as (6.8) was used). We show next a result from Nesterov (1997) that applies to any positive semi-definite matrix, at the expense of the constant of approximation. Precisely we are now interested in the following optimization problem:
Finally one can conclude using the fact if then . This can be seen by writing , , and thus
In other words is a Gram-matrix and, thus it is positive semi-definite.
7 Random walk based methods
Randomization naturally suggests itself in the center of gravity method (see Section 2.1), as a way to circumvent the exact calculation of the center of gravity. This idea was proposed and developed in Bertsimas and Vempala (2004). We give below a condensed version of the main ideas of this paper.
Thus if one can ensure that is in (near) isotropic position, and is small (say smaller than ), then the randomized center of gravity method (which replaces by ) will converge at the same speed than the original center of gravity method.
Let us now consider the issue of putting in near-isotropic position. Let . Rudelson (1999) showed that as long as , one has with high probability (say at least probability ) that the set is in near-isotropic position.
Thus it only remains to explain how to sample from a near-isotropic convex set . This is where random walk ideas come into the picture. The hit-and-run walkOther random walks are known for this problem but hit-and-run is the one with the sharpest theoretical guarantees. Curiously we note that one of those walks is closely connected to projected gradient descent, see Bubeck et al. (2015a). is described as follows: at a point , let be a line that goes through in a direction taken uniformly at random, then move to a point chosen uniformly at random in . Lovász (1998) showed that if the starting point of the hit-and-run walk is chosen from a distribution “close enough" to the uniform distribution on , then after steps the distribution of the last point is away (in total variation) from the uniform distribution on . In the randomized center of gravity method one can obtain a good initial distribution for by using the distribution that was obtained for . In order to initialize the entire process correctly we start here with (in Section 2.1 we used ), and thus we also have to use a separation oracle at iterations where , just like we did for the ellipsoid method (see Section 2.2).
Wrapping up the above discussion, we showed (informally) that to attain an -optimal point with the randomized center of gravity method one needs: iterations, each iterations requires random samples from (in order to put it in isotropic position) as well as a call to either the separation oracle or the first order oracle, and each sample costs steps of the random walk. Thus overall one needs calls to the separation oracle and the first order oracle, as well as steps of the random walk.