Convergence of gradient descent for deep neural networks

Sourav Chatterjee

A convergence criterion for gradient descent

with ϕ(0)=x0\phi(0)=x_{0}. Gradient descent and its many variants are indispensable tools in all branches of science and engineering, and particularly in modern machine learning and data science. The convergence properties of gradient descent are well-understood when the objective function ff is convex , and it is known that finding local minima of nonconvex functions by gradient descent is an NP-complete problem . In spite of this, gradient descent is widely used in practice to find local and global minima in highly nonconvex problems, especially in high dimensions. For example, it has been observed that gradient descent can often find global minima of training loss in deep learning , which is one of the reasons behind great success of the ‘deep learning revolution’ .

This article presents a novel criterion for convergence of gradient descent to a global minimum. The criterion is related to (and maybe seen as a strengthening of) the classical Kurdyka–Łojasiewicz inequality . It is also related to results from nonsmooth analysis, such as those in . Indeed, our proof idea bears close resemblance with the classical ‘Łojasiewicz trapping argument’ . Nevertheless, the convergence criterion stated below has not appeared in this exact form previously in the literature.

where ∣∇f(x)∣|\nabla f(x)| is the Euclidean norm of ∇f(x)\nabla f(x). If f(x)=0f(x)=0 for all x∈B(x0,r)x\in B(x_{0},r), then we let α(x0,r):=∞\alpha(x_{0},r):=\infty. Our main assumption is that for some r>0r>0,

Under the above assumption, we have two results. The first result, stated below, shows that the gradient flow started at x0x_{0} converges exponentially fast to a global minimum of ff in B(x0,r)B(x_{0},r) where ff is zero. The existence of a global minimum in B(x0,r)B(x_{0},r) is a part of the conclusion, and not an assumption.

Let ff, x0x_{0}, and α\alpha be as above. Assume that (1.1) holds for some r>0r>0, and let α:=α(x0,r)\alpha:=\alpha(x_{0},r). Then there is a unique solution of the gradient flow equation

on [0,∞)[0,\infty) with ϕ(0)=x0\phi(0)=x_{0}, and this flow stays in B(x0,r)B(x_{0},r) for all time, and converges to some x∗∈B(x0,r)x^{*}\in B(x_{0},r) where f(x∗)=0f(x^{*})=0. Moreover, for each t≥0t\geq 0, we have

Our second result is the analogue of Theorem 1.1 for gradient descent. It says that under the condition (1.1), gradient descent started at x0x_{0}, with a small enough step size, converges to a global minimum of ff in B(x0,r)B(x_{0},r). Again, the existence of a global minimum in B(x0,r)B(x_{0},r) is a part of the conclusion.

Let ff, x0x_{0}, and α\alpha be as above. Assume that (1.1) holds for some r>0r>0, and let α:=α(x0,r)\alpha:=\alpha(x_{0},r). Choose ϵ∈(0,1)\epsilon\in(0,1) such that

which is possible since (1.1) holds. Let L1L_{1} be a uniform upper bound on the magnitudes of the first-order derivatives of ff in B(x0,r)B(x_{0},r), and let L2L_{2} be a uniform upper bound on the magnitudes of the second-order derivatives of ff in B(x0,2r)B(x_{0},2r). Choose any η>0\eta>0 such that

for each k≥0k\geq 0. Then xk∈B(x0,r)x_{k}\in B(x_{0},r) for all kk, and as k→∞k\to\infty, xkx_{k} converges to a point x∗∈B(x0,r)x^{*}\in B(x_{0},r) where f(x∗)=0f(x^{*})=0. Moreover, with δ:=min⁡{1,(1−ϵ)αη}\delta:=\min\{1,(1-\epsilon)\alpha\eta\}, we have that for each kk,

Note that the above results have nothing to say about variants of gradient descent, such as stochastic gradient descent. Adding a stochastic component to the gradient descent algorithm has various benefits, such as helping it escape saddle points . Since it is known that stochastic gradient methods often asymptotically follow the path of a differential equation , it would be interesting to see if analogues of Theorems 1.1 and 1.2 can be proved for stochastic gradient descent. We also do not have anything to say about algorithms that aim to find critical points instead of global minima in nonconvex problems, such as the ones surveyed in . For a recent survey of the many variants of stochastic gradient descent and their applications in machine learning, see . For a comprehensive account of all variants of gradient descent, see . For some essential limitations of nonconvex optimization, see .

It is possible that Theorems 1.1 and 1.2 may be generalizable to what are variously called lower C2C^{2} functions, or proximal-regular functions, or weakly convex functions. However, the generalizations are not obvious (especially for Theorem 1.2), and are therefore left for future investigation.

Application to deep neural networks

A feedforward neural network consists of the following components:

A positive integer LL, which denotes the number of layers. It is sometimes called the depth of the network. The number L−1L-1 denotes the ‘number of hidden layers’. To avoid trivialities, we will assume that L≥2L\geq 2 — that is, there is at least one hidden layer.

A sequence of positive integers d1,…,dLd_{1},\ldots,d_{L}, denoting the dimensions of layers 1,…,L1,\ldots,L. The maximum of d1,…,dLd_{1},\ldots,d_{L} is called the width of the network. The dimension of the ‘output layer’, dLd_{L}, is often taken to be 11. We will henceforth take dL=1d_{L}=1.

where the activation functions σ1,…,σL\sigma_{1},\ldots,\sigma_{L} act componentwise on vectors of dimensions d1,…,dLd_{1},\ldots,d_{L}.

for k≥0k\geq 0, where η\eta is the step size, and ∇S\nabla S denotes the gradient of SS (assuming that the activation functions are differentiable).

There is an enormous body of literature on convergence properties of gradient descent for neural networks. The following are some of the most important contributions. Further references can be found in the review sections of these papers and also in the recent comprehensive survey .

Convergence for convex neural networks was studied in , and for linear networks in . Convergence in the absence of convexity and linearity has remained an open problem, except in one particular scenario — when the dimensions of the hidden layers 1,…,L−11,\ldots,L-1 are extremely large, where ‘extremely large’ may mean either tending to infinity, or larger than some large enough power of the sample size. This is now called the ‘infinite width’ or ‘overparametrized’ regime. Following some early results in , this approach was fully developed independently in the concurrent papers . The key idea here is that in the overparametrized regime, gradient descent for the neural net can be approximated by gradient descent in a linear setting. Since then, this idea has been widely applied in a variety of settings, for example, in . For some recent advances beyond the overparametrized regime, see and references therein.

We have two results about convergence of gradient descent for feedforward neural networks of bounded width and depth. The first theorem shows that under fairly general conditions, it is possible to find an exact fit to the data (i.e., a point w∗w^{*} where S(w∗)=0S(w^{*})=0) via gradient descent with suitable initialization and step size. The main requirement is that the input data x1,…,xnx_{1},\ldots,x_{n} have to be linearly independent, which necessarily means that the dimension dd of the input space has to be ≥n\geq n. This condition is inevitable, because if this is not true, then there may not exist any point w∗w^{*} where S(w∗)=0S(w^{*})=0 even for linear activation.

Recall that the width of the network is the number m:=max⁡{d1,…,dL}m:=\max\{d_{1},\ldots,d_{L}\}. It is important to note that this excludes the input dimension d0=dd_{0}=d. Theorem 2.1 implicitly needs d≥nd\geq n, but there is no requirement on the width mm. Most previous works need the width to grow with nn. For example, Soltanolkotabi et al. require m≥2nm\geq 2n (their result is only for networks with one hidden layer), while both Du et al. and Allen-Zhu et al. — who deal with networks of arbitrary depth — require mm to be at least as large as some polynomial in nn. One existing result result that might imply Theorem 2.1 is [19, Theorem 2.4], but it is completely clear if it does.

The class of activation functions allowed by Theorem 2.1 includes many functions used in common practice, such as linear activation (σ(x)=x\sigma(x)=x), bipolar sigmoid activation (σ(x)=(1−e−x)/(1+e−x)\sigma(x)=(1-e^{-x})/(1+e^{-x})), and tanh activation (σ(x)=tanh⁡x\sigma(x)=\tanh x). Moreover, the condition σ(0)=0\sigma(0)=0 is not a serious restriction, since the presence of the bias vectors implies that the class of models remains the same if we subtract off some constants from our activation functions to make them zero at the origin. In that sense, Theorem 2.1 also allows the sigmoid activation (σ(x)=1/(1+e−x)\sigma(x)=1/(1+e^{-x})), smoothed ReLU activation (σ(x)=log⁡(1+ex)\sigma(x)=\log(1+e^{x})), and complementary log-log activation (σ(x)=1−e−ex\sigma(x)=1-e^{-e^{x}}). It does not, however, allow activation functions that are not twice continuously differentiable, such as ReLU activation (σ(x)=max⁡{x,0}\sigma(x)=\max\{x,0\}), step activation (σ(x)=1\sigma(x)=1 if x>0x>0 and if x≤0x\leq 0), and piecewise linear activation.

Theorem 2.1 gives a criterion for convergence of gradient descent to a solution that perfectly interpolates the data. It does not, however, say anything about why such interpolating solutions sometimes have good generalization errors, or conditions under which deep learning performs well (or poorly). These are some of the other great mysteries of deep neural networks that have attracted much attention. For more about these, see and references therein.

The proof of Theorem 2.1 yields formulas for AA and η\eta, but they are quite complicated and possibly sub-optimal, and are therefore omitted from the discussion. In practice, it may be easiest to just choose AA and η\eta by trial and error after choosing W2,…,WL−1W_{2},\ldots,W_{L-1} with arbitrary positive entries, by increasing AA and decreasing η\eta until convergence is achieved.

Not many activation functions in common use satisfy the condition that the slope is uniformly bounded below by a positive constant. One prominent example that satisfies this condition is the leaky ReLU activation function (σ(x)=x\sigma(x)=x if x>0x>0 and σ(x)=ax\sigma(x)=ax if x≤0x\leq 0, where aa is some positive number less than 11). However, leaky ReLU activation is not twice continuously differentiable. We propose the following smooth modification of the leaky ReLU:

Here a∈(0,1)a\in(0,1), as in the definition of leaky ReLU. Note that as x→±∞x\to\pm\infty, smooth leaky ReLU has the same asymptotic behavior as leaky ReLU. Although σ(0)≠0\sigma(0)\neq 0, this can be easily fixed by subtracting a constant (which does not affect anything since we have bias vectors in our model).

Proof of Theorem 1.1

To avoid trivialities, let us assume that ff is not everywhere zero in B(x0,r)B(x_{0},r), and hence α<∞\alpha<\infty. Throughout this proof, we will denote the closed ball B(x,r0)B(x,r_{0}) by BB and its interior by UU.

Take any S<Tx−tS<T_{x}-t. Define g(s):=ϕx(t+s)g(s):=\phi_{x}(t+s) for s∈[0,S]s\in[0,S]. Then

Thus, h1h_{1} satisfies the integral equation for the gradient flow starting from yy in the interval [0,L][0,L]. Thus, by the uniqueness assumption for this flow in [0,L][0,L], we get that h1=ϕyh_{1}=\phi_{y} in this interval. Consequently, h=gh=g in [0,t+L][0,t+L]. Thus, the gradient flow starting from xx has a unique solution in [0,T][0,T] for every T≤t+ST\leq t+S. Since t+S>Txt+S>T_{x}, this contradicts the definition of TxT_{x}. ∎

Let K′K^{\prime} be the set of all points that are within distance 11 from KK. Note that K′⊇KK^{\prime}\supseteq K and K′K^{\prime} is compact. Since ff is in C2C^{2}, its first and second order derivatives are uniformly bounded on K′K^{\prime}. Thus, there is some LL such that for any x,y∈K′x,y\in K^{\prime},

Let A\mathcal{A} be the subset of B\mathcal{B} consisting of all gg such that g(0)=xg(0)=x and g(t)∈K′g(t)\in K^{\prime} for all t∈[0,T]t\in[0,T]. It is easy to see that A\mathcal{A} is a closed subset of B\mathcal{B}. Define a map Φ:A→B\Phi:\mathcal{A}\to\mathcal{B} as

where the second inequality holds because g(s)∈K′g(s)\in K^{\prime} for all ss, and the third inequality holds because t≤1/Lt\leq 1/L. Since all points at distance ≤1\leq 1 from xx are in K′K^{\prime}, this shows that Φ(A)⊆A\Phi(\mathcal{A})\subseteq\mathcal{A}. Next, note that for any g,h∈Ag,h\in\mathcal{A}, and any t∈[0,T]t\in[0,T],

We claim that g∗g^{*} is the only such map. To prove this, suppose that there exists another map hh with the above properties. If hh maps into K′K^{\prime}, then the uniqueness of the fixed point implies that h=g∗h=g^{*}. So, suppose that hh ventures outside K′K^{\prime}. Let

Since h(0)=x∈K′h(0)=x\in K^{\prime} and hh goes outside K′K^{\prime}, t0t_{0} is well-defined and finite. Moreover, since K′K^{\prime} is closed, h(t0)∈∂K′h(t_{0})\in\partial K^{\prime}. But note that since h(s)∈K′h(s)\in K^{\prime} for all s≤t0s\leq t_{0},

But this implies that the ball of radius 1/31/3 centered at h(t0)h(t_{0}) is completely contained in K′K^{\prime}, and hence, h(t0)∉∂K′h(t_{0})\notin\partial K^{\prime}. Thus, hh cannot venture outside K′K^{\prime}. We conclude that Tx≥1/LT_{x}\geq 1/L. ∎

The above lemma has several useful corollaries.

Suppose that ∣ϕx(t)∣↛∞|\phi_{x}(t)|\not\to\infty. Then there is a compact set KK such that for any ϵ>0\epsilon>0, we can find t∈[Tx−ϵ,Tx)t\in[T_{x}-\epsilon,T_{x}) with ϕx(t)∈K\phi_{x}(t)\in K. By Lemma 3.2, we can find ϵ∈(0,inf⁡y∈KTy)\epsilon\in(0,\inf_{y\in K}T_{y}). Take any t∈[Tx−ϵ,Tx)t\in[T_{x}-\epsilon,T_{x}) such that y:=ϕx(t)∈Ky:=\phi_{x}(t)\in K. Then, by Lemma 3.1, we have Ty=Tx−t<ϵT_{y}=T_{x}-t<\epsilon. But this contradicts the fact that Ty≥inf⁡z∈KTz>ϵT_{y}\geq\inf_{z\in K}T_{z}>\epsilon. ∎

If f(x)=0f(x)=0, then since ff is a nonnegative C2C^{2} function, ∇f(x)\nabla f(x) must also be zero. Thus, it suffices to prove the result under the assumption that ∇f(x)=0\nabla f(x)=0. Taking K={x}K=\{x\} in Lemma 3.2, we get that Tx>0T_{x}>0. Suppose that TxT_{x} is finite. Then let t:=Tx/2t:=T_{x}/2. Note that since ∇f(x)=0\nabla f(x)=0, g(s)≡xg(s)\equiv x is a solution of the gradient flow equation starting from xx in the interval [0,t][0,t]. By uniqueness, this shows that ϕx(s)=x\phi_{x}(s)=x for all s∈[0,t]s\in[0,t]. In particular, ϕx(t)=x\phi_{x}(t)=x. By Lemma 3.1, this implies that Tx=Tx−t=Tx/2T_{x}=T_{x}-t=T_{x}/2, which contradicts the fact that TxT_{x} is nonzero and finite. Thus, Tx=∞T_{x}=\infty. Again, the function g(t)≡xg(t)\equiv x is a solution of the flow equation in [0,∞)[0,\infty). Thus, by uniqueness, ϕx(t)=x\phi_{x}(t)=x for all xx. ∎

Take any tt such that ∇f(ϕx(t))=0\nabla f(\phi_{x}(t))=0 or f(ϕx(t))=0f(\phi_{x}(t))=0. Let y:=ϕx(t)y:=\phi_{x}(t). Then by Lemma 3.1, Ty=Tx−tT_{y}=T_{x}-t. But by Corollary 3.4, Ty=∞T_{y}=\infty. Thus, Tx=∞T_{x}=\infty. Also by Corollary 3.4, ϕy(u)=y=ϕx(t)\phi_{y}(u)=y=\phi_{x}(t) for all u>0u>0, and by Lemma 3.1, ϕy(u)=ϕx(t+u)\phi_{y}(u)=\phi_{x}(t+u) for all u>0u>0. Thus, ϕx(s)=ϕx(t)\phi_{x}(s)=\phi_{x}(t) for all s>ts>t. ∎

In the following, let us fix x0x_{0} and rr as in Theorem 1.1 and write ϕ\phi and TT instead of ϕx0\phi_{x_{0}} and Tx0T_{x_{0}}, for simplicity of notation.

If T<∞T<\infty, then ϕ\phi must visit the boundary of BB.

Suppose that T<∞T<\infty and ϕ\phi remains in UU throughout. By Lemma 3.2, there is some δ>0\delta>0 such that Ty≥δT_{y}\geq\delta for all y∈By\in B. Choose t∈(T−δ,T)t\in(T-\delta,T). Let z:=ϕ(t)z:=\phi(t). Then z∈Uz\in U, and therefore Tz≥δT_{z}\geq\delta. This shows that the gradient flow starting from x0x_{0} exists up to time at least t+δt+\delta. Since t+δ>Tt+\delta>T, this gives a contradiction which proves that ϕ\phi cannot remain in UU throughout. ∎

Let t≥0t\geq 0 be any number such that ϕ(s)∈B\phi(s)\in B for all s≤ts\leq t. Then for all s≤ts\leq t,

Note that by the flow equation for ϕ\phi,

By the definition of α\alpha, the right side is bounded above by −αf(ϕ(s))-\alpha f(\phi(s)) if ϕ(s)∈B\phi(s)\in B. It is now a standard exercise to deduce the claimed inequality. ∎

The flow ϕ\phi cannot visit the boundary of BB.

Suppose that ϕ\phi does visit the boundary of BB. Let t0:=inf⁡{t:ϕ(t)∈∂B}t_{0}:=\inf\{t:\phi(t)\in\partial B\}. Then ϕ(t0)∈∂B\phi(t_{0})\in\partial B and ϕ(s)∈U\phi(s)\in U for all s<t0s<t_{0}. By the flow equation,

Now, since ϕ(s)≠ϕ(t0)\phi(s)\neq\phi(t_{0}) for all s<t0s<t_{0}, Corollary 3.5 shows that f(ϕ(s))≠0f(\phi(s))\neq 0 for all s<t0s<t_{0}. Thus, the map

is differentiable in (0,t0)(0,t_{0}), with continuous derivative

where the second identity follows from (3.1). Thus, for any compact interval [a,b]⊆(0,t0)[a,b]\subseteq(0,t_{0}),

Since gg is continuous on [0,t0][0,t_{0}], we can take a→0a\to 0 and b→t0b\to t_{0}, and apply the monotone convergence theorem on the left, to get

Therefore, by (3.2) and the Cauchy–Schwarz inequality,

On the other hand, since ϕ(s)∈B\phi(s)\in B for all s≤t0s\leq t_{0}, Lemma 3.7 shows that

for all s≤t0s\leq t_{0}. Plugging this bound into the previous display, we get

But the last quantity is strictly less than rr, by assumption (1.1). Since ϕ(t0)∈∂B\phi(t_{0})\in\partial B, this gives a contradiction, which proves the lemma. ∎

Combining Lemma 3.6 and Lemma 3.8, we see that T=∞T=\infty. Moreover by Lemma 3.8, ϕ\phi stays in UU forever. Therefore, by Lemma 3.7, it follows that f(ϕ(s))≤e−αsf(x0)f(\phi(s))\leq e^{-\alpha s}f(x_{0}) for all ss. It remains to establish the convergence of the flow and the rate of convergence. Let

with the understanding that t0=∞t_{0}=\infty if f(ϕ(t))>0f(\phi(t))>0 for all tt. Then note that for any 0≤s<t<t00\leq s<t<t_{0},

where g(u)=f(ϕ(u))g(u)=\sqrt{f(\phi(u))}, as in the proof of Lemma 3.8. As in that proof, note that

Combining the last three displays, and invoking assumption (1.1), we get

Note that the bound does not depend on tt. If t0<∞t_{0}<\infty, then by Corollary 3.5, ϕ(t)=ϕ(t0)\phi(t)=\phi(t_{0}) for all t>t0t>t_{0}. Thus, in this case ϕ(t)\phi(t) converges to x∗:=ϕ(t0)x^{*}:=\phi(t_{0}). The rate of convergence is established by taking t→t0t\to t_{0} in (3.3). If t0=∞t_{0}=\infty, then (3.3) proves the Cauchy property of the flow ϕ\phi, which shows that ϕ(t)\phi(t) converges to some x∗∈Bx^{*}\in B as t→∞t\to\infty. Again, taking t→∞t\to\infty in (3.3) proves the rate of convergence. ∎

Proof of Theorem 1.2

If f(x0)=0f(x_{0})=0, then ∇f(x0)=0\nabla f(x_{0})=0 and therefore xk=x0x_{k}=x_{0} for all kk, and there is nothing to prove. So, let us assume that f(x0)>0f(x_{0})>0. The following lemma is the key step in the proof of Theorem 1.2.

For all k≥0k\geq 0, xk∈B(x0,r)x_{k}\in B(x_{0},r).

The proof of this lemma will be carried out via induction on kk. We have x0∈B(x0,r)x_{0}\in B(x_{0},r). Suppose that x1,…,xk−1∈B(x0,r)x_{1},\ldots,x_{k-1}\in B(x_{0},r) for some k≥1k\geq 1. We will use this hypothesis to show that xk∈B(x0,r)x_{k}\in B(x_{0},r), with kk remaining fixed henceforth.

Then for all 0≤j≤k−10\leq j\leq k-1, ∣Rj∣≤ϵη∣∇f(xj)∣2|R_{j}|\leq\epsilon\eta|\nabla f(x_{j})|^{2}.

Since xk−1∈B(x0,r)x_{k-1}\in B(x_{0},r), the assumed upper bound on η\eta implies that

Thus, ∣xk−xk−1∣≤r|x_{k}-x_{k-1}|\leq r, and therefore, xk∈B(x0,2r)x_{k}\in B(x_{0},2r). Consequently, the line segment joining xk−1x_{k-1} and xkx_{k} lies entirely in B(x0,2r)B(x_{0},2r). Since x0,…,xk−1∈B(x0,r)x_{0},\ldots,x_{k-1}\in B(x_{0},r), the line segments joining xjx_{j} and xj+1x_{j+1} lies in B(x0,r)B(x_{0},r) for all j≤k−2j\leq k-2. Thus, by Taylor expansion, we have that for any 0≤j≤k−10\leq j\leq k-1,

where xj∗x_{j}^{*} is a point on the line segment joining xjx_{j} and xj+1x_{j+1}, and ∇2f(xj∗)\nabla^{2}f(x_{j}^{*}) is the Hessian matrix of ff at xj∗x_{j}^{*}. Since L2L_{2} is an upper bound on the magnitudes of all second order derivatives of ff in B(x0,2r)B(x_{0},2r), and xj∗∈B(x0,2r)x_{j}^{*}\in B(x_{0},2r), this gives

which proves that ∣Rj∣≤12L2pη2∣∇f(xj)∣2|R_{j}|\leq\frac{1}{2}L_{2}p\eta^{2}|\nabla f(x_{j})|^{2}. Since L2pη≤2ϵL_{2}p\eta\leq 2\epsilon, this proves the claim. ∎

We have (1−ϵ)αη≤1(1-\epsilon)\alpha\eta\leq 1, and for any 0≤j≤k0\leq j\leq k,

Since xj∈B(x0,r)x_{j}\in B(x_{0},r) for j≤k−1j\leq k-1, Lemma 4.2 and the definition of α\alpha imply that for j≤k−1j\leq k-1,

Since f(x0)>0f(x_{0})>0 and f(x1)≥0f(x_{1})\geq 0, we can take j=0j=0 above and divide both sides by f(x0)f(x_{0}) to get that (1−ϵ)αη≤1(1-\epsilon)\alpha\eta\leq 1. Iterating the above inequality gives the desired upper bound for f(xj)f(x_{j}). ∎

Rearranging terms, we get the desired inequality. ∎

Let κ:=(1−ϵ)−1\kappa:=(1-\epsilon)^{-1}. By Lemma 4.4,

Also by Lemma 4.4, f(xj)≥f(xj+1)f(x_{j})\geq f(x_{j+1}) for each j≤k−1j\leq k-1. Thus, for any j≤k−1j\leq k-1, we use the Cauchy–Schwarz inequality to get

Using Lemma 4.3 to bound the terms on the right side, we get

Now, for t∈t\in, we have the inequality 1−t≤(1−t/2)21-t\leq(1-t/2)^{2}. This gives

Plugging this into the previous display completes the proof. ∎

We are now ready to complete the proof of Lemma 4.1 and then use it prove Theorem 1.2.

Applying Lemma 4.5 with j=0j=0, and recalling the criterion (1.2) used for choosing ϵ\epsilon, we get

This proves that xk∈B(x0,r)x_{k}\in B(x_{0},r), completing the induction step. ∎

By Lemma 4.1, we know that xk∈B(x0,r)x_{k}\in B(x_{0},r) for all kk. Thus, Lemma 4.5 holds for all k≥1k\geq 1 and all j<kj<k. In particular,

Note that the bound goes to zero as j→∞j\to\infty, and has no dependence on kk. Thus, {xk}k≥0\{x_{k}\}_{k\geq 0} is a Cauchy sequence in B(x0,r)B(x_{0},r), and therefore, converges to a limit x∗∈B(x0,r)x^{*}\in B(x_{0},r). Moreover, the above bound is also a bound for ∣x∗−xj∣|x^{*}-x_{j}|, since it has no dependence on kk. Lastly, since xk∈B(x0,r)x_{k}\in B(x_{0},r) for all kk, Lemma 4.3 also holds for any jj. This shows that f(x∗)=0f(x^{*})=0 and gives the required bound for f(xj)f(x_{j}). ∎

Proof of Theorem 2.1

Using this relation and the fact that DL=1D_{L}=1, we get

Then the definition of SS shows that for any ww where S(w)≠0S(w)\neq 0,

We obtain a lower bound on the above term by simply considering those jj’s that correspond to the entries of W1W_{1}. By (5.1), this gives

Now take any ww as in the statement of Theorem 2.1, that is,

the entries of W2,…,WL−1W_{2},\ldots,W_{L-1} are all strictly positive, and

irrespective of the values of W2,…,WLW_{2},\ldots,W_{L}. Let δ\delta be the minimum of all the entries of W2,…,WL−1W_{2},\ldots,W_{L-1} and KK be the maximum. Take any w′=(W1′,b1′,…,WL′,bL′)w^{\prime}=(W_{1}^{\prime},b_{1}^{\prime},\ldots,W_{L}^{\prime},b_{L}^{\prime}) such that ∣w−w′∣≤δ/2|w-w^{\prime}|\leq\delta/2. Then the entries of W2′,…,WL−1′W_{2}^{\prime},\ldots,W_{L-1}^{\prime} are all bounded below by δ/2\delta/2 and bounded above by

Thus, by (5.3) and (5.4), we see that for any w′∈B(w,δ/2)w^{\prime}\in B(w,\delta/2) where S(w′)≠0S(w^{\prime})\neq 0,

Since this holds for every w′∈B(w,δ/2)w^{\prime}\in B(w,\delta/2), and the numbers c1,…,cL−1c_{1},\ldots,c_{L-1} have no dependence on AA, it follows that if we fix δ\delta and KK, and take AA sufficiently large, then by (5.5), we can ensure that

which is the criterion (1.1) for this problem. By Theorems 1.1 and 1.2, this completes the proof of Theorem 2.1.

Proof of Theorem 2.2

In this proof, θ1,θ2,…\theta_{1},\theta_{2},\ldots will denote arbitrary positive constants whose values depend only on cc, C1C_{1}, C2C_{2}, α\alpha, β\beta, LL, and d1,…,dLd_{1},\ldots,d_{L}. (The important thing is that these constants do not depend on the input dimension dd.)

Thus, if EE happens, and we also have that

then (1.1) holds for SS in the ball B(w,M)B(w,M). Looking at the above inequality, it is clear that we can choose M=θ5M=\theta_{5} so large that the above event is implied by the event

for some suitably defined θ6\theta_{6}. Thus, if E∩FE\cap F happens, then (1.1) holds for SS in the ball B(w,θ5)B(w,\theta_{5}). Now note that since the entries of W1W_{1} are i.i.d. N(0,c/d)\mathcal{N}(0,c/d) random variables,

Acknowledgements

I thank Persi Diaconis, Dmitriy Drusvyatskiy, John Duchi, and Lexing Ying for useful feedback, references, and suggestions.

References