Information-theoretic lower bounds on the oracle complexity of stochastic convex optimization
Alekh Agarwal, Peter L. Bartlett, Pradeep Ravikumar, Martin J. Wainwright
Introduction
Convex optimization forms the backbone of many algorithms for statistical learning and estimation. Given that many statistical estimation problems are large-scale in nature—with the problem dimension and/or sample size being large—it is essential to make efficient use of computational resources. Stochastic optimization algorithms are an attractive class of methods, known to yield moderately accurate solutions in a relatively short time . Given the popularity of such stochastic optimization methods, understanding the fundamental computational complexity of stochastic convex optimization is thus a key issue for large-scale learning. A large body of literature is devoted to obtaining rates of convergence of specific procedures for various classes of convex optimization problems. A typical outcome of such analysis is an upper bound on the error—for instance, gap to the optimal cost—as a function of the number of iterations. Such analyses have been performed for many standard optimization algorithms, among them gradient descent, mirror descent, interior point programming, and stochastic gradient descent, to name a few. We refer the reader to various standard texts on optimization (e.g., ) for further details on such results.
On the other hand, there has been relatively little study of the inherent complexity of convex optimization problems. To the best of our knowledge, the first formal study in this area was undertaken in the seminal work of Nemirovski and Yudin , hereafter referred to as NY. One obstacle to a classical complexity-theoretic analysis, as these authors observed, is that of casting convex optimization problems in a Turing Machine model. They avoided this problem by instead considering a natural oracle model of complexity, in which at every round the optimization procedure queries an oracle for certain information on the function being optimized. This information can be either noiseless or noisy, depending on whether the goal is to lower bound the oracle complexity of deterministic or stochastic optimization algorithms. Working within this framework, the authors obtained a series of lower bounds on the computational complexity of convex optimization problems, both in deterministic and stochastic settings. In addition to the original text NY , we refer the interested reader to the book by Nesterov , and the lecture notes by Nemirovski for further background.
In this paper, we consider the computational complexity of stochastic convex optimization within this oracle model. In particular, we improve upon the work of NY for stochastic convex optimization in two ways. First, our lower bounds have an improved dependence on the dimension of the space. In the context of statistical estimation, these bounds show how the difficulty of the estimation problem increases with the number of parameters. Second, our techniques naturally extend to give sharper results for optimization over simpler function classes. We show that the complexity of optimization for strongly convex losses is smaller than that for convex, Lipschitz losses. Third, we show that for a fixed function class, if the set of optimizers is assumed to have special structure such as sparsity, then the fundamental complexity of optimization can be significantly smaller. All of our proofs exploit a new notion of the discrepancy between two functions that appears to be natural for optimization problems. They involve a reduction from a statistical parameter estimation problem to the stochastic optimization problem, and an application of information-theoretic lower bounds for the estimation problem. We note that special cases of the first two results in this paper appeared in the extended abstract , and that a related study was independently undertaken by Raginsky and Rakhlin .
The remainder of this paper is organized as follows. We begin in Section 2 with background on oracle complexity, and a precise formulation of the problems addressed in this paper. Section 3 is devoted to the statement of our main results, and discussion of their consequences. In Section 4, we provide the proofs of our main results, which all exploit a common framework of four steps. More technical aspects of these proofs are deferred to the appendices.
Background and problem formulation
We begin by introducing background on the oracle model of convex optimization, and then turn to a precise specification of the problem to be studied.
where is the method’s query at time . Note that by definition of as a minimizing argument, this error is a non-negative quantity.
2 Stochastic first-order oracles
Stochastic gradient methods are a widely used class of algorithms that can be understood as operating based on information provided by a stochastic first-order oracle. As a particular example, consider a function of the separable form , where each is differentiable. Functions of this form arise very frequently in statistical problems, where each term corresponds to a different sample and the overall cost function is some type of statistical loss (e.g., maximum likelihood, support vector machines, boosting etc.) The natural stochastic gradient method for this problem is to choose an index uniformly at random, and then to return the pair . Taking averages over the randomly chosen index yields , so that is an unbiased estimate of , with an analogous unbiased property holding for the gradient of .
3 Function classes of interest
Our first class consists of convex Lipschitz functions:
If we consider the case of a differentiable function , the unbiasedness condition in Definition 1 implies that
A second function class consists of strongly convex functions, defined as follows:
In order to establish this inequality, we note that strong convexity condition with implies that
As a third example, we study the oracle complexity of optimization over the class of convex functions that have sparse minimizers. This class of functions is well-motivated, since a large body of statistical work has studied the estimation of vectors, matrices and functions under various types of sparsity constraints. A common theme in this line of work is that the ambient dimension enters only logarithmically, and so has a mild effect. Consequently, it is natural to investigate whether the complexity of optimization methods also enjoys such a mild dependence on ambient dimension under sparsity assumptions.
Main results and their consequences
We begin by analyzing the minimax oracle complexity of optimization for the class of bounded and convex Lipschitz functions from Definition 2.
In general, our lower bounds cannot be improved, and hence specify the optimal minimax oracle complexity. We consider here some examples to illustrate their sharpness. Throughout we assume that is large enough to ensure that the term attains the lower bound and not the term. (This condition is reasonable given our goal of understanding the rate as increases, as opposed to the transient behavior over the first few iterations.)
As mentioned previously, the dimension-independent lower bound for the case was demonstrated in Chapter 5 of NY, and shown to be optimalThere is an additional logarithmic factor in the upper bounds for . since it is achieved using mirror descent with the prox-function . For the case of , the lower bounds are also unimprovable, since they are again achieved (up to constant factors) by stochastic gradient descent. See Appendix C for further details on these matching upper bounds.
Even though the results have been stated in a first-order stochastic oracle model, they actually hold in a stronger sense. Let denote the -order derivative of evaluated at , when it exists. With this notation, our results apply to an oracle that responds with a random function such that
along with appropriately bounded second moments of all the derivatives. Consequently, higher-order gradient information cannot improve convergence rates in a worst-case setting. Indeed, the result continues to hold even for the significantly stronger oracle that responds with a random function that is a noisy realization of the true function. In this sense, our result is close in spirit to a statistical sample complexity lower bound. Our proof technique is based on constructing a “packing set” of functions, and thus has some similarity to techniques used in statistical minimax analysis (e.g., ) and learning theory (e.g., ). A significant difference, as will be shown shortly, is that the metric of interest for optimization is very different than those typically studied in statistical minimax theory.
2 Oracle complexity for strongly convex Lipschitz functions
We now turn to the statement of lower bounds over the class of Lipschitz and strongly convex functions from Definition 3. In all these statements, we assume that , as is required for the definition of to be sensible.
3 Oracle complexity for convex Lipschitz functions with sparse optima
Finally, we turn to the oracle complexity of optimization over the class from Definition 4.
If for some (so that ), then this bound is sharp up to constant factors. In particular, suppose that we use mirror descent based on the norm with . As we discuss in more detail in Appendix C, it can be shown that this technique will achieve a solution accurate to within iterations; this achievable result matches our lower bound (14) up to constant factors under the assumed scaling . To the best of our knowledge, Theorem 3 provides the first tight lower bound on the oracle complexity of sparse optimization.
Proofs of results
We now turn to the proofs of our main results. We begin in Section 4.1 by outlining the framework and establishing some basic results on which our proofs are based. Sections 4.2 through 4.4 are devoted to the proofs of Theorems 1 through 3 respectively.
We begin by establishing a basic set of results that are exploited in the proofs of the main results. At a high-level, our main idea is to show that the problem of convex optimization is at least as hard as estimating the parameters of Bernoulli variables—that is, the biases of independent coins. In order to perform this embedding, for a given error tolerance , we start with an appropriately chosen subset of the vertices of a -dimensional hypercube, each of which corresponds to some values of the Bernoulli parameters. For a given function class, we then construct a “difficult” subclass of functions that are indexed by these vertices of the hypercube. We then show that being able to optimize any function in this subclass to -accuracy requires identifying the hypercube vertex. This is a multiway hypothesis test based on the observations provided by queries to the stochastic oracle, and we apply Fano’s inequality or Le Cam’s bound to lower bound the probability of error. In the remainder of this section, we provide more detail on each of steps involved in this embedding.
Our first step is to construct a subclass of functions that we use to derive lower bounds. Any such subclass is parametrized by a subset of the hypercube, chosen as follows. Recalling that denotes the Hamming metric, we let be a subset of the vertices of the hypercube such that
meaning that is a -packing in the Hamming norm. It is a classical fact (e.g., ) that one can construct such a set with cardinality .
Based on these functions and the packing set , we define the function class
Note that contains a total of functions by construction, and as mentioned previously, our choices of the base functions etc. will ensure that . We demonstrate specific choices of the class in the proofs of Theorems 1 through 3 to follow.
1.2 Optimizing well is equivalent to function identification
This discrepancy measure is non-negative, symmetric in its arguments, and satisfies if and only if , so that we may refer to it as a premetric. (It does not satisfy the triangle inequality nor the condition that if and only if , both of which are required for to be a metric.)
Given the subclass , we quantify how densely it is packed with respect to the premetric using the quantity
We denote this quantity by when the class is clear from the context. We now state a simple result that demonstrates the utility of maintaining a separation under among functions in .
Re-arranging yields the inequality , from which the claim (20) follows.
Suppose that for some fixed but unknown function , some method is allowed to make queries to an oracle with information function , thereby obtaining the information sequence
Our next lemma shows that if the method achieves a low minimax error over the class , then one can use its output to construct a hypothesis test that returns the true parameter at least of the time. (In this statement, we recall the definition (2) of the minimax error in optimization.)
Suppose that based on the data , there exists a method that achieves a minimax error satisfying
We have thus shown that having a low minimax optimization error over implies that the vertex can be identified most of the time.
1.3 Oracle answers and coin tosses
We now describe stochastic first order oracles for which the samples can be related to coin tosses. In particular, we associate a coin with each dimension , and consider the set of coin bias vectors lying in the set
Given a particular function —or equivalently, vertex —we consider two different types of stochastic first-order oracles , defined as follows:
1.4 Lower bounds on coin-tossing
Finally, we use information-theoretic methods to lower bound the probability of correctly estimating the true parameter in our model. At each round of either Oracle A or Oracle B, we can consider a set of coin tosses, with an associated vector of parameters. At any round, the output of Oracle A can (at most) reveal the instantiation of a randomly chosen index, whereas Oracle B can at most reveal the entire vector . Our goal is to lower bound the probability of estimating the true parameter , based on a sequence of length . As noted previously in remarks following Theorem 1, this part of our proof exploits classical techniques from statistical minimax theory, including the use of Fano’s inequality (e.g., ) and Le Cam’s bound (e.g., ).
where the probability is taken over both randomness in the oracle and the choice of .
By Fano’s inequality , we have the lower bound
By the independent and identically distributed nature of the sampling model, we have
By chain rule for mutual information , we have
Since the subset is chosen independently of , we have , and so it suffices to upper bound the first term. By definition of conditional mutual information , we have
The reader might have observed that Fano’s inequality yields a non-trivial lower bound only when is large enough. Since depends on the dimension for our construction, we can apply the Fano lower bound only for large enough. Smaller values of can be lower bounded by reduction to the case ; here we state a simple lower bound for estimating the bias of a single coin, which is a straightforward application of Le Cam’s bounding technique . In this special case, we have , and we recall that the estimator takes values in .
Given a sample size and a parameter , let be i.i.d Bernoulli variables with parameter . Let be any test function based on these samples and returning an element of . Then for any , we have the lower bound
where inequality (i) follows from the calculation following Equation 26 (see proof of Lemma 3), and uses our assumption that . Putting together the pieces, we obtain a lower bound on the probability of error
Equipped with these tools, we are now prepared to prove our main results.
2 Proof of Theorem 1
Furthermore, we note that for any , we have
When then , so that this co-ordinate does not make a contribution to the discrepancy function . On the other hand, when , we have
Consequently, any such co-ordinate yields a contribution of to the discrepancy. Recalling our packing set (15) with separation in Hamming norm, we conclude that for any distinct within our packing set,
so that by definition of , we have established the lower bound .
Recalling that , making the substitution , and performing some algebra yields
where and are universal constants. Combined with Theorem 5.3.1 of NY (or by using the lower bound of Lemma 4 instead of Lemma 3), we conclude that this lower bound holds for all dimensions .
As before, for any distinct pair , we have the lower bound
Substituting yields the scaling for all , and a universal constant . Recalling that , we obtain the bound (10). Combining this with Theorem 5.3.1 of NY (or by using the lower bound of Lemma 4 instead of Lemma 3) gives the claim for all dimensions.
3 Proof of Theorem 2
We now turn to the proof of lower bounds on the oracle complexity of the class of strongly convex functions from Definition 3. In this case, we work with the following family of base functions, parametrized by a scalar :
A key ingredient of the proof is a uniform lower bound on the discrepancy between pairs of these functions:
Using an ensemble based on the base functions (30), we have
The proof of this lemma is provided in Appendix A. Let us now proceed to the proofs of the main theorem claims.
Case 1: First suppose that , in which case Lemma 5 yields the lower bound
Simplifying the above expression yields that for , we have the lower bound
Finally, we observe that and which gives . Substituting the above relations in the lower bound (32) gives the first term in the stated result for .
To obtain lower bounds for dimensions , we use an argument based on . For this special case, we consider and to be the two functions of the single coordinate coming out of definition (30). The packing set consists of only two elements now, corresponding to and . Specializing the result of Lemma 5 to this case, we see that the two functions are separated. Now we again apply Lemma 2 to get an upper bound on the error probability and Lemma 4 to get a lower bound, which gives the result for .
Case 2: On the other hand, suppose that . In this case, appealing to Lemma 5 gives us that for . Recalling that , we set the desired accuracy . From this point onwards, we mimic the proof of Theorem 1; doing so yields that for all , we have
corresponding to the second term in Theorem 1 for a universal constant .
Finally, the third and fourth terms are obtained just like Theorem 1 by checking the condition in the two cases above. Overall, this completes the proof for the case .
As with the proof of Theorem 1(b), we use Oracle B that returns -dimensional values and gradients in this case, with the base functions defined in equation 30. With this choice, we have the upper bound
Rearranging terms and substituting , we obtain for
for a universal constant . The stated result can now be attained by recalling and for and . For , the cases of and are identical up to constant factors in the lower bounds we state. This completes the proof for .
Finally, the case for involves similar modifications as part(a) by using the different expression for . Thus we have completed the proof of this theorem.
4 Proof of Theorem 3
We begin by constructing an appropriate subset of over which the Fano method can be applied. Let be a set of vectors, such that each satisfies
It can be shown that there exists such a packing set with |\mathcal{V}(k)|\geq\exp\big{(}\frac{k}{2}\log\frac{d-k}{k/2}\big{)} elements (e.g., see Lemma 5 in Raskutti et al. ).
For any , we define the function
In this definition, the quantity is a pre-factor to be chosen later, and is a given error tolerance. Observe that each function is convex, and Lipschitz with parameter with respect to the norm.
Central to the remainder of the proof is the function class . In particular, we need to control the discrepancy for this class. The following result, proven in Appendix B, provides a suitable lower bound:
Using Lemma 6, we may complete the proof of Theorem 3. Define the base functions
Consider Oracle B, which returns -dimensional gradients based on the function
Our next step is to use Fano’s inequality to lower bound the probability of error in the multiway testing problem associated with this stochastic oracle, following an argument similar to (but somewhat simpler than) the proof of Lemma 3. Fano’s inequality yields the lower bound
By Lemma 6 and our choice , we have
for a universal constant , where the second step uses the relation for . As long as , we have , which gives the result for . The result for follows Theorem 1(b) applied with , completing the proof.
Discussion
In this paper, we have studied the complexity of convex optimization within the stochastic first-order oracle model. We derived lower bounds for various function classes, including convex functions, strongly convex functions, and convex functions with sparse optima. As we discussed, our lower bounds are sharp in general, since there are matching upper bounds achieved by known algorithms, among them stochastic gradient descent and stochastic mirror descent. Our bounds also reveal various dimension-dependent and geometric aspects of the stochastic oracle complexity of convex optimization. An interesting aspect of our proof technique is the use of tools common in statistical minimax theory. In particular, our proofs are based on constructing packing sets, defined with respect to a pre-metric that measures how the degree of separation between the optima of different functions. We then leveraged information-theoretic techniques, in particular Fano’s inequality and its variants, in order to establish lower bounds.
There are various directions for future research. It would be interesting to consider the effect of memory constraints on the complexity of convex optimization, or to derive lower bounds for problems of distributed optimization. We suspect that the proof techniques developed in this paper may be useful for studying these related problems.
AA and PLB gratefully acknowledge partial support from NSF awards DMS-0707060 and DMS-0830410 and DARPA-HR0011-08-2-0002. AA was also supported in part by a Microsoft Research Fellowship. MJW and PR were partially supported by funding from the National Science Foundation (DMS-0605165, and DMS-0907632). In addition, MJW received funding from the Air Force Office of Scientific Research (AFOSR-09NL184). We also thank the anonymous reviewers for helpful suggestions, and corrections to our results and for pointing out the optimality of our bounds in the primal-dual norm setting.
Appendix A Proof of Lemma 5
Consequently, using the definition (30) of the base functions, some algebra yields the relations
Using these expressions for and , we obtain
A little calculation shows that constrained minimum of the univariate function over the interval is achieved at
where we have recalled that takes values in . Substituting the minimizing argument , we find that the minimum value is given by
Summing over all co-ordinates , we obtain
Here we begin by observing that for any two , we have
As in our previous calculation, the only coordinates that contribute to are the ones where , and for such coordinates, the function above is minimized at . Furthermore, the minimum value for any such coordinate is .
We split the remainder of our analysis into two cases: first, if we suppose that , or equivalently that , then equation (39) yields that
Combined with our earlier expression (38) for the single function infimum, we obtain that the discrepancy is given by
On the other hand, if we assume that , or equivalently that , then we obtain
Combined with our earlier expression (38) for the single function infimum, we obtain
where step (i) uses the bound . Noting that completes the proof of the lemma.
Appendix B Proof of Lemma 6
We now consider one of the individual terms arising in the definition (16) of the function . Using the relation (40), it can be written as
Let us consider the minimizer of the term in this summation. First, suppose that , in which case there are two possibilities.
If and neither nor is zero, then we must have , so that the minimum value of is achieved at .
Otherwise, suppose that and . In this case, we see from Equation (42) that it is equivalent to minimizing . Setting achieves the minimum value of .
In the remaining two cases, we have .
If , then the component is minimized at and the minimum value along the component is .
If , then the minimum value is , achieved at .
Consequently, accumulating all of these individual cases into a single expression, we obtain
Finally, combining equations (41) and (43) in the definition of , we find that
where the second equality follows since and have exactly non-zero elements each. Finally, since is an -packing set in Hamming distance, we have , which completes the proof.
Appendix C Upper bounds via mirror descent
This appendix is devoted to background on the family of mirror descent methods. We first describe the basic form of the algorithm and some known convergence results, before showing that different forms of mirror descent provide matching upper bounds for several of the lower bounds established in this paper, as discussed in the main text.
We assume that is a function of Legendre type , which implies that the conjugate dual is differentiable on its domain with \nabla\Phi^{*}=\big{(}\nabla\Phi\big{)}^{-1}. For a given proximal function, we let be the Bregman divergence induced by , given by
where is a stepsize. In case of stochastic optimization, is simply replaced by the noisy version .
A special case of this algorithm is obtained by choosing the proximal function , which is -strongly convex with respect to the Euclidean norm. The associated Bregman divergence is simply the Euclidean norm, so that the updates (45) correspond to a standard projected gradient descent method. If one receives only an unbiased estimate of the gradient , then this algorithm corresponds to a form of projected stochastic gradient descent. Moreover, other choices of the proximal function lead to different stochastic algorithms, as discussed below.
Note that this averaged convergence is a little different from the convergence of discussed in our lower bounds. In order to relate the two quantities, observe that by Jensen’s inequality
Consequently, based on mirror descent for rounds, we may set so as to obtain the same convergence bounds up to constant factors. In the following discussion, we assume this choice of for comparing the mirror descent upper bounds to our lower bounds.
C.2 Matching upper bounds
Now consider the form of mirror descent obtained by choosing the proximal function
For this case, we use mirror descent based on the proximal function with . Under the condition , a condition which holds in our lower bounds, we obtain
which matches the lower bound from Theorem 1(b) (we note that there is an additional log factor here just like the preceding discussion when which we ignore here).
In order to recover matching upper bounds in this case, we use the function from Equation (47) with . In this case, the resulting upper bound (46) on the convergence rate takes the form
since . Based on the conditions of Theorem 3, we are guaranteed that is -sparse, with every component bounded by in absolute value, so that , where the final inequality follows since . Substituting this upper bound back into Equation (48) yields
Note that whenever for some , then we have , in which case this upper bound matches the lower bound from Theorem 3 up to constant factors, as claimed.