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 xTx_{T} is the method’s query at time TT. Note that by definition of xf∗x^{*}_{f} 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 f(x)=1n∑i=1nhi(x)f(x)=\frac{1}{n}\sum_{i=1}^{n}h_{i}(x), where each hih_{i} is differentiable. Functions of this form arise very frequently in statistical problems, where each term ii 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 i∈{1,2,…,n}i\in\{1,2,\ldots,n\} uniformly at random, and then to return the pair (hi(x),∇hi(x))(h_{i}(x),\nabla h_{i}(x)). Taking averages over the randomly chosen index ii yields 1n∑i=1nhi(x)=f(x)\frac{1}{n}\sum_{i=1}^{n}h_{i}(x)=f(x), so that hi(x)h_{i}(x) is an unbiased estimate of f(x)f(x), with an analogous unbiased property holding for the gradient of hi(x)h_{i}(x).

3 Function classes of interest

Our first class consists of convex Lipschitz functions:

If we consider the case of a differentiable function ff, 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 α=1/2\alpha=1/2 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 dd 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 Fcv⁡\mathcal{F}_{\operatorname{\scriptsize{cv}}} 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 TT is large enough to ensure that the 1/T1/\sqrt{T} term attains the lower bound and not the L/144L/144 term. (This condition is reasonable given our goal of understanding the rate as TT increases, as opposed to the transient behavior over the first few iterations.)

As mentioned previously, the dimension-independent lower bound for the case p≥2p\geq 2 was demonstrated in Chapter 5 of NY, and shown to be optimalThere is an additional logarithmic factor in the upper bounds for p=Ω(log⁡d)p=\Omega(\log d). since it is achieved using mirror descent with the prox-function ∥⋅∥q\|\cdot\|_{q}. For the case of 1≤p<21\leq p<2, 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 ∇if(x)\nabla^{i}f(x) denote the ithi_{th}-order derivative of ff evaluated at xx, when it exists. With this notation, our results apply to an oracle that responds with a random function f^t\hat{f}_{t} 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 Fscv⁡\mathcal{F}_{\operatorname{\scriptsize{scv}}} from Definition 3. In all these statements, we assume that γ2≤4Ld−1/pr\gamma^{2}\leq\frac{4Ld^{-1/p}}{r}, as is required for the definition of Fscv⁡\mathcal{F}_{\operatorname{\scriptsize{scv}}} 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 Fsp⁡\mathcal{F}_{\operatorname{\scriptsize{sp}}} from Definition 4.

If k=O(d1−δ)k=\mathcal{O}(d^{1-\delta}) for some δ∈(0,1)\delta\in(0,1) (so that log⁡dk=Θ(log⁡d)\log\frac{d}{k}=\Theta(\log d)), then this bound is sharp up to constant factors. In particular, suppose that we use mirror descent based on the ∥⋅∥1+ε\|\cdot\|_{1+\varepsilon} norm with ε=2log⁡d/(2log⁡d−1)\varepsilon=2\log d/(2\log d-1). As we discuss in more detail in Appendix C, it can be shown that this technique will achieve a solution accurate to O(k2log⁡dT)\mathcal{O}(\sqrt{\frac{k^{2}\log d}{T}}) within TT iterations; this achievable result matches our lower bound (14) up to constant factors under the assumed scaling k=O(d1−δ)k=\mathcal{O}(d^{1-\delta}) . 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 dd independent coins. In order to perform this embedding, for a given error tolerance ϵ\epsilon, we start with an appropriately chosen subset of the vertices of a dd-dimensional hypercube, each of which corresponds to some values of the dd 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 ϵ\epsilon-accuracy requires identifying the hypercube vertex. This is a multiway hypothesis test based on the observations provided by TT 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 G⊆F\mathcal{G}\subseteq\mathcal{F} that we use to derive lower bounds. Any such subclass is parametrized by a subset V⊆{−1,+1}d\mathcal{V}\subseteq\{-1,+1\}^{d} of the hypercube, chosen as follows. Recalling that ΔH\Delta_{H} denotes the Hamming metric, we let V={α1,…,αM}\mathcal{V}=\{\alpha^{1},\ldots,\alpha^{M}\} be a subset of the vertices of the hypercube such that

meaning that V\mathcal{V} is a d4\frac{d}{4}-packing in the Hamming norm. It is a classical fact (e.g., ) that one can construct such a set with cardinality ∣V∣≥(2/e)d/2|\mathcal{V}|\geq(2/\sqrt{e})^{d/2}.

Based on these functions and the packing set V\mathcal{V}, we define the function class

Note that G(δ)\mathcal{G}(\delta) contains a total of ∣V∣|\mathcal{V}| functions by construction, and as mentioned previously, our choices of the base functions etc. will ensure that G(δ)⊆F\mathcal{G}(\delta)\subseteq\mathcal{F}. We demonstrate specific choices of the class G(δ)\mathcal{G}(\delta) 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 ρ(f,g)=0\rho(f,g)=0 if and only if xf∗=xg∗x^{*}_{f}=x^{*}_{g}, so that we may refer to it as a premetric. (It does not satisfy the triangle inequality nor the condition that ρ(f,g)=0\rho(f,g)=0 if and only if f=gf=g, both of which are required for ρ\rho to be a metric.)

Given the subclass G(δ)\mathcal{G}(\delta), we quantify how densely it is packed with respect to the premetric ρ\rho using the quantity

We denote this quantity by ψ(δ)\psi(\delta) when the class G\mathcal{G} is clear from the context. We now state a simple result that demonstrates the utility of maintaining a separation under ρ\rho among functions in G(δ)\mathcal{G}(\delta).

Re-arranging yields the inequality gβ(x~)−gβ(xβ∗)≥23 ψ(δ)g_{\beta}(\widetilde{x})-g_{\beta}(x^{*}_{\beta})\geq\frac{2}{3}\,\psi(\delta), from which the claim (20) follows.

Suppose that for some fixed but unknown function gα∗∈G(δ)g_{\alpha^{*}}\in\mathcal{G}(\delta), some method MT\mathcal{M}_{T} is allowed to make TT queries to an oracle with information function ϕ(⋅ ; gα∗)\phi(\cdot\,;\,g_{\alpha^{*}}), thereby obtaining the information sequence

Our next lemma shows that if the method MT\mathcal{M}_{T} achieves a low minimax error over the class G(δ)\mathcal{G}(\delta), then one can use its output to construct a hypothesis test that returns the true parameter α∗\alpha^{*} at least 2/32/3 of the time. (In this statement, we recall the definition (2) of the minimax error in optimization.)

Suppose that based on the data ϕ(x1T;gα∗)\phi(x_{1}^{T};g_{\alpha}^{*}), there exists a method MT\mathcal{M}_{T} that achieves a minimax error satisfying

We have thus shown that having a low minimax optimization error over G(δ)\mathcal{G}(\delta) implies that the vertex α∗∈V\alpha^{*}\in\mathcal{V} can be identified most of the time.

1.3 Oracle answers and coin tosses

We now describe stochastic first order oracles ϕ\phi for which the samples ϕ(x1T;gα)\phi(x_{1}^{T};g_{\alpha}) can be related to coin tosses. In particular, we associate a coin with each dimension i∈{1,2,…,d}i\in\{1,2,\ldots,d\}, and consider the set of coin bias vectors lying in the set

Given a particular function gα∈G(δ)g_{\alpha}\in\mathcal{G}(\delta)—or equivalently, vertex α∈V\alpha\in\mathcal{V}—we consider two different types of stochastic first-order oracles ϕ\phi, 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 α∗∈V\alpha^{*}\in\mathcal{V} in our model. At each round of either Oracle A or Oracle B, we can consider a set of dd coin tosses, with an associated vector θ∗=(12+α1∗δ,…,12+αd∗δ)\theta^{*}=(\frac{1}{2}+\alpha^{*}_{1}\delta,\ldots,\frac{1}{2}+\alpha^{*}_{d}\delta) of parameters. At any round, the output of Oracle A can (at most) reveal the instantiation bi∈{0,1}b_{i}\in\{0,1\} of a randomly chosen index, whereas Oracle B can at most reveal the entire vector (b1,b2,…,bd)(b_{1},b_{2},\ldots,b_{d}). Our goal is to lower bound the probability of estimating the true parameter α∗\alpha^{*}, based on a sequence of length TT. 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 α∗\alpha^{*}.

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 UU is chosen independently of α∗\alpha^{*}, we have I(α∗;U)=0I(\alpha^{*};U)=0, 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 ∣V∣|\mathcal{V}| is large enough. Since ∣V∣|\mathcal{V}| depends on the dimension dd for our construction, we can apply the Fano lower bound only for dd large enough. Smaller values of dd can be lower bounded by reduction to the case d=1d=1; 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 V={1/2+δ,1/2−δ}\mathcal{V}=\left\{1/2+\delta,1/2-\delta\right\}, and we recall that the estimator α^(MT)\widehat{\alpha}(\mathcal{M}_{T}) takes values in V\mathcal{V}.

Given a sample size T≥1T\geq 1 and a parameter α∗∈V\alpha^{*}\in\mathcal{V}, let {X1,…,XT}\{X_{1},\ldots,X_{T}\} be TT i.i.d Bernoulli variables with parameter α∗\alpha^{*}. Let α^\widehat{\alpha} be any test function based on these samples and returning an element of V\mathcal{V}. Then for any δ∈(0,1/4]\delta\in(0,1/4], 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 δ∈(0,1/4]\delta\in(0,1/4]. 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 α≠β\alpha\neq\beta, we have

When αi=βi\alpha_{i}=\beta_{i} then xα(i)=xβ(i)=−αi/2x_{\alpha}(i)=x_{\beta}(i)=-\alpha_{i}/2, so that this co-ordinate does not make a contribution to the discrepancy function ρ(gα,gβ)\rho(g_{\alpha},g_{\beta}). On the other hand, when αi≠βi\alpha_{i}\neq\beta_{i}, we have

Consequently, any such co-ordinate yields a contribution of 2cδ/d2c\delta/d to the discrepancy. Recalling our packing set (15) with d/4d/4 separation in Hamming norm, we conclude that for any distinct α≠β\alpha\neq\beta within our packing set,

so that by definition of ψ\psi, we have established the lower bound ψ(δ)≥cδ2\psi(\delta)\geq\frac{c\delta}{2}.

Recalling that c=L2c=\frac{L}{2}, making the substitution δ=18ϵc=36ϵL\delta=\frac{18\epsilon}{c}=\frac{36\epsilon}{L}, and performing some algebra yields

where c0c_{0} and c1c_{1} 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 dd.

As before, for any distinct pair α,β∈V\alpha,\beta\in\mathcal{V}, we have the lower bound

Substituting δ=18ϵ/c\delta=18\epsilon/c yields the scaling ϵ≥c0 cT\epsilon\geq c_{0}\,\frac{c}{\sqrt{T}} for all d≥11d\geq 11, ϵ≤c/72\epsilon\leq c/72 and a universal constant c0c_{0}. Recalling that c=Ld1−1/pc=Ld^{1-1/p}, 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 θ∈[0,1)\theta\in[0,1):

A key ingredient of the proof is a uniform lower bound on the discrepancy ρ\rho 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 1−θ≥4δ/(1+2δ)1-\theta\geq 4\delta/(1+2\delta), in which case Lemma 5 yields the lower bound

Simplifying the above expression yields that for d≥11d\geq 11, we have the lower bound

Finally, we observe that L=crL=cr and γ2=(1−θ)c/(4d)\gamma^{2}=(1-\theta)c/(4d) which gives 1−θ=4drγ2/L1-\theta=4dr\gamma^{2}/L. Substituting the above relations in the lower bound (32) gives the first term in the stated result for d≥11d\geq 11.

To obtain lower bounds for dimensions d<11d<11, we use an argument based on d=1d=1. For this special case, we consider f+f^{+} and f−f^{-} to be the two functions of the single coordinate coming out of definition (30). The packing set V\mathcal{V} consists of only two elements now, corresponding to α=1\alpha=1 and α=−1\alpha=-1. Specializing the result of Lemma 5 to this case, we see that the two functions are 2cδ2r2/(1−θ)2c\delta^{2}r^{2}/(1-\theta) 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 d≤11d\leq 11.

Case 2: On the other hand, suppose that 1−θ≤4δ/(1+2δ)1-\theta\leq 4\delta/(1+2\delta). In this case, appealing to Lemma 5 gives us that ρ(gα,β)≥cδr2/4\rho(g_{\alpha},\beta)\geq c\delta r^{2}/4 for α≠β∈V\alpha\neq\beta\in\mathcal{V}. Recalling that L=crL=cr, we set the desired accuracy ϵ:=cδr2/36=Lδr/36\epsilon:=c\delta r^{2}/36=L\delta r/36. From this point onwards, we mimic the proof of Theorem 1; doing so yields that for all δ∈(0,1/4)\delta\in(0,1/4), we have

corresponding to the second term in Theorem 1 for a universal constant c0c_{0}.

Finally, the third and fourth terms are obtained just like Theorem 1 by checking the condition δ<1/4\delta<1/4 in the two cases above. Overall, this completes the proof for the case p=1p=1.

As with the proof of Theorem 1(b), we use Oracle B that returns dd-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 ϵ=cδ2r218(1−θ)\epsilon=\frac{c\delta^{2}r^{2}}{18(1-\theta)}, we obtain for d≥11d\geq 11

for a universal constant c0c_{0}. The stated result can now be attained by recalling c=Ld1−1/p/rc=Ld^{1-1/p}/r and γ2=Ld−1/p(1−θ)/r\gamma^{2}=Ld^{-1/p}(1-\theta)/r for 1−θ≥4δ/(1+2δ)1-\theta\geq 4\delta/(1+2\delta) and d≥11d\geq 11. For d<11d<11, the cases of p>2p>2 and p=1p=1 are identical up to constant factors in the lower bounds we state. This completes the proof for 1−θ≥4δ/(1+2δ)1-\theta\geq 4\delta/(1+2\delta).

Finally, the case for 1−θ<4δ/(1+2δ)1-\theta<4\delta/(1+2\delta) involves similar modifications as part(a) by using the different expression for ρ(gα,gβ)\rho(g_{\alpha},g_{\beta}). Thus we have completed the proof of this theorem.

4 Proof of Theorem 3

We begin by constructing an appropriate subset of Fsp⁡(k)\mathcal{F}_{\operatorname{\scriptsize{sp}}}(k) over which the Fano method can be applied. Let V(k):={α1,…,αM}\mathcal{V}(k):=\{\alpha^{1},\ldots,\alpha^{M}\} be a set of vectors, such that each αj∈{−1,0,+1}d\alpha^{j}\in\{-1,0,+1\}^{d} 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 α∈V(k)\alpha\in\mathcal{V}(k), we define the function

In this definition, the quantity c>0c>0 is a pre-factor to be chosen later, and δ∈(0,14]\delta\in(0,\frac{1}{4}] is a given error tolerance. Observe that each function gα∈G(δ;k)g_{\alpha}\in\mathcal{G}(\delta;k) is convex, and Lipschitz with parameter cc with respect to the ∥⋅∥∞\|\cdot\|_{\infty} norm.

Central to the remainder of the proof is the function class G(δ;k):={gα,  α∈V(k)}\mathcal{G}(\delta;k):=\{g_{\alpha},\;\alpha\in\mathcal{V}(k)\}. In particular, we need to control the discrepancy ψ(δ;k):=ψ(G(δ;k))\psi(\delta;k):=\psi(\mathcal{G}(\delta;k)) 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 dd-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 c=L/3c=L/3, we have

for a universal constant c0c_{0}, where the second step uses the relation δ=108ϵLkr\delta=\frac{108\epsilon}{Lkr} for k,d≥11k,d\geq 11. As long as k≤⌊d/2⌋k\leq\lfloor d/2\rfloor, we have log⁡d−kk/2=Θ(log⁡dk)\log\frac{d-k}{k/2}=\Theta\left(\log\frac{d}{k}\right), which gives the result for k,d≥11k,d\geq 11. The result for k,d≤11k,d\leq 11 follows Theorem 1(b) applied with p=∞p=\infty, 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 fi+f^{+}_{i} and fi−f^{-}_{i}, we obtain

A little calculation shows that constrained minimum of the univariate function hih_{i} over the interval [−r,r][-r,r] is achieved at

where we have recalled that αi\alpha_{i} takes values in {−1,+1}\{-1,+1\}. Substituting the minimizing argument x∗(i)x^{*}(i), we find that the minimum value is given by

Summing over all co-ordinates i∈{1,2,…,d}i\in\{1,2,\ldots,d\}, we obtain

Here we begin by observing that for any two α,β∈V\alpha,\beta\in\mathcal{V}, we have

As in our previous calculation, the only coordinates that contribute to ρ(gα,gβ)\rho(g_{\alpha},g_{\beta}) are the ones where αi≠βi\alpha_{i}\neq\beta_{i}, and for such coordinates, the function above is minimized at x∗(i)=0x^{*}(i)=0. Furthermore, the minimum value for any such coordinate is (1+3θ)cr2/(2d)(1+3\theta)cr^{2}/(2d).

We split the remainder of our analysis into two cases: first, if we suppose that 1−θ1+θ≥2δ\frac{1-\theta}{1+\theta}\geq 2\delta, or equivalently that 1−θ≥4δ/(1+2δ)1-\theta\geq 4\delta/(1+2\delta), 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 1−θ1+θ<2δ\frac{1-\theta}{1+\theta}<2\delta, or equivalently that 1−θ<4δ/(1+2δ)1-\theta<4\delta/(1+2\delta), then we obtain

Combined with our earlier expression (38) for the single function infimum, we obtain

where step (i) uses the bound 1−θ<2δ(1+θ)1-\theta<2\delta(1+\theta). Noting that θ≥0\theta\geq 0 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 gαg_{\alpha}. Using the relation (40), it can be written as

Let us consider the minimizer of the ithi^{th} term in this summation. First, suppose that αi≠βi\alpha_{i}\neq\beta_{i}, in which case there are two possibilities.

If αi≠βi\alpha_{i}\neq\beta_{i} and neither αi\alpha_{i} nor βi\beta_{i} is zero, then we must have αi+βi=0\alpha_{i}+\beta_{i}=0, so that the minimum value of 2r2r is achieved at x(i)=0x(i)=0.

Otherwise, suppose that αi≠0\alpha_{i}\neq 0 and βi=0\beta_{i}=0. In this case, we see from Equation (42) that it is equivalent to minimizing αix(i)+∣x(i)∣\alpha_{i}x(i)+|x(i)|. Setting x(i)=−αix(i)=-\alpha_{i} achieves the minimum value of 2r2r.

In the remaining two cases, we have αi=βi\alpha_{i}=\beta_{i}.

If αi=βi≠0\alpha_{i}=\beta_{i}\neq 0, then the component is minimized at x(i)=−αirx(i)=-\alpha_{i}r and the minimum value along the component is 2r(1−δ)2r(1-\delta).

If αi=βi=0\alpha_{i}=\beta_{i}=0, then the minimum value is 2r2r, achieved at x(i)=0x(i)=0.

Consequently, accumulating all of these individual cases into a single expression, we obtain

Finally, combining equations (41) and (43) in the definition of ρ\rho, we find that

where the second equality follows since α\alpha and β\beta have exactly kk non-zero elements each. Finally, since V\mathcal{V} is an k/2k/2-packing set in Hamming distance, we have ΔH(α,β)≥k/2\Delta_{H}(\alpha,\beta)\geq k/2, 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 Φ\Phi is a function of Legendre type , which implies that the conjugate dual Φ∗\Phi^{*} is differentiable on its domain with \nabla\Phi^{*}=\big{(}\nabla\Phi\big{)}^{-1}. For a given proximal function, we let DΦD_{\Phi} be the Bregman divergence induced by Φ\Phi, given by

where ηt>0\eta_{t}>0 is a stepsize. In case of stochastic optimization, ∇f(xt)\nabla f(x_{t}) is simply replaced by the noisy version z^(xt)\widehat{z}(x_{t}).

A special case of this algorithm is obtained by choosing the proximal function Φ(x)=12∥x∥22\Phi(x)=\frac{1}{2}\|x\|_{2}^{2}, which is 11-strongly convex with respect to the Euclidean norm. The associated Bregman divergence DΦ(x,y)=12∥x−y∥22D_{\Phi}(x,y)=\frac{1}{2}\|x-y\|_{2}^{2} 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 ∇f(xt)\nabla f(x_{t}), 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 xTx_{T} discussed in our lower bounds. In order to relate the two quantities, observe that by Jensen’s inequality

Consequently, based on mirror descent for T−1T-1 rounds, we may set xT=1T−1∑t=1T−1xtx_{T}=\frac{1}{T-1}\sum_{t=1}^{T-1}x_{t} so as to obtain the same convergence bounds up to constant factors. In the following discussion, we assume this choice of xTx_{T} 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 Φa\Phi_{a} with a=qa=q. Under the condition ∥x∗∥∞≤1\|x^{*}\|_{\infty}\leq 1, 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 p=O(log⁡d)p=\mathcal{O}(\log d) which we ignore here).

In order to recover matching upper bounds in this case, we use the function Φa\Phi_{a} from Equation (47) with a=2log⁡d2log⁡d−1a=\frac{2\log d}{2\log d-1}. In this case, the resulting upper bound (46) on the convergence rate takes the form

since 1a−1=2log⁡d−1\frac{1}{a-1}=2\log d-1. Based on the conditions of Theorem 3, we are guaranteed that x∗x^{*} is kk-sparse, with every component bounded by 11 in absolute value, so that ∥x∗∥a2≤k2/a≤k2\|x^{*}\|_{a}^{2}\leq k^{2/a}\leq k^{2}, where the final inequality follows since a>1a>1. Substituting this upper bound back into Equation (48) yields

Note that whenever k=O(d1−δ)k=\mathcal{O}(d^{1-\delta}) for some δ>0\delta>0, then we have log⁡d=Θ(log⁡dk)\log d=\Theta(\log\frac{d}{k}), in which case this upper bound matches the lower bound from Theorem 3 up to constant factors, as claimed.

References