Lower Bounds for Finding Stationary Points II: First-Order Methods
Yair Carmon, John C. Duchi, Oliver Hinder, Aaron Sidford
Introduction
In Part I of this series , we establish the complexity of finding an -stationary point (1) for algorithms that, at a query point , have access to all derivatives of . In contrast, in this paper we focus on first-order methods, which only query function values and gradients.
First-order methods are important in large-scale optimization for many reasons. Perhaps the two most salient are that each iteration is often inexpensive, and that on many problems, the number of iterations grows slowly (or not at all) with the problem dimension . From a theoretical perspective, the latter property is captured by dimension-free convergence rates, where the worst case iteration count depends polynomially on the desired accuracy and measures of function regularity but has no explicit dependence on . In non-convex optimization problems, regularity often comes by assuming bounded function value at the initial point , i.e. for some , and that is -Lipschitz continuous. Under these conditions, classical gradient descent finds an -stationary point in iterations , a dimension-free guarantee.
Developing first-order methods for finding stationary points of non-convex functions with improved dimension-free rates of convergence is an area of active research . Under the additional assumption of Lipschitz second derivatives, we and Agarwal et al. propose randomized first-order methods with nearly dimension free rate (ignoring other problem-dependant constants). In a later paper , we propose a deterministic accelerated gradient-based method with complexity , and under the further assumption of that has Lipschitz third derivatives, we show the same method attains rates of . This raises the main question we address in this paper: how much further can this dependence can be improved, and what Lipschitz continuity assumptions are necessary?
In Table 1 we summarize our results, along with corresponding known upper bounds. We establish lower bounds on the worst-case oracle complexity of finding -stationary points, where algorithms may access only through queries to an information oracle, that returns the value and some number of (or potentially all) derivatives of at the queried point. A lower bound means that for every algorithm , there exists a function in the allowed function class (e.g. functions with and -Lipschitz gradient) for which requires at least oracle queries before returning an -stationary point of .
In Part I of this series we prove that no algorithm, even one given all derivatives of at each iteration, can improve on the rate of gradient descent for the class of functions with bounded initial value and Lipschitz continuous gradient. Therefore, in distinction with the convex case, acceleration of gradient descent for non-convex optimization fundamentally depends on higher-order smoothness assumptions. We further show that, for the class of functions with th order Lipschitz derivatives, no method can improve the rate achieved by a th-order method . However, this does not get at the crux of the issue we consider here—what is the best possible rate for first-order methods, given that higher-order derivatives are Lipschitz?
cubic-regularized Newton’s method first-order methods first-order methods gradient descent Thus, we establish two separations. First, no deterministic first-order method can achieve the rate of convergence of Newton’s method. Second, the rate we achieve requires the assumption of Lipschitz third derivatives, as first-order methods assuming only Lipschitz Hessian must compute at least function values and gradients to find an -stationary point. We also show that the optimal rate for finding -stationary points of convex functions with bounded initial value (i.e. ) and -Lipschitz gradient is . Given a bound where , as is standard for convex optimization, the optimal rate is . The two rates are not directly comparable. Finding stationary points is thus fundamentally easier for convex functions.
The starting point of our development is Nesterov’s [18, § 2.1.2] “worst function in the world,”
Throughout, we use Pi. to reference an item of Part I of this sequence , as we build off of many ideas there. In Section 2 we briefly summarize our framework (Sections Pi.LABEL:sec:prelims and Pi.LABEL:sec:anatomy). Section 3 begins the new analysis and contains lower bounds for finding stationary points of convex functions. In Section 4 we construct our hard non-convex instance, while in Section 5 we use this function to establish our main result: a lower bound on the complexity of finding stationary points using deterministic first-order methods. In Sections 6 we discuss some difficulties in sharpening or extending our lower bounds. Section 7 concludes by situating our work in the current literature and reflecting on its implications for future research.
Notation
A framework for lower bounds
For ease of reference, this section provides a condensed version of Sections Pi.LABEL:sec:prelims and Pi.LABEL:sec:anatomy of the first part of this series that lays out the notation, concepts and strategy we use to prove lower bounds. Here, we are deliberately brief; see for motivation, intuition and background for our definitions, as well as exposition of randomized and higher-order methods.
where is the th derivative of . We occasionally refer to a function with Lipschitz th order derivatives as th-order smooth.
Let , and . Then the set
We also require the following important invariance notion [17, Ch. 7.2].
Every function class we consider is orthogonally invariant.
2 Algorithm classes
3 Complexity measures
We define the complexity of algorithm class on function class as
Table 1 provides upper and lower bounds on the quantity (4) for different choices of and . For example, gradient descent guarantees \mathcal{T}_{\epsilon}\big{(}\mathcal{A}^{(1)}_{\textnormal{{det}}}\cap\mathcal{A}^{(1)}_{\textnormal{{zr}}},\mathcal{F}_{1}(\Delta,L_{1})\big{)}\leq 2\Delta L_{1}\epsilon^{-2}.
4 How to show a lower bound
The last step in our preliminaries is to give an overview of our proof strategy; this is an abbreviated version of Section Pi.LABEL:sec:anatomy. There, we abstract classical techniques for lower bounds in convex optimization , presenting a generic method for proving lower bounds on deterministic methods (of any order) applied to functions in any orthogonally invariant class.
Our starting point is what we call a zero-chain, which distills the “chain-like” structure of Nesterov’s construction (2).
In Definition Pi.LABEL:def:zero-chain , we extend zero-chains to higher orders; in our terminology Nesterov’s function (2) is a first-order zero-chain, but not a second-order zero-chain. A first-order zero-chain limits the rate that zero-respecting algorithms acquire information from derivatives, forcing them to “discover” coordinates one by one, as the following observation makes clear.
The important insight, essentially due to Nemirovski and Yudin is that by using a resisting oracle that can adversarially rotate the function , any lower bound for zero-respecting algorithms implies an identical bound for all deterministic algorithms:
Let be an orthogonally invariant function class. Then
See Proposition Pi.LABEL:prop:prelims-det-zr for a more general version of this result.
This strategy highlights the importance of “dimension freedom,” because we take the dimension of to be at least , which must thus grow inversely with .
Lower bounds for finding stationary points of convex functions
While for convex optimization guarantees of small gradients are atypical topics of study, we nonetheless begin by considering the complexity of finding stationary points of smooth convex functions. This serves two purposes. First, it is a baseline for finding stationary points in the non-convex setting; based on algorithmic upper bounds due to Nesterov , we see that convexity makes this task fundamentally easier. Second, our lower bound construction for convex problems underpins our construction and analysis for general smooth (non-convex) functions in the sequel, allowing us to demonstrate our techniques in a simpler setting. Of course, in convex optimization, it is typically more useful to find points with small optimality gap, . Convexity allows efficient algorithms for guaranteeing such optimality, and typically one ignores questions of the magnitude of the gradient in favor of small optimality or duality gaps . Nonetheless, in some situations—such as certifying (near) dual feasibility or small constraint residuals in primal-dual or operator splitting algorithms [e.g. 7]—achieving small gradients is important.
We proceed as follows. In Section 3.1 we define the class of convex functions under consideration and a quadratic subclass. In Section 3.2, we construct a hard quadratic instance, and verify its key properties. Finally, in Section 3.3, we state, discuss and prove our lower bounds.
The collections of functions we consider are the following.
is the set of convex quadratic functions satisfying the above conditions.
Our results, following Nemirovski and Yudin and Nesterov , demonstrate that for deterministic first-order methods, the class is “hard enough,” in that it provides nearly sharp lower bounds for first-order methods, which immediately apply to and . We also have or any .
In addition to functions restricted by initial optimality gap, we consider the following initial distance-based definition.
is the set of convex quadratic functions satisfying the above conditions.
2 The worst function in the (convex) world
For , this is Nesterov’s “worst function in the world” [18, § 2.1.2]. The parameter allows us to control and thus provides a degree of freedom in satisfying the constraint for our lower bounds. By inspection,
is the unnormalized graph Laplacian of the simple path on vertices (see ) plus the term in the position , and .
Let us now verify that meets the three requirements of our lower bound strategy.
Zero-chain is a first-order zero-chain.
has -Lipschitz continuous gradient.
The unique minimizer of is , and .
Substituting and into Eq. (7), we have that implies
3 Scaling argument and final bound
With our hard instance in place, we provide our lower bounds for finding stationary points of convex functions. We note that the lower bound for the class also follows from the standard lower bounds on finding -suboptimal points, since for every an -stationary point is also -suboptimal.
Let , and be positive. Then
Let us discuss Theorem 1 briefly. Nesterov shows that for any , accelerated gradient descent applied to a regularized version of yields a point satisfying after at most iterations. For , a similar technique to Nesterov’s, which we provide for completeness in Appendix A.1, yields an upper complexity bound of . Thus, to within logarithmic factors both bounds of Theorem 1 are sharp. It is illustrative to compare Theorem 1 to our results for non-convex but smooth functions, and we do so in detail in Sec. 7.1. The comparison shows that finding stationary points of smooth convex functions with first-order methods is fundamentally easier than finding stationary points of non-convex functions, even with higher-order smoothness and using higher-order methods.
4 Proof of Theorem 1
To define the difficult zero-chain , we scale using two scalar parameters , which we determine later, defining
We use the parameter to control the first-order smoothness of , as , while the parameter controls the lower bound on for . We first show how to choose , depending on , , , and . By Lemma 1.iii, for every with we have
guarantees for any such that and hence for all .
All that remains is to choose , and to guarantee that belongs to the appropriate quadratic class. By Lemma 1.ii, has Lipschitz gradient, so we take
and guarantee that has -Lipschitz gradient. To guarantee that , Lemma 1.ii yields
where we have substituted our choice of and in the final equality. Defining
so to guarantee , it suffices to choose
This gives the first part (8a) of the theorem. For inequality (8b), we must have . Let denote the minimizer of , so that
where again we have substituted our choices of and in the final equality. Consequently, to guarantee it suffices to take
Constructing the non-convex hard instance
We illustrate the construction in Figure 1; it is the sum of the convex hard instance (5) (with ) and a separable non-convex function. In the following lemma, which we prove in Appendix B.1, we list the important properties of .
The function satisfies the following.
We have
For all , , and for all , .
For every , for every
Before formally stating the properties of , we provide a high-level explanation of the choice of . First, a necessary and sufficient condition for to be a first-order zero-chain is that . Second, examining the proof Lemma 1.iii we see that the gradient of the quadratic chain is smallest for vectors with entries that slowly decrease from 1 to 0. We design to “punish” such slowly varying vectors, by demanding that be large for any far from both 0 and 1 (Lemma 2.iv); this is the key to improving Lemma 1.iii and the most important property of . Third, for every finite all the derivatives of are Lipschitz, and as increases converges to a quartic polynomial; in the limit we have . This allows us to establish that Lipschitz continuity of derivatives beyond the third does not alter the dependence of our bounds. However, we cannot simply use , as its first three derivatives are unbounded. Lastly, we place the minimum of at , so that the all-ones vector is the global minimizer of , and ; this is simply convenient for our analysis.
With our considerations explained, we verify the three components of our general strategy: is a first-order zero-chain, belongs to the relevant function classes, and has large gradient whenever . We begin with the zero-chain property, which follows trivially from Lemma 2.i.
Crucially, is only a first-order zero-chain (see Definition Pi.LABEL:def:zero-chain); were it a second-order zero-chain, the resulting lower bounds would apply to second-order algorithms as well, where Newton’s method achieves the rate , which is strictly better than all of our lower bounds. We next show that any point for which has large gradient. This is the core technical result of our analysis.
We defer the full proof of this lemma to Appendix B.2 and sketch its main idea here. We may view any vector meeting the conditions of the lemma as a sequence going from to . Every such sequence must have a “transition region”, which we define roughly as the subsequence starting after the last such that and ending at the first (subsequent) such that (see Figure 2). Letting denote the length of this subsequence and ignoring constant factors, we establish that
The bound comes from the quadratic chain in , which has large gradient for any sequence with sharp transitions; this is essentially Lemma 1.iii with and . The bound is due to the non-convex terms in , which by Lemma 2.iv contribute a term of magnitude to every entry of in the transition region. These two bounds intersect at , so the gradient has norm at least for every value of .
Finally, we list the boundedness properties of our construction.
The function satisfies the following.
The first part of the lemma follows from Lemma 2, which shows that , while . The second part of the lemma follows directly from Lemma 2.v and that the quadratic chain has 4-Lipschitz gradient and 0-Lipschitz higher order derivatives. ∎
Lower bounds for first-order methods
We now give our main result: lower bounds for the complexity of finding -stationary using the class of first-order deterministic and/or zero-respecting algorithms, applied to functions in the class
We begin by sketching the argument for . In this case, we may take , and . This choice guarantees that has -Lipschitz gradient and -Lipschitz Hessian when , which we later verify using the assumption . We then use Observation 2, Lemma 3 and Observation 1 to show that \mathsf{T}_{\epsilon}\big{(}\mathsf{A},f\big{)}\geq T+1 for every whenever , and conclude that may scale as (since ). By Lemma 4.i we have
so we can take to guarantee , where we assume without loss of generality that (otherwise Theorem 1 dominates our bound). Substituting the expressions for and into the expression for gives the result for .
For we require a more careful argument, as we must simultaneously handle all orders of smoothness. To do so, we let and , and show how to take and independently of (depending only on ). This allows us to obtain identical -dependence for all .
To better understand the theorem, we give a few additional remarks.
In the paper , we propose the method “convex until proven guilty,” which augments Nesterov’s accelerated gradient method with implicit negative curvature descent. For the function classes and , it achieves rates of convergence and , respectively. These results nearly match our lower bounds in Theorem 2; in the case of , the gap (in terms of ) is of order , while for , the gap is of order . See further discussion in Sec. 7.1.
Choice of function class
The focus on the more restricted function classes —rather than the classes we study in Part I —makes our lower bounds stronger, and it is necessary for non-trivial results, since for any and , the class contains functions impossible for first-order methods. Indeed, the class of -bounded -smooth convex quadratics is a subset of for any and . Therefore, by Theorem 1,
We thus limit our scope to functions with smooth lower order derivatives.
Conditions on the accuracy ϵbold-italic-ϵ\boldsymbol{\epsilon}
In Theorem 2 we require that for all . For each , we may rewrite this as . In other words, these conditions ensure that th order regularization-based methods have stronger convergence guarantees than gradient descent .
The case 𝒑=𝟏𝒑1\boldsymbol{p=1}
We state our bounds in Theorem 2 for . It is possible to use the construction (9) to prove a lower bound of on the time necessary for a deterministic first-order algorithm to find an -stationary point for the class . As Theorem Pi.LABEL:thm:fullder-final shows this lower bound holds for all randomized high-order algorithms, we do not pursue this.
The case 𝒑=𝟑𝒑3\boldsymbol{p=3}
We can slightly strengthen our lower bound in the case , making it independent of for sufficiently small . To achieve this we set in the definition of , take , and argue that that the resulting construction has -Lipschitz continuous Hessian, and tends to zero as . For sufficiently small , we can then replace the minimum over in the first claim of Theorem 2 with .
1 Lower bounds based on distance to optimality
For convex optimization problems, typical convergence guarantees depend on the distance of the initial point to the globally optimal set ; the dependence on this distance may be polynomial for general convex optimization problems , while for smooth and strongly convex problems, the convergence guarantees depend only logarithmically on it. In the non-convex case, we can provide lower bounds that depend on the distance rather than the gap . To that end, we consider the class
functions with -Lipschitz th derivatives (for each ) and all global minima satisfying . We obtain the bound, analogously to our results in Section Pi.LABEL:sec:fullder-distance, by “hiding” a sharp global minimum near the origin.
To state the theorem, we require an additional piece of notation. Let be the lower bound Theorem 2 provides on \mathcal{T}_{\epsilon}\big{(}\mathcal{A}_{\textnormal{{det}}}\cup\mathcal{A}^{(1)}_{\textnormal{{zr}}},\mathcal{F}_{1:p}(\Delta,L_{1},...,L_{p})\big{)}, so
where we take if is larger than the settings Theorem 2 requires. Then by a reduction from our lower bounds on the complexity of , we obtain the following result.
We prove Theorem 3 in Appendix B.3. Theorem 3 shows that the lower bounds of Theorem 2 apply almost identically (to constant factors), except that we replace the function gap in the lower bound with the quantity . As the dependence of the lower bound on does not change, distance-based assumptions seem unlikely to help in the design of efficient optimization algorithms for non-convex functions.
2 Proof of Theorem 2
We must choose these parameters to guarantee the membership
For the bounded values constraint , by Lemma 4.i it suffices to take
Thus, so long as we choose the constants to satisfy inequality (12), the preceding choice of guarantees .
With this membership guaranteed, we consider the choices for , and such that after iterations of a zero respecting first-order method, we have . Indeed, Observations 2 and 1 imply that if are the sequence of iterates produced by applying any zero-respecting (first-order) method to , then for all . Lemma 3 implies that for any such iterate. Therefore, if we choose , and such that
then for all . We thus obtain the guarantee
The same bound for the class then follows from Proposition 1. Our strategy is now the obvious one: we select , and to satisfy the function membership constraints (12) and the large gradient guarantee (14). Substituting our choices into the boud (15) will then yield the lower bound in the theorem. We begin with the general case and later provide a tighter construction for .
To simplify the derivation, we define, for any
We choose and to guarantee that is appropriately smooth; in the sequel, we will choose and so that the gradient bound condition (14) holds. In this sense, we may choose and without consideration of . Taking guarantees the inequality (18) holds for . Substituting this choice into the identical inequality for shows that we must have for each such . Thus, the choice
satisfies inequality (18), and consequently, the smoothness condition (12) as well. We may therefore write and as
It remains to choose , depending on , to guarantee our gradient lower bound condition (14) holds, i.e. . We thus set
We can now substitute back into our definitions and in Eq. (17) and verify that and For , we have
We now consider two cases; and . In the first case (which holds for sufficiently small ), we substitute our choices of and into the time lower bound (15),
which is the desired bound, where in step we made use of . When , we show that the above bound is in fact smaller than the convex lower bound in Theorem 1. Indeed, substituting in our choices of and , we see that implies
Taking a square root and substituting to our lower bound gives
completing the proof in the general case.
Functions with Lipschitz Hessian
For , we keep the definitions (16) but replace the particular rescaling choices (17) with
Using , the above parameter setting satisfies inequality (12); has -Lipschitz th-order derivatives for . To satisfy the gradient lower bound (14), i.e. , we set
We can substitute into the definition to verify that :
If does not hold, we have
Taking a square root and substituting to our lower bound gives
due to Theorem 1, establishing the case .
The challenge of strengthening Theorem 2
The lower bounds in Theorem 2 leave two avenues for improvement. The first is tightening our and lower bounds to match the known upper bounds of and , for and , respectively. The second improvement is to extend our lower bounds to randomized algorithms, as we did for the case of full derivative information in Section Pi.LABEL:sec:fullder-random. We discuss each of these in turn.
The core of our first-order lower bounds is Lemma 3, which establishes a lower bound of the form for vectors such that (i.e. any point that a first-order zero-respecting method can produce after iterations), where is our unscaled hard instance (see definition (9)). Here we consider a slightly more general form,
The next lemma, whose proof we provide in Appendix B.4, shows such gradient norm upper bound for constructions of the form (19).
Summarizing, tightening our lower bounds seems to require a construction that is not of the form (19). This does not eliminate more general (non-convex) interactions, e.g. of the form rather than . The proof technique of Lemma 3 should provide useful “sanity checks” when considering alternative constructions.
2 A bound for randomized algorithms
In Section Pi.LABEL:sec:fullder-random we extend our lower bound for \mathsf{T}_{\epsilon}\big{(}\mathcal{A}_{\textnormal{{det}}}\cup\mathcal{A}_{\textnormal{{zr}}},\mathcal{F}_{p}(\Delta,L_{p})\big{)} to the broader class of randomized algorithms with access to all derivatives at query point . We do this by making our hard function insensitive: the individual “linking” terms are identically zero for near 0. A natural question is whether the same methodology (originally proposed in ) can extend Theorem 2 to the class of randomized first-order algorithms, . Direct application of that technique cannot work in our case, for a simple reason: it applies to randomized algorithms of any order. In other words, if we modify our hard instance construction (9) to be a robust zero-chain (Definition Pi.LABEL:def:robust-zero-chain), any lower bounds it implies hold for all algorithms in , where rates are achievable, so we could not provide sharper lower bounds than Theorem Pi.LABEL:thm:fullder-final.
Nevertheless, the ideas introduced in Section Pi.LABEL:sec:fullder-random might still be of use. Specifically, consider a modification of the construction (9) where is identically zero for sufficiently small , say , while still satisfying Lemma 2, thus making the non-convex component of insensitive. As explained above, also making the convex quadratic component of insensitive (as Woodworth and Srebro do) is unworkable in our setting, as it results in a robust zero-chain equally hard for all high-order algorithms. Instead, we may keep the quadratic component unchanged—and hence sensitive—and try to carry out the proof of Lemma Pi.LABEL:lem:fullder-rand-slow. Doing so, we see that the inductive argument allows us to ignore the insensitive non-convex component of , leaving us to contend only with the (randomly rotated) quadratic chain. Thus, the difficulty here appears closely related to proving a lower bound for randomized first-order methods applied for optimization of convex quadratics. Such lower bounds remain elusive, and we believe finding them is an important open problem.
Concluding remarks
Here we situate our work in the literature, discuss its implications, and provide a few possible extensions.
In conjunction with known upper bounds, our lower bounds characterize the optimal rates for finding stationary points. Our lower bounds are sharp to within constant factors for algorithms with full derivative information , and (perhaps) slightly loose for first-order algorithms. These characterizations yield a few insights.
For the class of -smooth functions, first-order methods—specifically gradient descent—attain the optimal rate ; no higher-order randomized method can attain improved performance over the entire function class. The intuition here is that contains functions whose Hessian and higher order derivatives may vary arbitrarily sharply, and providing no useful information for optimization.
When higher-order derivatives are also Lipschitz continuous the picture changes fundamentally: there is a strict separation between (deterministic) second-order and first-order methods. In particular, cubic regularization of Newton’s method achieves dependence for functions with Lipschitz Hessian, while no deterministic first-order method can have better time complexity than , regardless of how many derivatives are Lipschitz. Note that when the Hessian is Lipschitz, our definition of first-order algorithms allows for algorithms that rely on Hessian-vector products, as they can be estimated to arbitrary accuracy in two gradient evaluations.
The effect of high-order smoothness on first-order methods
For , the class of functions with Lipschitz gradient and Hessian, our lower bound scales as , while for the class of functions with Lipschitz third order derivative our “convex until proven guilty” method achieves the rate . As , this proves a separation between the optimal rate for first-order methods with second- and third-order smoothness.
In contrast, orders of smoothness beyond the third offer limited room for improvement in dependence; the lower bound holds for all function classes with , while the method does not enjoy improved guarantees with Lipschitz fourth-order derivatives. The “robustness” of the lower bound to higher-order smoothness stems from the fact that our hard instance becomes a quartic polynomial in the limit , and we choose inversely proportional to . As we discuss in [9, Lemma 4], our guarantee cannot improve using fourth-order smoothness because of symmetries in the fourth-order Taylor expansion. Quartic polynomials thus appear to play a central role in the complexity of first-order methods for smooth optimization.
Convex vs. non-convex functions
Our results show that convexity makes finding stationary points fundamentally—and significantly—easier. For first-order methods and functions with bounded initial sub-optimality, the rate is achievable for first-order smooth convex functions, while the lower bound holds for non-convex functions with arbitrarily high-order smoothness. For methods using higher-order derivatives, our lower bounds for finding stationary points of non-convex functions are as the order of smoothness grows. However, similar to Appendix A.1, the results can show that for convex functions with Lipschitz Hessian, a second-order method achieves the strictly better rate .
Another striking difference between convex and non-convex functions is the effect of replacing the bound on the initial function value (i.e. ) with a bound on the initial distance to the global minimizer (i.e. ). For non-convex function classes, we show lower bounds with the same dependence regardless of which type of bound is used. In contrast, for convex function the optimal rates scale as and , again a gap in dependence. The rates are not directly comparable; one can construct families of functions where grows with the dimension while remains constant.
Returning to -value-bounded function classes, we see one more large difference between the convex and non-convex case; convex rates scale as while all the non-convex rates scale linearly with . This arises from fundamental differences in the convergence “mechanism” for convex and non-convex optimization. The analysis of non-convex optimization schemes typically revolves around a progress argument, where one shows that, as long as , the guarantee holds for some quantity (e.g. for gradient descent ). The number of iterations to find an -stationary point is therefore at most , which scales linearly in . By our lower bounds, such progress arguments are, in a sense, optimal. Conversely, in convex optimization we may control either the gap or the distance , and this interplay (see Appendix A.1) allows stronger arguments than those based purely on function progress.
2 Further research
There exists a gap in polynomial dependence between our lower bounds (Theorem 2) and the best known upper bounds for first-order methods with higher-order smoothness. We do not believe the upper bounds of are improvable by different analysis or by any algorithmic change that maintains the general structure of alternating between accelerated gradient descent and negative curvature exploitation. In conjunction with our arguments in Section 6.1 about the structure of our lower bounds, resolution of the optimal rate will likely provide either a method with a substantially different approach to accelerating gradient descent in the smooth non-convex setting or a new lower bound construction.
Finite sum and stochastic problems
Smooth, non-convex, finite-sum and stochastic optimization problems are important, arising (for example) in the training of neural networks. This motivates the design and analysis of efficient methods for finding stationary points in such problems, and researchers have successfully developed variance reduction and acceleration techniques for these settings . However, no corresponding lower bounds are available. Woodworth and Srebro , show how to establish lower bounds for convex finite sum problems. Combined with the developments in our paper, we believe their techniques should extend to finding stationary points of non-convex problems. An important conclusion of is that randomized selection of the component function is crucial to efficient convergence: in contrast to our results, they show a separation between deterministic and randomized finite sum complexity.
Second-order stationary points
Approximate stationary points are not always close to local minima, and so it is interesting to consider stronger convergence guarantees. Second-order stationarity (also known as the second-order necessary condition for local optimality) is the most popular example; for a function , a point is -second-order stationary if and . Efficient first-order methods for finding second-order stationary points exist . Moreover, it is possible to generically transform methods for finding -stationary points into methods that find -second-order stationary points, for some , without changing the dependence of the complexity [9, Appendix C], but such modifications introduce dependence logarithmic in the problem dimension .
Clearly, lower bounds for finding -stationary points also apply to finding -second-order stationary points. However, attaining second-order stationarity with first-order methods is fundamentally more difficult than attaining only stationarity. There are no dimension-free guarantees: the results of Simchowitz et al. imply dimension dependence for all randomized first-order algorithms that escape saddle points. Moreover, for deterministic first-order algorithms it is easy to construct a resisting oracle that forces dimension dependence (consider with and adversarially chosen rotation ), implying strong separation between deterministic and randomized first-order methods for finding second-order stationary points. It will be interesting to investigate such issues further.
Acknowledgments
OH was supported by the PACCAR INC fellowship. YC and JCD were partially supported by the SAIL-Toyota Center for AI Research, NSF-CAREER award 1553086, and a Sloan Foundation Fellowship in Mathematics. YC was partially supported by the Stanford Graduate Fellowship and the Numerical Technologies Fellowship.
References
Appendix A Additional results for convex functions
Here we give a first-order method that finds -stationary points of a function in iterations. The method consists of Nesterov’s accelerated gradient descent (AGD) applied on the sum of and a standard quadratic regularizer.
Our starting point is AGD for strongly convex functions; a function is -strongly convex if
Moreover, for any such we have [6, Eq. (9.14)], and consequently
Using our complexity notation, we may rewrite this as
Now suppose that is convex with -Lipschitz gradient but not necessarily strongly-convex. We can add strong convexity to by means of a proximal term; for any , the function
is -strongly-convex with -Lipschitz gradient. With this in mind, we define a proximal version of AGD as follows,
With a careful choice of , achieves the desired upper bound.
Let and be positive, and let . Then, algorithm satisfies
For any , recall that and let
be the sequence of iterates produces on . Then by guarantee (21), we have
For any point such that , we have
Clearly, and by guarantee (20) we also have . Consequently,
In inequality we substituted bounds (22) and (24), and in we used . We conclude that \mathsf{T}_{\epsilon}\big{(}\mathsf{PAGD}_{\sigma,L_{1}},f\big{)}\leq T, and substituting (24) and the definition of into (23) we have
Without loss of generality, we may assume , as otherwise \mathsf{T}_{\epsilon}\big{(}\mathsf{PAGD}_{\sigma,L_{1}},f\big{)}=1. We thus simplify the expression slightly to obtain the proposition. ∎
A.2 The impossibility of approximate optimality without a bounded domain
We make the following inductive claim: if , then
for all . Indeed, each term in the sum (25) defining is non-negative, so for the base case of the induction , we have , or . For , assuming that satisfies the bound (26), we have that , which implies
which is the desired claim (26) for .
The bound (26) implies for all whenever . Therefore, we choose to satisfy , that is
for which since we assume . Thus, we guarantee that when we must have , giving the result. ∎
Appendix B Technical results
Parts i and ii are evident from inspection, as
To see the part iii, note that is non-increasing for every and non-decreasing for every and therefore is its global minimum. That is immediate from its definition, and, for every , . To see part iv, note that for every , and a calculation shows for (see Figure 1).
To see the fifth part of the claim, note that
where the functions and are and . We thus bound the derivatives of and . We begin with , which we can write as the composition where and . Let denote the collection of all partitions of where each element of the partition has at most indices. That is, if , then for some , the are disjoint, , and . The cardinality is the number of matchings in the complete graph on vertices, or the th telephone number, which has bound [12, Lemma 2]
We may then apply Faà di Bruno’s formula for the chain rule to obtain
where denotes the number of sets in with precisely elements. Of course, we have , and thus
The proof of the upper bound on is similar ( with and as defined above), so for every and , the -th derivative of has the bound
where is a numerical constant. ∎
B.2 Proof of Lemma 3
so that for every . Note that when for every . This is a somewhat special case due to the coefficient of the first “link” in the quadratic chain term in (9). To handle it cleanly we define
Continuing with construction of the transition region, we make the following definition.
and let , so . Roughly, our transition region consists of the indices , but for technical reasons we attach to it the following decreasing ‘tail’.
With these definitions, is well-defined and , since . We denote the transition region and associated length by
We illustrate our definition of the transition region in Figure 3.
Let us describe the transition region. In the “head” of the region, we have for every ; a total of indices. The “tail” of the transition region is strictly decreasing, . Moreover, for any such that , the decrease is rapid; . This descriptions leads us to the following technical properties.
and .
We defer the proof of the lemma to the end of this section, continuing the proof assuming it.
We now lower bound . For notational convenience, define , and recalling that , we see that the norm of the gradient of is
where we made use of the notation if and if . We obtain a lower bound for the final sum of squares (28) by fixing , , and , then minimizing the quadratic form explicitly over the variables . We obtain
where the matrix and vector have definitions
We now bring to bear the properties of the transition region Lemma 7 supplies. By Lemma 7.i,
and by Lemma 7.ii, using ,
Substituting and the bounds (30) and (31) into the gradient lower bound (29), we have that
A quick computation reveals that , which gives the result. ∎
Proof of Lemma 7. We have by definition that and . To see that
holds, consider the two cases that or . In the first case that , by definition so . The second case that is a bit more subtle. By definition of the sequence , we have
Combining this bound on and the inequality due to the construction of , we obtain
B.3 Proof of Theorem 3
The proof builds off of those of Theorems 2 and Pi.LABEL:thm:fullder-final-dist. We begin by recalling the following bump function construction
Adding a scaled version of to our hard instance construction allows us to “plant” a global minimum that is both close to the origin and essentially invisible to zero-respecting method. For convenience, we restate Lemma Pi.LABEL:lem:fullder-hhard-props,
The function satisfies the following.
We note that by Lemma 8.ii, is identically 0 at a neighborhood of any with , which immediate implies that and are zero-chains. Therefore for any producing iterates when operating on , we have for any . Thus, by our choices of and , for every , and so
To establish that , it remains to show that every global minimizer of has norm at most . Let denote a global minimizer of , and temporarily assume that
Therefore, and , as otherwise we have the contradiction . By the definition (33), implies that , and therefore . To verify the assumed inequality (36), we use Lemma 8.i to obtain
B.4 Proof of Lemma 5
We construct as follows. We let , and for let (with ),
where for we used and to write . Since is 1-Lipschitz, we have
Moreover, one can readily verify that for every and that for every . Therefore, using using and we have that , which gives the overall bound
Taking , we have
where we have used since . Thus, holds for . For , since , we have and therefore holds as required (since for every ). In the edge case we have and therefore yields . ∎