Nonparametric regression using deep neural networks with ReLU activation function

Johannes Schmidt-Hieber

Introduction

Multilayer (or deep) neural networks have been successfully trained recently to achieve impressive results for complicated tasks such as object detection on images and speech recognition. Deep learning is now considered to be the state-of-the art for these tasks. But there is a lack of mathematical understanding. One problem is that fitting a neural network to data is highly non-linear in the parameters. Moreover, the function class is non-convex and different regularization methods are combined in practice.

This article is inspired by the idea to build a statistical theory that provides some understanding of these procedures. As the full method is too complex to be theoretically tractable, we need to make some selection of important characteristics that we believe are crucial for the success of the procedure.

The statistical analysis for the ReLU activation function is quite different from earlier approaches and we discuss this in more detail in the overview on related literature in Section 6. Viewed as a nonparametric method, ReLU networks have some surprising properties. To explain this, notice that deep networks with ReLU activation produce functions that are piecewise linear in the input. Nonparametric methods which are based on piecewise linear approximations are typically not able to capture higher-order smoothness in the signal and are rate-optimal only up to smoothness index two. Interestingly, ReLU activation combined with a deep network architecture achieves near minimax rates for arbitrary smoothness of the regression function.

The number of hidden layers of state-of-the-art network architectures has been growing over the past years, cf. . There are versions of the recently developed deep network ResNet which are based on 152152 layers, cf. . Our analysis indicates that for the ReLU activation function the network depth should be scaled with the sample size. This suggests, that for larger samples, additional hidden layers should be added.

Recent deep architectures include more network parameters than training samples. The well-known AlexNet for instance is based on 6060 million network parameters using only 1.2 million samples. We account for high-dimensional parameter spaces in our analysis by assuming that the number of potential network parameters is much larger than the sample size. For noisy data generated from the nonparametric regression model, overfitting does not lead to good generalization errors and incorporating regularization or sparsity in the estimator becomes essential. In the deep networks literature, one option is to make the network thinner assuming that only few parameters are non-zero (or active), cf. , Section 7.10. Our analysis shows that the number of non-zero parameters plays the role of the effective model dimension and - as is common in non-parametric regression - needs to be chosen carefully.

Existing statistical theory often requires that the size of the network parameters tends to infinity as the sample size increases. In practice, estimated network weights are, however, rather small. We can incorporate small parameters in our theory, proving that it is sufficient to consider neural networks with all network parameters bounded in absolute value by one.

Multilayer neural networks are typically applied to high-dimensional input. Without additional structure in the signal besides smoothness, nonparametric estimation rates are then slow because of the well-known curse of dimensionality. This means that no statistical procedure can do well regarding pointwise reconstruction of the signal. Multilayer neural networks are believed to be able to adapt to many different structures in the signal, therefore avoiding the curse of dimensionality and achieving faster rates in many situations. In this work, we stick to the regression setup and show that deep ReLU networks can indeed attain faster rates under a hierarchical composition assumption on the regression function, which includes (generalized) additive models and the composition models considered in .

Parts of the success of multilayer neural networks can be explained by the fast algorithms that are available to estimate the network weights from data. These iterative algorithms are based on minimization of some empirical loss function using stochastic gradient descent. Because of the non-convex function space, gradient descent methods might get stuck in a saddle point or converge to one of the potentially many local minima. derives a heuristic argument showing that the risk of most of the local minima is not much larger than the risk of the global minimum. Despite the huge number of variations of the stochastic gradient descent, the common objective of all approaches is to reduce the empirical loss. In our framework we associate to any network reconstruction method a parameter quantifying the expected discrepancy between the achieved empirical risk and the global minimum of the energy landscape. The main theorem then states that a network estimator is minimax rate optimal (up to log factors) if and only if the method almost minimizes the empirical risk.

We also show that wavelet series estimators are unable to adapt to the underlying structure under the composition assumption on the regression function. By deriving lower bounds, it is shown that the rates are suboptimal by a polynomial factor in the sample size n.n. This provides an example of a function class for which fitting a neural network outperforms wavelet series estimators.

Our setting deviates in two aspects from the computer science literature on deep learning. Firstly, we consider regression and not classification. Secondly, we restrict ourselves in this article to multilayer feedforward artificial neural networks, while most of the many recent deep learning applications have been obtained using specific types of networks such as convolutional or recurrent neural networks.

The article is structured as follows. Section 2 introduces multilayer feedforward artificial neural networks and discusses mathematical modeling. This section also contains the definition of the network classes. The considered function classes for the regression function and the main result can be found in Section 3. Specific structural constraints such as additive models are discussed in Section 4. In Section 5 it is shown that wavelet estimators can only achieve suboptimal rates under the composition assumption. We give an overview of relevant related literature in Section 6. The proof of the main result together with additional discussion can be found in Section 7.

Notation: Vectors are denoted by bold letters, e.g. x:=(x1,,xd).\mathbf{x}:=(x_{1},\ldots,x_{d})^{\top}. As usual, we define xp:=(i=1dxip)1/p,|\mathbf{x}|_{p}:=(\sum_{i=1}^{d}|x_{i}|^{p})^{1/p}, x:=maxixi,|\mathbf{x}|_{\infty}:=\max_{i}|x_{i}|, x0:=i1(xi0),|\mathbf{x}|_{0}:=\sum_{i}\mathbf{1}(x_{i}\neq 0), and write fp:=fLp(D)\|f\|_{p}:=\|f\|_{L^{p}(D)} for the LpL^{p}-norm on D,D, whenever there is no ambiguity of the domain D.D. For two sequences (an)n(a_{n})_{n} and (bn)n,(b_{n})_{n}, we write anbna_{n}\lesssim b_{n} if there exists a constant CC such that anCbna_{n}\leq Cb_{n} for all n.n. Moreover, anbna_{n}\asymp b_{n} means that anbna_{n}\lesssim b_{n} and bnan.b_{n}\lesssim a_{n}. We denote by log2\log_{2} the logarithm with respect to the basis two and write x\lceil x\rceil for the smallest integer x.\geq x.

Mathematical definition of multilayer neural networks

In computer science, neural networks are more commonly introduced via their representation as directed acyclic graphs, cf. Figure 1. Using this equivalent definition, the nodes in the graph (also called units) are arranged in layers. The input layer is the first layer and the output layer the last layer. The layers that lie in between are called hidden layers. The number of hidden layers corresponds to LL and the number of units in each layer generates the width vector p.\mathbf{p}. Each node/unit in the graph representation stands for a scalar product of the incoming signal with a weight vector which is then shifted and applied to the activation function.

Mathematical modeling of deep network characteristics: Given a network function f(x)=WLσvLWL1σvL1W1σv1W0x,f(\mathbf{x})=W_{L}\sigma_{\mathbf{v}_{L}}W_{L-1}\sigma_{\mathbf{v}_{L-1}}\cdots W_{1}\sigma_{\mathbf{v}_{1}}W_{0}\mathbf{x}, the network parameters are the entries of the matrices (Wj)j=0,,L(W_{j})_{j=0,\ldots,L} and vectors (vj)j=1,,L.(\mathbf{v}_{j})_{j=1,\ldots,L}. These parameters need to be estimated/learned from the data.

The aim of this article is to consider a framework that incorporates essential features of modern deep network architectures. In particular, we allow for large depth LL and large number of potential network parameters. For the main result, no upper bound on the number of network parameters is needed. Thus we consider high-dimensional settings with more parameters than training data.

Another characteristic of trained networks is that the size of the learned network parameters is typically not very large. Common network initialization methods initialize the weight matrices WjW_{j} by a (nearly) orthogonal random matrix if two successive layers have the same width, cf. , Section 8.4. In practice, the trained network weights are typically not far from the initialized weights. Since in an orthogonal matrix all entries are bounded in absolute value by one, also the trained network weights will not be large.

Existing theoretical results, however, often require that the size of the network parameters tends to infinity. If large parameters are allowed, one can for instance easily approximate step functions by ReLU networks. To be more in line with what is observed in practice, we consider networks with all parameters bounded by one. This constraint can be easily build into the deep learning algorithm by projecting the network parameters in each iteration onto the interval ..

If Wj\|W_{j}\|_{\infty} denotes the maximum-entry norm of Wj,W_{j}, the space of network functions with given network architecture and network parameters bounded by one is

with the convention that v0\mathbf{v}_{0} is a vector with all components equal to zero.

In deep learning, sparsity of the neural network is enforced through regularization or specific forms of networks. Dropout for instance sets randomly units to zero and has the effect that each unit will be active only for a small fraction of the data, cf. , Section 7.2. In our notation this means that each entry of the vectors σvkWk1W1σv1W0x,\sigma_{\mathbf{v}_{k}}W_{k-1}\cdots W_{1}\sigma_{\mathbf{v}_{1}}W_{0}\mathbf{x}, k=1,,Lk=1,\ldots,L is zero over a large range of the input space xd.\mathbf{x}\in^{d}. Convolutional neural networks filter the input over local neighborhoods. Rewritten in the form (2) this essentially means that the WiW_{i} are banded Toeplitz matrices. All network parameters corresponding to higher off-diagonal entries are thus set to zero.

In this work we model the network sparsity assuming that there are only few non-zero/active network parameters. If Wj0\|W_{j}\|_{0} denotes the number of non-zero entries of WjW_{j} and f\||f|_{\infty}\|_{\infty} stands for the sup-norm of the function xf(x),\mathbf{x}\mapsto|f(\mathbf{x})|_{\infty}, then the ss-sparse networks are given by

The upper bound on the uniform norm of ff is most of the time dispensable and therefore omitted in the notation. We consider cases where the number of network parameters ss is small compared to the total number of parameters in the network.

In deep learning, it is common to apply variations of stochastic gradient descent combined with other techniques such as dropout to the loss induced by the log-likelihood (see Section 6.2.1.1 in ). For nonparametric regression with normal errors, this coincides with the least-squares loss (in machine learning terms this is the cross-entropy for this model, cf. , p.129). The common objective of all reconstruction methods is to find networks ff with small empirical risk 1ni=1n(Yif(Xi))2.\tfrac{1}{n}\sum_{i=1}^{n}(Y_{i}-f(\mathbf{X}_{i}))^{2}. For any estimator f^n\widehat{f}_{n} that returns a network in the class F(L,p,s,F)\mathcal{F}(L,\mathbf{p},s,F) we define the corresponding quantity

To evaluate the statistical performance of an estimator f^n,\widehat{f}_{n}, we derive bounds for the prediction error

with X=DX1\mathbf{X}\stackrel{{\scriptstyle\mathcal{D}}}{{=}}\mathbf{X}_{1} being independent of the sample (Xi,Yi)i.(\mathbf{X}_{i},Y_{i})_{i}.

The term Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) can be related via empirical process theory to constant ×(R(f^n,f0)R(f^nERM,f0))+\times(R(\widehat{f}_{n},f_{0})-R(\widehat{f}_{n}^{\operatorname{ERM}},f_{0}))+ remainder, with f^nERM\widehat{f}_{n}^{\operatorname{ERM}} an empirical risk minimizer. Therefore, Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) is the key quantity that together with the minimax estimation rate sharply determines the convergence rate of f^n\widehat{f}_{n} (up to logn\log n-factors). Determining the decay of Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) in nn for commonly employed methods such as stochastic gradient descent is an interesting problem in its own. We only sketch a possible proof strategy here. Because of the potentially many local minima and saddle points of the loss surface or energy landscape, gradient descent based methods have only a small chance to reach the global minimum without getting stuck in a local minimum first. By making a link to spherical spin glasses, provide a heuristic suggesting that the loss of any local minima lies in a band that is lower bounded by the loss of the global minimum. The width of the band depends on the width of the network. If the heuristic argument can be made rigorous, then the width of the band provides an upper bound for Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) for all methods that converge to a local minimum. This would allow us then to study deep learning without an explicit analysis of the algorithm. For more on the energy landscape, see .

Main results

The theoretical performance of neural networks depends on the underlying function class. The classical approach in nonparametric statistics is to assume that the regression function is β\beta-smooth. The minimax estimation rate for the prediction error is then n2β/(2β+d).n^{-2\beta/(2\beta+d)}. Since the input dimension dd in neural network applications is very large, these rates are extremely slow. The huge sample sizes often encountered in deep learning applications are by far not sufficient to compensate the slow rates.

We therefore consider a function class that is natural for neural networks and exhibits some low-dimensional structure that leads to input dimension free exponents in the estimation rates. We assume that the regression function f0f_{0} is a composition of several functions, that is,

with gi:[ai,bi]di[ai+1,bi+1]di+1.g_{i}:[a_{i},b_{i}]^{d_{i}}\rightarrow[a_{i+1},b_{i+1}]^{d_{i+1}}. Denote by gi=(gij)j=1,,di+1g_{i}=(g_{ij})_{j=1,\ldots,d_{i+1}}^{\top} the components of gig_{i} and let tit_{i} be the maximal number of variables on which each of the gijg_{ij} depends on. Thus, each gijg_{ij} is a tit_{i}-variate function. As an example consider the function f0(x1,x2,x3)=g11(g01(x1,x3),g02(x1,x2))f_{0}(x_{1},x_{2},x_{3})=g_{11}(g_{01}(x_{1},x_{3}),g_{02}(x_{1},x_{2})) for which d0=3,d_{0}=3, t0=2,t_{0}=2, d1=t1=2,d_{1}=t_{1}=2, d2=1.d_{2}=1. We always must have tidit_{i}\leq d_{i} and for specific constraints such as additive models, tit_{i} might be much smaller than di.d_{i}. The single components g0,,gqg_{0},\ldots,g_{q} and the pairs (βi,ti)(\beta_{i},t_{i}) are obviously not identifiable. As we are only interested in estimation of f0f_{0} this causes no problems. Among all possible representations, one should always pick one that leads to the fastest estimation rate in Theorem 1 below.

It is conceivable that for many of the problems for which neural networks perform well a hidden hierarchical input-output relationship of the form (6) is present with small values ti,t_{i}, cf. . Slightly more specific function spaces, which alternate between summations and composition of functions, have been considered in . We provide below an example of a function class that can be decomposed in the form (6) but is not contained in these spaces.

Recall that a function has Hölder smoothness index β\beta if all partial derivatives up to order β\lfloor\beta\rfloor exist and are bounded and the partial derivatives of order β\lfloor\beta\rfloor are ββ\beta-\lfloor\beta\rfloor Hölder, where β\lfloor\beta\rfloor denotes the largest integer strictly smaller than β.\beta. The ball of β\beta-Hölder functions with radius KK is then defined as

We assume that each of the functions gijg_{ij} has Hölder smoothness βi.\beta_{i}. Since gijg_{ij} is also tit_{i}-variate, gijCtiβi([ai,bi]ti,Ki)g_{ij}\in\mathcal{C}_{t_{i}}^{\beta_{i}}([a_{i},b_{i}]^{t_{i}},K_{i}) and the underlying function space becomes

with d:=(d0,,dq+1),\mathbf{d}:=(d_{0},\ldots,d_{q+1}), t:=(t0,,tq),\mathbf{t}:=(t_{0},\ldots,t_{q}), β:=(β0,,βq).\bm{\beta}:=(\beta_{0},\ldots,\beta_{q}).

For estimation rates in the nonparametric regression model, the crucial quantity is the smoothness of f.f. Imposing smoothness on the functions gi,g_{i}, we must then find the induced smoothness on f.f. If, for instance, q=1,q=1, β0,β11,\beta_{0},\beta_{1}\leq 1, d0=d1=t0=t1=1,d_{0}=d_{1}=t_{0}=t_{1}=1, then f=g1g0f=g_{1}\circ g_{0} and ff has smoothness β0β1,\beta_{0}\beta_{1}, cf. . We should then be able to achieve at least the convergence rate n2β0β1/(2β0β1+1).n^{-2\beta_{0}\beta_{1}/(2\beta_{0}\beta_{1}+1)}. For β1>1,\beta_{1}>1, the rate changes. Below we see that the convergence of the network estimator is described by the effective smoothness indices

Recall the definition of Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) in (5). We can now state the main result.

Consider the dd-variate nonparametric regression model (1) for composite regression function (6) in the class G(q,d,t,β,K).\mathcal{G}(q,\mathbf{d},\mathbf{t},\bm{\beta},K). Let f^n\widehat{f}_{n} be an estimator taking values in the network class F(L,(pi)i=0,,L+1,s,F)\mathcal{F}(L,(p_{i})_{i=0,\ldots,L+1},s,F) satisfying

i=0qlog2(4ti4βi)log2nLnϕn,\sum_{i=0}^{q}\log_{2}(4t_{i}\vee 4\beta_{i})\log_{2}n\leq L\lesssim n\phi_{n},

nϕnmini=1,,Lpi,n\phi_{n}\lesssim\min_{i=1,\ldots,L}p_{i},

There exist constants C,CC,C^{\prime} only depending on q,d,t,β,F,q,\mathbf{d},\mathbf{t},\bm{\beta},F, such that if Δn(f^n,f0)CϕnLlog2n,\Delta_{n}(\widehat{f}_{n},f_{0})\leq C\phi_{n}L\,\log^{2}n, then

and if Δn(f^n,f0)CϕnLlog2n,\Delta_{n}(\widehat{f}_{n},f_{0})\geq C\phi_{n}L\,\log^{2}n, then

In order to minimize the rate ϕnLlog2n,\phi_{n}L\,\log^{2}n, the best choice is to choose LL of the order of log2n.\log_{2}n. The rate in the regime Δn(f^n,f0)Cϕnlog3n\Delta_{n}(\widehat{f}_{n},f_{0})\leq C\phi_{n}\,\log^{3}n becomes then

The convergence rate in Theorem 1 depends on ϕn\phi_{n} and Δn(f^n,f0).\Delta_{n}(\widehat{f}_{n},f_{0}). Below we show that ϕn\phi_{n} is a lower bound for the minimax estimation risk over this class. Recall that the term Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) is large if f^n\widehat{f}_{n} has a large empirical risk compared to an empirical risk minimizer. Having this term in the convergence rate is unavoidable as it also appears in the lower bound in (9). Since for any empirical risk minimizer the Δn\Delta_{n}-term is zero by definition, we have the following direct consequence of the main theorem.

Let f~nargminfF(L,p,s,F)i=1n(Yif(Xi))2\widetilde{f}_{n}\in\mathop{\rm arg\min}_{f\in\mathcal{F}(L,\mathbf{p},s,F)}\sum_{i=1}^{n}(Y_{i}-f(\mathbf{X}_{i}))^{2} be an empirical risk minimizer. Under the same conditions as for Theorem 1, there exists a constant C,C^{\prime}, only depending on q,d,t,β,F,q,\mathbf{d},\mathbf{t},\bm{\beta},F, such that

The number of network parameters in a fully connected network is of the order i=0Lpipi+1.\sum_{i=0}^{L}p_{i}p_{i+1}. This shows that Theorem 1 requires sparse networks. More specifically, the network has at least i=1Lpis\sum_{i=1}^{L}p_{i}-s completely inactive nodes, meaning that all incoming signal is zero. The choice snϕnlogns\asymp n\phi_{n}\log n in condition (iv)(iv) balances the squared bias and the variance. From the proof of the theorem convergence rates can also be derived if ss is chosen of a different order.

For convenience, Theorem 1 is stated without explicit constants. The proofs, however, are non-asymptotic although we did not make an attempt to minimize the constants. It is well-known that deep learning outperforms other methods only for large sample sizes. This indicates that the method might be able to adapt to underlying structure in the signal and therefore achieving fast convergence rates but with large constants or remainder terms which spoil the results for small samples.

The proof of the risk bounds in Theorem 1 is based on the following oracle-type inequality.

Consider the dd-variate nonparametric regression model (1) with unknown regression function f0,f_{0}, satisfying f0F\|f_{0}\|_{\infty}\leq F for some F1.F\geq 1. Let f^n\widehat{f}_{n} be any estimator taking values in the class F(L,p,s,F)\mathcal{F}(L,\mathbf{p},s,F) and let Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) be the quantity defined in (5). For any ε(0,1],\varepsilon\in(0,1], there exists a constant Cε,C_{\varepsilon}, only depending on ε,\varepsilon, such that with

One consequence of the oracle inequality is that the upper bounds on the risk become worse if the number of layers increases. In practice it also has been observed that too many layers lead to a degradation of the performance, cf. , , Section 4.4 and , Section 4. Residual networks can overcome this problem. But they are not of the form (2) and will need to be analyzed separately.

One may wonder whether there is anything special about ReLU networks compared to other activation functions. A close inspection of the proof shows that two specific properties of the ReLU function are used.

One of the advantages of deep ReLU networks is the projection property

that we can use to pass a signal without change through several layers in the network. This is important since the approximation theory is based on the construction of smaller networks for simpler tasks that might not all have the same network depth. To combine these subnetworks we need to synchronize the network depths by adding hidden layers that do not change the output. This can be easily realized by choosing the weight matrices in the network to be the identity (assuming equal network width in successive layers) and using (11), see also (18). This property is not only a theoretical tool. To pass an outcome without change to a deeper layer is also often helpful in practice and realized by so called skip connections in which case they do not need to be learned from the data. A specific instance are residual networks with ReLU activation function that are successfully applied in practice. The difference to standard feedforward networks is that if all networks parameters are set to zero in a residual network, the network becomes essentially the identity map. For other activation functions it is much harder to approximate the identity.

Another advantage of the ReLU activation is that all network parameters can be taken to be bounded in absolute value by one. If all network parameters are initialized by a value in ,, this means that each network parameter only need to be varied by at most two during training. It is unclear whether other results in the literature for non-ReLU activation functions hold for bounded network parameters. An important step is the approximation of the square function xx2.x\mapsto x^{2}. For any twice differentiable and non-linear activation function, the classical approach to approximate the square function by a network is to use rescaled second order differences (σ(t+2xh)2σ(t+xh)+σ(xh))/(h2σ(t))x2(\sigma(t+2xh)-2\sigma(t+xh)+\sigma(xh))/(h^{2}\sigma^{\prime\prime}(t))\rightarrow x^{2} for h0h\rightarrow 0 and a tt with σ(t)0\sigma^{\prime\prime}(t)\neq 0. To achieve a sufficiently good approximation, we need to let hh tend to zero with the sample size, making some of the network parameters necessarily very large.

The log2n\log^{2}n-factor in the convergence rate ϕnLlog2n\phi_{n}L\log^{2}n is likely an artifact of the proof. Next we show that ϕn\phi_{n} is a lower bound for the minimax estimation risk over the class G(q,d,t,β,K)\mathcal{G}(q,\mathbf{d},\mathbf{t},\bm{\beta},K) in the interesting regime timin(d0,,di1)t_{i}\leq\min(d_{0},\ldots,d_{i-1}) for all i.i. This means that no dimensions are added on deeper abstraction levels in the composition of functions. In particular, it avoids that tit_{i} is larger than the input dimension d0.d_{0}. Outside of this regime, it is hard to determine the minimax rate and in some cases it is even possible to find another representation of ff as a composition of functions which yields a faster convergence rate.

Consider the nonparametric regression model (1) with Xi\mathbf{X}_{i} drawn from a distribution with Lebesgue density on d^{d} which is lower and upper bounded by positive constants. For any non-negative integer q,q, any dimension vectors d\mathbf{d} and t\mathbf{t} satisfying timin(d0,,di1)t_{i}\leq\min(d_{0},\ldots,d_{i-1}) for all i,i, any smoothness vector β\bm{\beta} and all sufficiently large constants K>0,K>0, there exists a positive constant c,c, such that

where the inf\inf is taken over all estimators f^n.\widehat{f}_{n}.

The L2L^{2}-minimax rate coincides in most regimes with the sup-norm rate obtained in Section 4.1 of for composition of two functions. But unlike the classical nonparametric regression model, the minimax estimation rates for L2L^{2}-loss and sup-norm loss differ for some setups by a polynomial power in the sample size n.n.

There are several recent results in approximation theory that provide lower bounds on the number of required network weights ss such that all functions in a function class can be approximated by a ss-sparse network up to some prescribed error, cf. for instance . Results of this flavor can also be quite easily derived by combining the minimax lower bound with the oracle inequality. The argument is that if the same approximation rates would hold for networks with less parameters then we would obtain rates that are faster than the minimax rates, which clearly is a contradiction. This provides a new statistical route to establish approximation theoretic properties.

for some εc2,\varepsilon\leq c_{2}, then for any width vector p\mathbf{p} with p0=dp_{0}=d and pL+1=1,p_{L+1}=1,

A more refined argument using Lemma 4 instead of Theorem 2 yields also lower bounds for L2.L^{2}.

Examples of specific structural constraints

In this section we discuss several well-studied special cases of compositional constraints on the regression function.

Additive models: In an additive model the regression function has the form

This can be written as a composition of functions

Suppose that fiC1β(,K)f_{i}\in\mathcal{C}_{1}^{\beta}(,K) for i=1,,d.i=1,\ldots,d. Then, f:dg0[K,K]dg1[Kd,Kd].f:\,^{d}\stackrel{{\scriptstyle g_{0}}}{{\longrightarrow}}[-K,K]^{d}\stackrel{{\scriptstyle g_{1}}}{{\longrightarrow}}[-Kd,Kd]. Since for any γ>1,\gamma>1, g1Cdγ([K,K]d,(K+1)d),g_{1}\in\mathcal{C}_{d}^{\gamma}([-K,K]^{d},(K+1)d),

For network architectures F(L,p,s,F)\mathcal{F}(L,\mathbf{p},s,F) satisfying F(K+1)d,F\geq(K+1)d, 2log2(4(β2)d)lognLlogn,2\log_{2}(4(\beta\vee 2)d)\log n\leq L\lesssim\log n, n1/(2β+1)minipin^{1/(2\beta+1)}\lesssim\min_{i}p_{i} and sn1/(2β+1)logn,s\asymp n^{1/(2\beta+1)}\log n, we thus obtain by Theorem 1,

This coincides up to the log3n\log^{3}n-factor with the minimax estimation rate.

Generalized additive models: Suppose the regression function is of the form

For network architectures satisfying the assumptions of Theorem 1, the bound on the estimation rate becomes

Theorem 3 shows that n2β(γ1)/(2β(γ1)+1)+n2γ/(2γ+1)n^{-2\beta(\gamma\wedge 1)/(2\beta(\gamma\wedge 1)+1)}+n^{-2\gamma/(2\gamma+1)} is also a lower bound. Let us also remark that for the special case β=γ2\beta=\gamma\geq 2 and β,γ\beta,\gamma integers, Theorem 2.1 of establishes the estimation rate n2β/(2β+1).n^{-2\beta/(2\beta+1)}.

Sparse tensor decomposition: Assume that the regression function f0f_{0} has the form

For networks with architectures satisfying 3log2(4(β+1)(d+1)N)log2nLlogn,3\log_{2}(4(\beta+1)(d+1)N)\log_{2}n\leq L\lesssim\log n, n1/(2β+1)minipin^{1/(2\beta+1)}\lesssim\min_{i}p_{i} and sn1/(2β+1)logn,s\asymp n^{1/(2\beta+1)}\log n, Theorem 1 yields the rate

and the exponent in the rate does not depend on the input dimension d.d.

Suboptimality of wavelet series estimators

As argued before the composition assumption in (6) is very natural and generalizes many structural constraints such as additive models. In this section, we show that wavelet series estimators are unable to take advantage from the underlying composition structure in the regression function and achieve in some setups much slower convergence rates.

More specifically, we consider general additive models of the form f0(x)=h(x1++xd)f_{0}(\mathbf{x})=h(x_{1}+\ldots+x_{d}) with hC1α([0,d],K).h\in\mathcal{C}_{1}^{\alpha}([0,d],K). This can also be viewed as a special instance of the single index model, where the aim is not to estimate hh but f0.f_{0}. Using (13), the prediction error of neural network reconstructions with small empirical risk and depth LlognL\asymp\log n is then bounded by n2α/(2α+1)log3n.n^{-2\alpha/(2\alpha+1)}\log^{3}n. The lower bound below shows that wavelet series estimators cannot converge faster than with the rate n2α/(2α+d).n^{-2\alpha/(2\alpha+d)}. This rate can be much slower if dd is large. Wavelet series estimators thus suffer in this case from the curse of dimensionality while neural networks achieve fast rates.

To construct a counterexample, it is enough to consider the nonparametric regression model Yi=f0(Xi)+εi,Y_{i}=f_{0}(\mathbf{X}_{i})+\varepsilon_{i}, i=1,,ni=1,\ldots,n with uniform design Xi:=(Ui,1,,Ui,d)Unifd.\mathbf{X}_{i}:=(U_{i,1},\ldots,U_{i,d})\sim\operatorname{Unif}^{d}. The empirical wavelet coefficients are

Because of E[d^λ1λd(f0)]=dλ1λd(f0),E[\widehat{d}_{\bm{\lambda}_{1}\ldots\bm{\lambda}_{d}}(f_{0})]=d_{\bm{\lambda}_{1}\ldots\bm{\lambda}_{d}}(f_{0}), this gives unbiased estimators for the wavelet coefficients. By the law of total variance,

For the lower bounds we may assume that the smoothness indices are known. For estimation, we can truncate the series expansion on a resolution level that balances squared bias and variance of the total estimator. More generally, we study estimators of the form

for an arbitrary subset IΛ××ΛI\subset\Lambda\times\ldots\times\Lambda. Using that the design is uniform,

Let qq be as above and set ν:=log2d+1.\nu:=\lceil\log_{2}d\rceil+1. For any 0<α10<\alpha\leq 1 and any K>0,K>0, there exists a non-zero constant c(ψ,d)c(\psi,d) only depending on dd and properties of the wavelet function ψ\psi such that for any j,j, we can find a function fj,α(x)=hj,α(x1++xd)f_{j,\alpha}(\mathbf{x})=h_{j,\alpha}(x_{1}+\ldots+x_{d}) with hj,αC1α([0,d],K)h_{j,\alpha}\in\mathcal{C}_{1}^{\alpha}([0,d],K) satisfying

for all p1,,pd{0,1,,2jqν1}.p_{1},\ldots,p_{d}\in\{0,1,\ldots,2^{j-q-\nu}-1\}.

If f^n\widehat{f}_{n} denotes the wavelet estimator (15) for a compactly supported wavelet ψ\psi and an arbitrary index set I,I, then, for any 0<α10<\alpha\leq 1 and any Hölder radius K>0,K>0,

A close inspection of the proof shows that the theorem even holds for 0<αr0<\alpha\leq r with rr the smallest positive integer for which xrψ(x)dx0.\int x^{r}\psi(x)dx\neq 0.

A brief summary of related statistical theory for neural networks

This section is intended as a condensed overview on related literature summarizing main proving strategies for bounds on the statistical risk. An extended summary of the work until the late 90’s is given in . To control the stochastic error of neural networks, bounds on the covering entropy and VC dimension can be found in the monograph . A challenging part in the analysis of neural networks is the approximation theory for multivariate functions. We first recall results for shallow neural networks, that is, neural networks with one hidden layer.

Shallow neural networks: A shallow network with one output unit and width vector (d,m,1)(d,m,1) can be written as

The universal approximation theorem states that a neural network with one hidden layer can approximate any continuous function ff arbitrarily well with respect to the uniform norm provided there are enough hidden units, cf. . If ff has a derivative ff^{\prime}, then the derivative of the neural network also approximates f.f^{\prime}. The number of required hidden units might be, however, extremely large, cf. and . There are several proofs for the universal approximation theorem based on the Fourier transform, the Radon transform and the Hahn-Banach theorem .

The proofs can be sharpened in order to obtain rates of convergence. In the convergence rate n2β/(2β+d+5)n^{-2\beta/(2\beta+d+5)} is derived. Compared with the minimax estimation rate this is suboptimal by a polynomial factor. The reason for the loss of performance with this approach is that rewriting the function as a network requires too many parameters.

In a similar strategy is used to derive the rate Cf(dlognn)1/2C_{f}(d\tfrac{\log n}{n})^{1/2} for the squared L2L^{2}-risk, where Cf:=ω1Ff(ω)dωC_{f}:=\int|\omega|_{1}|\mathcal{F}f(\omega)|d\omega and Ff\mathcal{F}f denotes the Fourier transform of f.f. If Cf<C_{f}<\infty and dd is fixed the rate is always n1/2n^{-1/2} up to logarithmic factors. Since iifCf,\sum_{i}\|\partial_{i}f\|_{\infty}\leq C_{f}, this means that Cf<C_{f}<\infty can only hold if ff has Hölder smoothness at least one. This rate is difficult to compare with the standard nonparametric rates except for the special case d=1,d=1, where the rate is suboptimal compared with the minimax rate n2/(2+d)n^{-2/(2+d)} for dd-variate functions with smoothness one. More interestingly, the rate Cf(dlognn)1/2C_{f}(d\tfrac{\log n}{n})^{1/2} shows that neural networks can converge fast if the underlying function satisfies some additional structural constraint. The same rate can also be obtained by a Fourier series estimator, see , Section 1.7. In a similar fashion, studies abstract function spaces on which shallow networks achieve fast convergence rates.

Results for multilayer neural networks: In it is shown how to approximate a polytope by a neural network with two hidden layers. Based on this result, uses two-layer neural networks with sigmoidal activation function and achieves the nonparametric rate n2β/(2β+d)n^{-2\beta/(2\beta+d)} up to logn\log n-factors for β1.\beta\leq 1. This is extended in to a composition assumption and further generalized to β>1\beta>1 in the recent article . Unfortunately, the result requires that the activation function is at least as smooth as the signal, cf. Theorem 1 in and therefore rules out the ReLU activation function.

The activation function σ(x)=x2\sigma(x)=x^{2} is not of practical relevance but has some interesting theory. Indeed, with one hidden layer, we can generate quadratic polynomials and with LL hidden layers polynomials of degree 2L.2^{L}. For this activation function, the role of the network depth is the polynomial degree and we can use standard results to approximate functions in common function classes. A natural generalization is the class of activation functions satisfying limxxkσ(x)=0\lim_{x\rightarrow-\infty}x^{-k}\sigma(x)=0 and limx+xkσ(x)=1.\lim_{x\rightarrow+\infty}x^{-k}\sigma(x)=1.

If the growth is at least quadratic (k2k\geq 2), the approximation theory has been derived in for deep networks with number of layers scaling with logd.\log d. The same class has also been considered recently in . For the approximations to work, the assumption k2k\geq 2 is crucial and the same approach does not generalize to the ReLU activation function, which satisfies the growth condition with k=1k=1, and always produces functions that are piecewise linear in the input.

Approximation theory for the ReLU activation function has been only recently developed in . The key observation is that there are specific deep networks with few units which approximate the square function well. In particular, the function approximation presented in is essential for our approach and we use a similar strategy to construct networks that are close to a given function. We are, however, interested in a somehow different question. Instead of deriving existence of a network architecture with good approximation properties, we show that for any network architecture satisfying the conditions of Theorem 1 good approximation rates are obtainable. An additional difficulty in our approach is the boundedness of the network parameters.

Proofs

For the approximation of a function by a network, we first construct smaller networks computing simpler objects. Let p=(p0,,pL+1)\mathbf{p}=(p_{0},\ldots,p_{L+1}) and p=(p0,,pL+1).\mathbf{p}^{\prime}=(p_{0}^{\prime},\ldots,p_{L+1}^{\prime}). To combine networks, we make frequently use of the following rules.

Enlarging: F(L,p,s)F(L,q,s)\mathcal{F}(L,\mathbf{p},s)\subseteq\mathcal{F}(L,\mathbf{q},s^{\prime}) whenever pq\mathbf{p}\leq\mathbf{q} componentwise and ss.s\leq s^{\prime}.

Additional layers/depth synchronization: To synchronize the number of hidden layers for two networks, we can add additional layers with identity weight matrix, such that

Parallelization: Suppose that f,gf,g are two networks with the same number of hidden layers and the same input dimension, that is, fF(L,p)f\in\mathcal{F}(L,\mathbf{p}) and gF(L,p)g\in\mathcal{F}(L,\mathbf{p}^{\prime}) with p0=p0.p_{0}=p_{0}^{\prime}. The parallelized network (f,g)(f,g) computes ff and gg simultaneously in a joint network in the class F(L,(p0,p1+p1,,pL+1+pL+1)).\mathcal{F}(L,(p_{0},p_{1}+p_{1}^{\prime},\ldots,p_{L+1}+p_{L+1}^{\prime})).

To see this, let f(x)=WLσvLWL1σv1W0xF(L,p,s).f(\mathbf{x})=W_{L}\sigma_{\mathbf{v}_{L}}W_{L-1}\ldots\sigma_{\mathbf{v}_{1}}W_{0}\mathbf{x}\in\mathcal{F}(L,\mathbf{p},s). If all entries of the jj-th column of WiW_{i} are zero, then we can remove this column together with the jj-th row in Wi1W_{i-1} and the jj-th entry of vi\mathbf{v}_{i} without changing the function. This shows then that fF(L,(p0,,pi1,pi1,pi+1,,pL+1),s).f\in\mathcal{F}(L,(p_{0},\ldots,p_{i-1},p_{i}-1,p_{i+1},\ldots,p_{L+1}),s). Because there are ss active parameters, we can iterate this procedure at least pisp_{i}-s times for any i=1,,L.i=1,\ldots,L. This proves f\in\mathcal{F}\big{(}L,(p_{0},p_{1}\wedge s,p_{2}\wedge s,\ldots,p_{L}\wedge s,p_{L+1}),s\big{)}.

For any function fCrβ(r,K)f\in\mathcal{C}_{r}^{\beta}(^{r},K) and any integers m1m\geq 1 and N(β+1)r(K+1)er,N\geq(\beta+1)^{r}\vee(K+1)e^{r}, there exists a network

The proof of the theorem is given in the supplement. The idea is to first build networks that for given input (x,y)(x,y) approximately compute the product xy.xy. We then split the input space into small hyper-cubes and construct a network that approximates a local Taylor expansion on each of these hyper-cubes.

Based on Theorem 5, we can now construct a network that approximates f=gqg0.f=g_{q}\circ\ldots\circ g_{0}. In a first step, we show that ff can always be written as composition of functions defined on hypercubes ti.^{t_{i}}. As in the previous theorem, let gijCtiβi([ai,bi]ti,Ki)g_{ij}\in\mathcal{C}_{t_{i}}^{\beta_{i}}([a_{i},b_{i}]^{t_{i}},K_{i}) and assume that Ki1.K_{i}\geq 1. For i=1,,q1,i=1,\ldots,q-1, define

Here, 2Ki1xKi12K_{i-1}\mathbf{x}-K_{i-1} means that we transform the entries by 2Ki1xjKi12K_{i-1}x_{j}-K_{i-1} for all j.j. Clearly,

Using the definition of the Hölder balls Crβ(D,K),C_{r}^{\beta}(D,K), it follows that h0jh_{0j} takes values in ,, h0jCt0β0(t0,1),h_{0j}\in\mathcal{C}_{t_{0}}^{\beta_{0}}(^{t_{0}},1), hijCtiβi(ti,(2Ki1)βi)h_{ij}\in\mathcal{C}_{t_{i}}^{\beta_{i}}(^{t_{i}},(2K_{i-1})^{\beta_{i}}) for i=1,,q1,i=1,\ldots,q-1, and hqjCtqβq(tq,Kq(2Kq1)βq).h_{qj}\in\mathcal{C}_{t_{q}}^{\beta_{q}}(^{t_{q}},K_{q}(2K_{q-1})^{\beta_{q}}). Without loss of generality, we can always assume that the radii of the Hölder balls are at least one, that is, Ki1.K_{i}\geq 1.

Let hijh_{ij} be as above with Ki1K_{i}\geq 1. Then, for any functions h~i=(h~ij)j\widetilde{h}_{i}=(\widetilde{h}_{ij})_{j}^{\top} with h~ij:ti,\widetilde{h}_{ij}:^{t_{i}}\rightarrow,

Define Hi=hih0H_{i}=h_{i}\circ\ldots\circ h_{0} and H~i=h~ih~0.\widetilde{H}_{i}=\widetilde{h}_{i}\circ\ldots\circ\widetilde{h}_{0}. If QiQ_{i} is an upper bound for the Hölder semi-norm of hij,h_{ij}, j=1,,di+1,j=1,\ldots,d_{i+1}, we find using triangle inequality,

Together with the inequality (y+z)αyα+zα(y+z)^{\alpha}\leq y^{\alpha}+z^{\alpha} which holds for all y,z0y,z\geq 0 and all α,\alpha\in, the result follows. ∎

It is enough to prove the result for all sufficiently large n.n. Throughout the proof CC^{\prime} is a constant only depending on (q,d,t,β,F)(q,\mathbf{d},\mathbf{t},\bm{\beta},F) that may change from line to line. Combining Theorem 2 with the assumed bounds on the depth LL and the network sparsity s,s, it follows for n3,n\geq 3,

where we used ε=1/2\varepsilon=1/2 for the lower bound and ε=1\varepsilon=1 for the upper bound. Taking C=8C,C=8C^{\prime}, we find that 18Δn(f^n,f0)R(f^,f0)\tfrac{1}{8}\Delta_{n}(\widehat{f}_{n},f_{0})\leq R(\widehat{f},f_{0}) whenever Δn(f^n,f0)CϕnLlog2n.\Delta_{n}(\widehat{f}_{n},f_{0})\geq C\phi_{n}L\log^{2}n. This proves the lower bound in (9).

To derive the upper bounds in (8) and (9) we need to bound the approximation error. To do this, we rewrite the regression function f0f_{0} as in (21), that is, f0=hqh0f_{0}=h_{q}\circ\ldots h_{0} with hi=(hij)j,h_{i}=(h_{ij})_{j}^{\top}, hijh_{ij} defined on ti,^{t_{i}}, and for any i<q,i<q, hijh_{ij} mapping to ..

We apply Theorem 5 to each function hijh_{ij} separately. Take m=log2nm=\lceil\log_{2}n\rceil and let Li:=8+(log2n+5)(1+log2(tiβi)).L^{\prime}_{i}:=8+(\lceil\log_{2}n\rceil+5)(1+\lceil\log_{2}(t_{i}\vee\beta_{i})\rceil). This means there exists a network h~ijF(Li,(ti,6(ti+βi)N,,6(ti+βi)N,1),si)\widetilde{h}_{ij}\in\mathcal{F}(L^{\prime}_{i},(t_{i},6(t_{i}+\lceil\beta_{i}\rceil)N,\ldots,6(t_{i}+\lceil\beta_{i}\rceil)N,1),s_{i}) with si141(ti+βi+1)3+tiN(log2n+6),s_{i}\leq 141(t_{i}+\beta_{i}+1)^{3+t_{i}}N(\lceil\log_{2}n\rceil+6), such that

where QiQ_{i} is any upper bound of the Hölder norms of hij.h_{ij}. If i<q,i<q, then we apply to the output the two additional layers 1(1x)+.1-(1-x)_{+}. This requires four additional parameters. Call the resulting network hijF(Li+2,(ti,6(ti+βi)N,,6(ti+βi)N,1),si+4)h_{ij}^{*}\in\mathcal{F}(L^{\prime}_{i}+2,(t_{i},6(t_{i}+\lceil\beta_{i}\rceil)N,\ldots,6(t_{i}+\lceil\beta_{i}\rceil)N,1),s_{i}+4) and observe that σ(hij)=(h~ij(x)0)1.\sigma(h_{ij}^{*})=(\widetilde{h}_{ij}(x)\vee 0)\wedge 1. Since hij(x),h_{ij}(\mathbf{x})\in, we must have

If the networks hijh_{ij}^{*} are computed in parallel, hi=(hij)j=1,,di+1h_{i}^{*}=(h_{ij}^{*})_{j=1,\ldots,d_{i+1}} lies in the class

with ri:=maxidi+1(ti+βi).r_{i}:=\max_{i}d_{i+1}(t_{i}+\lceil\beta_{i}\rceil). Finally, we construct the composite network f=h~q1σ(hq1)σ(h0)f^{*}=\widetilde{h}_{q1}\circ\sigma(h_{q-1}^{*})\circ\ldots\circ\sigma(h_{0}^{*}) which by the composition rule in Section 7.1 can be realized in the class

with E:=3(q1)+i=0qLi.E:=3(q-1)+\sum_{i=0}^{q}L^{\prime}_{i}. Observe that there is an AnA_{n} that is bounded in nn such that E=An+log2n(i=0qlog2(tiβi)+1).E=A_{n}+\log_{2}n(\sum_{i=0}^{q}\lceil\log_{2}(t_{i}\vee\beta_{i})\rceil+1). Using that x<x+1,\lceil x\rceil<x+1, Ei=0q(log2(4)+log2(tiβi))log2nLE\leq\sum_{i=0}^{q}(\log_{2}(4)+\log_{2}(t_{i}\vee\beta_{i}))\log_{2}n\leq L for all sufficiently large n.n. By (18) and for sufficiently large n,n, the space (25) can be embedded into F(L,p,s)\mathcal{F}(L,\mathbf{p},s) with L,p,sL,\mathbf{p},s satisfying the assumptions of the theorem by choosing N=cmaxi=0,,qnti2βi+tiN=\lceil c\max_{i=0,\ldots,q}n^{\frac{t_{i}}{2\beta_{i}^{*}+t_{i}}}\rceil for a sufficiently small constant c>0c>0 only depending on q,d,t,β.q,\mathbf{d},\mathbf{t},\bm{\beta}. Combining Lemma 3 with (23) and (24)

For the approximation error in (22) we need to find a network function that is bounded in sup-norm by F.F. By the previous inequality there exists a sequence (f~n)n(\widetilde{f}_{n})_{n} such that for all sufficiently large n,n, f~nF(L,p,s)\widetilde{f}_{n}\in\mathcal{F}(L,\mathbf{p},s) and f~nf022Cmaxi=0,,qc2βi/tin(2βi)/(2βi+ti).\|\widetilde{f}_{n}-f_{0}\|_{\infty}^{2}\leq 2C\max_{i=0,\ldots,q}c^{-2\beta_{i}^{*}/t_{i}}n^{-(2\beta_{i}^{*})/(2\beta_{i}^{*}+t_{i})}. Define fn=f~n(f0/f~n1).f^{*}_{n}=\widetilde{f}_{n}(\|f_{0}\|_{\infty}/\|\widetilde{f}_{n}\|_{\infty}\newline \wedge 1). Then, fnf0=gqKF,\|f^{*}_{n}\|_{\infty}\leq\|f_{0}\|_{\infty}=\|g_{q}\|_{\infty}\leq K\leq F, where the last inequality follows from assumption (i).(i). Moreover, fnF(L,p,s,F).f^{*}_{n}\in\mathcal{F}(L,\mathbf{p},s,F). Writing fnf0=(fnf~n)+(f~nf0),f_{n}^{*}-f_{0}=(f_{n}^{*}-\widetilde{f}_{n})+(\widetilde{f}_{n}-f_{0}), we obtain fnf02f~nf0.\|f_{n}^{*}-f_{0}\|_{\infty}\leq 2\|\widetilde{f}_{n}-f_{0}\|_{\infty}. This shows that (26) also holds (with constants multiplied by 88) if the infimum is taken over the smaller space F(L,p,s,F).\mathcal{F}(L,\mathbf{p},s,F). Together with (22) the upper bounds in (8) and (9) follow for any constant CC. This completes the proof. ∎

2 Proof of Theorem 2

Several oracle inequalities for the least-squares estimator are know, cf. . The common assumption of bounded response variables is, however, violated in the nonparametric regression model with Gaussian measurement noise. Additionally we provide also a lower bound of the risk and give a proof that can be easily generalized to arbitrary noise distributions. Let N(δ,F,)\mathcal{N}(\delta,\mathcal{F},\|\cdot\|_{\infty}) be the covering number, that is, the minimal number of \|\cdot\|_{\infty}-balls with radius δ\delta that covers F\mathcal{F} (the centers do not need to be in F\mathcal{F}).

Consider the dd-variate nonparametric regression model (1) with unknown regression function f0.f_{0}. Let f^\widehat{f} be any estimator taking values in F.\mathcal{F}. Define

and assume {f0}F{f:d[F,F]}\{f_{0}\}\cup\mathcal{F}\subset\{f:^{d}\rightarrow[-F,F]\} for some F1.F\geq 1. If Nn:=N(δ,F,)3,\mathcal{N}_{n}:=\mathcal{N}(\delta,\mathcal{F},\|\cdot\|_{\infty})\geq 3, then,

The proof of the lemma can be found in the supplement. Next, we prove a covering entropy bound. Recall the definition of the network function class F(L,p,s,F)\mathcal{F}(L,\mathbf{p},s,F) in (4).

For a proof see the supplement. A related result is Theorem 14.5 in .

The assertion follows from Lemma 5 with δ=1/n,\delta=1/n, Lemma 4 and Remark 1 since F1.F\geq 1.

3 Proof of Theorem 3

Throughout this proof, \|\cdot\|_{2}=\|\cdot\|_{L^{2}^{d}}. By assumption there exist positive γΓ\gamma\leq\Gamma such that the Lebesgue density of X\mathbf{X} is lower bounded by γ\gamma and upper bounded by Γ\Gamma on d.^{d}. For such design, R(f^n,f0)γf^nf022.R(\widehat{f}_{n},f_{0})\geq\gamma\|\widehat{f}_{n}-f_{0}\|_{2}^{2}. Denote by PfP_{f} the law of the data in the nonparametric regression model (1). For the Kullback-Leibler divergence we have KL(Pf,Pg)=nE[(f(X1)g(X1))2]Γnfg22.\operatorname{KL}(P_{f},P_{g})=nE[(f(\mathbf{X}_{1})-g(\mathbf{X}_{1}))^{2}]\leq\Gamma n\|f-g\|_{2}^{2}. Theorem 2.7 in states that if for some M1M\geq 1 and κ>0,\kappa>0, f(0),,f(M)G(q,d,t,β,K)f_{(0)},\ldots,f_{(M)}\in\mathcal{G}(q,\mathbf{d},\mathbf{t},\bm{\beta},K) are such that

f(j)f(k)2κϕn\|f_{(j)}-f_{(k)}\|_{2}\geq\kappa\sqrt{\phi_{n}} for all 0j<kM0\leq j<k\leq M

nj=1Mf(j)f(0)22Mlog(M)/(9Γ),n\sum_{j=1}^{M}\|f_{(j)}-f_{(0)}\|_{2}^{2}\leq M\log(M)/(9\Gamma),

then there exists a positive constant c=c(κ,γ),c=c(\kappa,\gamma), such that

Hence ψuCtβ(t,(β)tt).\psi_{\mathbf{u}}\in\mathcal{C}_{t^{*}}^{\beta^{*}}(^{t^{*}},(\beta^{*})^{t^{*}}t^{*}). For a vector w=(wu)uUn{0,1}Un,\mathbf{w}=(w_{\mathbf{u}})_{\mathbf{u}\in\mathcal{U}_{n}}\in\{0,1\}^{|\mathcal{U}_{n}|}, define

By construction, ψu\psi_{\mathbf{u}} and ψu,\psi_{\mathbf{u}^{\prime}}, u,uUn,\mathbf{u},\mathbf{u}^{\prime}\in\mathcal{U}_{n}, uu\mathbf{u}\neq\mathbf{u}^{\prime} have disjoint support. As a consequence ϕwCtβ(t,2(β)tt).\phi_{\mathbf{w}}\in\mathcal{C}_{t^{*}}^{\beta^{*}}(^{t^{*}},2(\beta^{*})^{t^{*}}t^{*}).

and fwG(q,d,t,β,K)f_{\mathbf{w}}\in\mathcal{G}(q,\mathbf{d},\mathbf{t},\bm{\beta},K) provided KK is taken sufficiently large.

For all u,\mathbf{u}, ψu22=hn2β+tKB22t.\|\psi_{\mathbf{u}}\|_{2}^{2}=h_{n}^{2\beta^{**}+t^{*}}\|K^{B}\|_{2}^{2t^{*}}. If Ham(w,w)=uUn1(wuwu)\operatorname{Ham}(\mathbf{w},\mathbf{w}^{\prime})=\sum_{\mathbf{u}\in\mathcal{U}_{n}}\mathbf{1}(w_{\mathbf{u}}\neq w_{\mathbf{u}^{\prime}}) denotes the Hamming distance, we find

By the Varshamov - Gilbert bound (, Lemma 2.9) and because of mnt=Un,m_{n}^{t^{*}}=|\mathcal{U}_{n}|, we conclude that there exists a subset W{0,1}mnt\mathcal{W}\subset\{0,1\}^{m_{n}^{t^{*}}} of cardinality W2mnt/8,|\mathcal{W}|\geq 2^{m_{n}^{t^{*}}/8}, such that Ham(w,w)mnt/8\operatorname{Ham}(\mathbf{w},\mathbf{w}^{\prime})\geq m_{n}^{t^{*}}/8 for all w,wW,\mathbf{w},\mathbf{w}^{\prime}\in\mathcal{W}, ww.\mathbf{w}\neq\mathbf{w}^{\prime}. This implies that for κ=KB2t/(8ρβ),\kappa=\|K^{B}\|_{2}^{t^{*}}/(\sqrt{8}\rho^{\beta^{**}}),

Using the definition of hnh_{n} and ρ,\rho,

This shows that the functions fwf_{\mathbf{w}} with wW\mathbf{w}\in\mathcal{W} satisfy (i)(i) and (ii).(ii). The assertion follows. ∎

4 Proof of Lemma 1

We will choose c21.c_{2}\leq 1. Since f0K,\|f_{0}\|_{\infty}\leq K, it is therefore enough to consider the infimum over F(L,p,s,F)\mathcal{F}(L,\mathbf{p},s,F) with F=K+1.F=K+1. Let f~n\widetilde{f}_{n} be an empirical risk minimizer. Recall that Δn(f~n,f0)=0.\Delta_{n}(\widetilde{f}_{n},f_{0})=0. Because of the minimax lower bound in Theorem 3, there exists a constant c3c_{3} such that c3n2β/(2β+d)supf0C1β(,K)R(f~n,f0)c_{3}n^{-2\beta/(2\beta+d)}\leq\sup_{f_{0}\in\mathcal{C}_{1}^{\beta}(,K)}R(\widetilde{f}_{n},f_{0}) for all sufficiently large n.n. Because of p0=dp_{0}=d and pL+1=1,p_{L+1}=1, Theorem 2 yields

for some constant C.C. Given ε,\varepsilon, set nε:=(8ε/c3)(2β+d)/β.n_{\varepsilon}:=\lfloor(\sqrt{8}\varepsilon/\sqrt{c_{3}})^{-(2\beta+d)/\beta}\rfloor. Observe that for εc3/8,\varepsilon\leq\sqrt{c_{3}/8}, nε12(8ε/c3)(2β+d)/βn_{\varepsilon}^{-1}\leq 2(\sqrt{8}\varepsilon/\sqrt{c_{3}})^{(2\beta+d)/\beta} and 8ε2/c3nε2β/(2β+d).8\varepsilon^{2}/c_{3}\leq n_{\varepsilon}^{-2\beta/(2\beta+d)}. For sufficiently small c2>0c_{2}>0 and all εc2,\varepsilon\leq c_{2}, we can insert nεn_{\varepsilon} in the previous inequality and find

for constants C1,C2C_{1},C_{2} depending on K,β,K,\beta, and d.d. The result follows using the condition sc1εd/β/(Llog(1/ε))s\leq c_{1}\varepsilon^{-d/\beta}/(L\log(1/\varepsilon)) and choosing c1c_{1} small enough. ∎

5 Proofs for Section 5

For a real number u,u, denote by {u}\{u\} the fractional part of u.u.

We need to study the cases μ00\mu_{0}\neq 0 and μ0=0\mu_{0}=0 separately. If μ00,\mu_{0}\neq 0, define g(u)=r1ur1[0,1/2](u)+r1(1u)r1(1/2,1](u).g(u)=r^{-1}u^{r}\mathbf{1}_{[0,1/2]}(u)+r^{-1}(1-u)^{r}\mathbf{1}_{(1/2,1]}(u). Notice that gg is Lipschitz with Lipschitz constant one. Let hj,α(u)=K2jα1g({2jqνu})h_{j,\alpha}(u)=K2^{-j\alpha-1}g(\{2^{j-q-\nu}u\}) with qq and ν\nu as defined in the statement of the lemma. For a TT-periodic function us(u)u\mapsto s(u) the α\alpha-Hölder semi-norm for α1\alpha\leq 1 can be shown to be supuv,uvTs(u)s(v)/uvα.\sup_{u\neq v,|u-v|\leq T}|s(u)-s(v)|/|u-v|^{\alpha}. Since gg is 11-Lipschitz, we have for u,vu,v with uv2q+νj,|u-v|\leq 2^{q+\nu-j},

Since hj,αK/2,\|h_{j,\alpha}\|_{\infty}\leq K/2, hj,αC1α([0,d],K).h_{j,\alpha}\in\mathcal{C}_{1}^{\alpha}([0,d],K). Let fj,α(x)=hj,α(x1++xd).f_{j,\alpha}(\mathbf{x})=h_{j,\alpha}(x_{1}+\ldots+x_{d}). Recall that the support of ψ\psi is contained in [0,2q][0,2^{q}] and 2ν2d.2^{\nu}\geq 2d. By definition of the wavelet coefficients, Equation (27), the definitions of hj,α,h_{j,\alpha}, and using μr=xrψ(x)dx,\mu_{r}=\int x^{r}\psi(x)dx, we find for p1,,pd{0,1,,2jq21},p_{1},\ldots,p_{d}\in\{0,1,\ldots,2^{j-q-2}-1\},

where we used for the last identity that by definition of r,r, μ1==μr1=0.\mu_{1}=\ldots=\mu_{r-1}=0.

In the case that μ0=0,\mu_{0}=0, we can take g(u)=(dr)1udr1[0,1/2](u)+(dr)1(1u)dr1(1/2,1](u).g(u)=(dr)^{-1}u^{dr}\mathbf{1}_{[0,1/2]}(u)+(dr)^{-1}(1-u)^{dr}\mathbf{1}_{(1/2,1]}(u). Following the same arguments as before and using the multinomial theorem, we obtain

Let c(ψ,d)c(\psi,d) be as in Lemma 2. Choose an integer jj^{*} such that

This means that 2j12(c(ψ,d)2K2n)1/(2α+d).2^{j^{*}}\geq\tfrac{1}{2}(c(\psi,d)^{2}K^{2}n)^{1/(2\alpha+d)}. By Lemma 2, there exists a function fj,αf_{j^{*},\alpha} of the form h(x1++xd),h(x_{1}+\ldots+x_{d}), hC1α([0,d],K),h\in\mathcal{C}_{1}^{\alpha}([0,d],K), such that with (16),

Acknowledgments

The author is grateful to all the insights, comments and suggestions that arose from discussions on the topic with other researchers. In particular, he wants to thank the AE, two referees, Thijs Bos, Hyunwoong Chang, Konstantin Eckle, Kenji Fukumizu, Maximilian Graf, Roy Han, Masaaki Imaizumi, Michael Kohler, Matthias Löffler, Patrick Martin, Hrushikesh Mhaskar, Gerrit Oomens, Tomaso Poggio, Richard Samworth, Taiji Suzuki, Dmitry Yarotsky and Harry van Zanten.

The author is very grateful to the discussants for sharing their viewpoints on the article. The discussant contributions highlight the gaps in the theoretical understanding and outline many possible directions for future research in this area. The rejoinder is structured according to topics. We refer to [GMMM], [K], [KL], and [S] for the discussant contributions by Ghorbani et al., Kutyniok, Kohler & Langer, and Shamir, respectively.

Overparametrization and implicit regularization

One of the general claims about deep learning is that even for extreme overfitting the method still generalizes well. There are numerous experiments showing that running the training error to zero and therefore interpolating all data points results in state of the art generalization performance. The rationale behind this is that among all solutions interpolating the data points - of which most result in bad generalization behavior - stochastic gradient descent (SGD) picks a minimum norm interpolant. This is also known as implicit regularization. While this is well known for stochastic gradient descent applied to linear regression, for deep networks some progress has been made recently in finding the norm minimized by (S)GD, see .

It is now reasonable to wonder whether the notion of network sparsity could be removed in the article if implicit regularization would have been taken into account. [GMMM] write that ”model complexity is not controlled by an explicit penalty or procedure, but by the dynamics of stochastic gradient descent (SGD) itself.” [S] mentions implicit regularization to show that statistical guarantees should involve specific learning methods.

We conjecture that for additive error models, such as the nonparametric regression model considered in the article, implicit regularization in the overfitted regime is insufficient to achieve even consistency. To support our conjecture, we provide the following two step argument. In the first step, we argue that for one-dimensional input and shallow networks with fixed parameters in the first layer, SGD will converge to a variant of the natural cubic spline interpolant. In the second step we show that this reconstruction leads to an inconsistent estimator if additive noise is present.

A shallow ReLU network with one input and one output node can be written as xj=1maj(bjxcj)+.x\mapsto\sum_{j=1}^{m}a_{j}(b_{j}x-c_{j})_{+}. We now study an even more simplified setup where bjb_{j} is always one. For small δ>0,\delta>0, (xcj)+cjcj+δ(xu)+du/δ.(x-c_{j})_{+}\approx\int_{c_{j}}^{c_{j}+\delta}(x-u)_{+}du/\delta. This motivates to study smoothed shallow ReLU networks of the form

with parameter vector a=(a1,,am)\mathbf{a}=(a_{1},\ldots,a_{m}) and fixed t0<t1<<tm.t_{0}<t_{1}<\ldots<t_{m}. For convenience, we have rescaled the parameters aja_{j} so that the normalization factor becomes 1/tjtj1.1/\sqrt{t_{j}-t_{j-1}}. We consider the overparametrized regime mnm\geq n assuming that for any i,i, there lies at least one tjt_{j} in the interval [X(i1),X(i))[X_{(i-1)},X_{(i)}) with X(i)X_{(i)} the ii-th order statistic of the sample X1,,XnX_{1},\ldots,X_{n} and X(0)=.X_{(0)}=-\infty. Under overparametrization, this is a rather weak assumption and ensures existence of a shallow ReLU network faf_{\mathbf{a}}^{*} perfectly interpolating the data in the sense that fa(Xi)=Yif_{\mathbf{a}}^{*}(X_{i})=Y_{i} for all i.i.

For initialization at zero and properly chosen learning rate, SGD with respect to the least squares loss converges to the minimum norm interpolant with parameter vector

(this result is due to for overdetermined linear systems but can be extended to the underdetermined case, see also the generalizations in ). Because of fa(x)=aj/tjtj1f_{\mathbf{a}}^{\prime\prime}(x)=a_{j}/\sqrt{t_{j}-t_{j-1}} for all x(tj1,tj),x\in(t_{j-1},t_{j}), we find a2=faL2[t0,tm].\|\mathbf{a}\|_{2}=\|f_{\mathbf{a}}^{\prime\prime}\|_{L^{2}[t_{0},t_{m}]}. It is known that the natural cubic spline interpolant LL is the interpolant with the smallest L2L^{2}-norm on the second derivative. Moreover, we have that fL22=LL22+LfL22\|f^{\prime\prime}\|_{L^{2}}^{2}=\|L^{\prime\prime}\|_{L^{2}}^{2}+\|L^{\prime\prime}-f^{\prime\prime}\|_{L^{2}}^{2} for all twice differentiable interpolating functions ff, see Equation (2.9) in . Since faf_{\mathbf{a}^{*}} and LL are both interpolants, this implies that the SGD limit faf_{\mathbf{a}^{*}} will be close to the natural cubic spline interpolant.

In the nonparametric regression model with additive errors, the distance between the true function values and the response variables YiY_{i} is of the order of the noise level (which is assumed to be fixed). The natural cubic spline interpolates the YiY_{i}’s. If in a neighborhood, the YiY_{i}’s lie all on one side of the regression function, the average distance between the natural cubic spline interpolant and the true regression function will be lower bounded by a constant. Since this happens on a subset with Lebesgue measure bounded from below, the natural cubic spline interpolant is inconsistent for estimating the regression function. As the SGD limit approximates the natural cubic spline interpolant, this indicates that stochastic gradient descent should lead to inconsistent estimators.

We believe that this also holds true for deep networks. In this case, it is expected that SGD still converges to a spline interpolant but not necessarily to the natural cubic spline interpolant, see also for a related argument.

While it has been observed that there are nonparametric estimators that can interpolate and also achieve fast convergence rates in the nonparametric regression model (), the argument above indicates that implicit regularization in the overfitted regime will not do that. To obtain rate optimal estimators, more regularization has to be imposed forcing the network to do smoothing.

Network sparsity

The article identifies sparsity of the network weights as a complexity measure to achieve optimal convergence rates under a hierarchical composition assumption. As sparsity is a non-standard assumption, there are several comments on it in the reports. [GMMM] show that the empirical distribution of the weights in the first fully connected layer of the VGG-19 network is nearly Gaussian. [KL] mention a recent result proving optimal estimation rates for very deep networks with fully connected layers.

After the original version of this article was drafted, a large body of applied work emerged dealing either with compression through sparsifying dense networks or proposing methods that directly train a sparse neural network. Below we briefly summarize some of these approaches.

One method to achieve sparsity in neural networks is by pruning a fully connected network after training. A simple approach would be to replace small network weights by zero, but more sophisticated approaches based on the second derivative have been proposed as well, see . proposes an iterative pruning procedure, see also . These approaches allow to reduce the number of parameters in fully connected layers by about 90%90\% without loss of efficiency.

Although Theorem 1 is formulated in terms of network sparsity, the proof explicitly constructs a network topology, that is, the graph structure defined by the non-zero connections between successive layers, for which the minimax estimation rate is attained (up to log-factors). Instead of searching over all ss-sparse networks, it is therefore in principle possible to start with this network topology and only learn the non-zero weights. By fixing one sparse network topology, a lot of the flexibility of networks to adapt to the underlying structure in the data might be lost. An intermediate constraint would be to impose an individual sparsity parameter for each weight matrix or to bound the indegree and outdegree for each individual unit in the network. In the applied literature, choosing a sparse network topology beforehand has been proposed recently in . The latter article makes an interesting connection between sparsely connected neural networks and decision trees. Related to an initial choice of a sparse network topology is the evolutionary algorithm inspired by biological neural networks proposed in . It starts with sparse weight matrices. In every iteration, the smallest weights are removed and new random connections are added so that the network topology changes but the overall network sparsity is kept constant. The method proposed in is also inspired by the sparsity observed in biological networks. It starts with a sparse network topology and increases the sparsity by only keeping the units in each hidden layer that channel most of the signal to the next layer.

The recent work on weight agnostic neural networks takes this one step further. No training is done and the weights are fixed to the initialized values at all times. Only the network topology is learned by an iterative procedure. In each step of the iteration, we have a set of candidate models. For each of those models a score is computed. ”Around” the models with the highest scores a new set of randomly generated candidate models is generated.

Theorem 1 in [KL] considers neural networks with fixed width and depth increasing polynomially in the sample size. It is shown that for such extremely deep networks, the empirical risk minimizer over fully-connected layers achieves the optimal estimation rate and no sparsity is needed. Such architectures are, however, in many aspects quite different compared to the neural networks considered in practice. In , it has been observed that for such extremely deep networks, one needs discontinuous weight assignments to achieve the best possible approximation rate. This is a strange phenomenon which could hint at some issues with the stability during learning of the network weights.

Classification and nonparametric regression

While the article deals with data from the nonparametric regression model, the overwhelming part of the literature on deep learning is on classification. Nonparametric regression and estimation of the conditional class probabilities in classification is similar, if a fraction of the data is mislabeled which prevents the conditional class probabilities to be close to zero or one. For the commonly considered classification tasks in deep learning, this is, however, not the case as most of the data are correctly labeled. As the randomness due to mislabeling is negligible in those cases, the only remaining randomness is in the distribution of the design/inputs and reconstruction becomes rather an interpolation than a denoising problem. If the different classes are also well separated from each other, much faster convergence rates can be achieved. This explains why the sample complexity in the nonparametric regression model is much higher than what is observed in deep learning for object recognition tasks, see also Report [S].

Concerning the statistical properties there are some differences. For image classification problems, deep learning is for instance not robust to Gaussian perturbations, see . In the nonparametric regression model, Gaussian perturbations just increase the noise level. Since the noise level appears in the estimation risk bounds through the constants, the estimation rates for the class of estimators considered in the article will not change under additive noise perturbations.

We would like to stress again that the structure of the data is essential for the behavior of deep learning and the properties of the reconstructions. One of the challenges for future research will be to study estimation in models beyond nonparametric regression.

Algorithms

[GMMM] and [S] question whether one can disentangle the algorithm from the statistical analysis. We would like to stress that Theorem 1 is not about one fixed estimator. It provides bounds for any estimator which, given data, returns a sparsely connected neural network. The method/estimator determines the term Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) defined in Equation (5) and Theorem 1 shows that Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) tightly controls the estimation risk from above and below. This is different than the case of data interpolation and training error zero, where Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) is not sufficient anymore to fully characterize the statistical properties, see also Report [S] and .

We agree that the difficulty is shifted to a precise estimate of the term Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) and we hope to study this term in more detail in future work. This term might heavily depend on the learning rate, the initialization and the energy landscape. Regarding a question in [K], the expectation in the definition of Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) (Equation (5) in the article) can be taken over all the randomness, including additional randomization in the algorithm.

While it would be desirable to have precise theoretical bounds for the performance of the most popular deep learning methods such as Adam, we believe that some amount of idealization and simplification is unavoidable. In statistical theory, this seems to be widely accepted. For instance, most of the theory on the LASSO deals with regularization parameters derived from large deviations bounds although the standard software implementations choose the regularization parameter by ten-fold cross validation.

High-dimensional input

[GMMM] mentions that for the current proof strategy and the case of additive models, the dependence of the dimension on the constants is dd.d^{d}. As mentioned in the article, the results focus on the convergence rates and no attempt has been made to minimize the constants appearing in the proofs. In fact by a variation of the original argument, the dependence on the dimension for additive models f(x)=i=1dfi(xi)f(\mathbf{x})=\sum_{i=1}^{d}f_{i}(x_{i}) can be shown to be linear. To see this, we can build for any given N1,N\geq 1, dd separate networks with sNlogNs\asymp N\log N parameters, computing the functions f1(x1),,fd(xd)f_{1}(x_{1}),\ldots,f_{d}(x_{d}) up to an approximation error of the order O(Nβ).O(N^{-\beta}). Using the parallelization rule mentioned on p.21 of the article, one can then combine the individual networks into a large neural network computing the sum i=1nfi(xi)\sum_{i=1}^{n}f_{i}(x_{i}) up to an approximation error of order O(dNβ)O(dN^{-\beta}) using sdNlogNs\asymp dN\log N many network parameters. It then follows from Theorem 2 that the rate is upper bounded by dn2β/(2β+1)log3ndn^{-2\beta/(2\beta+1)}\log^{3}n if Δn(f^n,f0)\Delta_{n}(\widehat{f}_{n},f_{0}) is sufficiently small and dd is bounded by a power of the sample size.

As another result on high-dimensional input, [S] mentions a theorem proving that basis expansions have difficulties to approximate functions generated by a single neuron. Either huge coefficients are needed or the number of basis functions has to be exponential in the input dimension.

Since the input dimension dd in deep learning applications is typically extremely large, a possible future direction would be to analyze neural networks with high-dimensional d=dnd=d_{n}\uparrow\infty and comparing the rates to other nonparametric procedures.

Function classes

With respect to the considered function class, [K] emphasizes that the function classes should be detached from the method. On the contrary, [S] favors an alternative approach where the underlying function class consists itself of neural network functions. We believe that both approaches have advantages and disadvantages.

The imposed class of composition functions in the article appears of course naturally given the composition structure of deep networks. Compositions are fundamental operations and as mentioned in the article, many widely studied function classes in nonparametric statistics such as (generalized) additive models occur as special cases of the imposed composition constraint.

For a recent result in the statistical literature with function class consisting of neural network functions, we refer to . One possibility for future research would be to determine the maxisets for neural networks, that is, the largest possible function class for which a prespecified estimation rate can be obtained, see . The main advantage of generic function spaces such as Hölder classes is that we can compare the estimation rates achieved by different methods and therefore learn something about the strength and weaknesses of these methods. The article shows for instance that wavelet methods have a slower rate of convergence for generalized additive models than sparsely connected deep ReLU networks.

To obtain fast estimation rates, an alternative is to impose structure on the design, see .

Choice of the activation function

On p. 12 in the article we highlight several specific properties of the ReLU activation function such as the possibility to easily learn skip connections. [KL] mention that results for ReLU networks automatically transfer to other activation functions. The argument, however, requires that the network parameters will become large. In the meantime, we better and better understand how SGD leads to norm control on the parameters. To model this, we think that it is important to control the magnitude of the weights in the network classes. In the article, the network parameters are bounded in absolute value by one. This is a convenient choice, but as our understanding of the norm control induced by SGD improves, more realistic constraints are imaginable. It is well-known that training does not move the parameters far from the initialized values. To analyze the effect of different initialization strategies one possibility would be to study network classes generated by all parameters in a neighborhood of a (random) initializer.

Real data

[GMMM] report the results of a simulation study which seemingly contradict the theory in the article. They study the noisefree case and up to three hidden layers showing that a certain smooth function cannot be learned. We would like to refer to the simulation study in , which finds that for regression problems, the performance of deep neural networks is not far off from the theoretical bounds. This article also examines the finite sample performance of the multiplication network in Lemma A.2 which forms an essential part in the proof of Theorem 1. To a certain extent, even such specific constructions can be picked up by deep learning. This, however, only works for a careful initialization. It might be necessary to reinitialize the procedure if the algorithm gets stuck in a local minimum with large training error.

Acknowledgments

The author would like to thank Misha Belkin and Dirk Lorentz for fruitful discussions on overfitting and SGD.

Appendix A Network approximation of polynomials

In this section we describe the construction of deep networks that approximate monomials of the input.

In a first step, we construct a network, with all network parameters bounded by one, which approximately computes xyxy given input xx and y.y. Let Tk:[0,222k][0,22k],T^{k}:[0,2^{2-2k}]\rightarrow[0,2^{-2k}],

The next result shows that k=1mRk(x)\sum_{k=1}^{m}R^{k}(x) approximates x(1x)x(1-x) exponentially fast in mm and that in particular x(1x)=k=1Rk(x)x(1-x)=\sum_{k=1}^{\infty}R^{k}(x) in L.L^{\infty}. This lemma can be viewed as a slightly sharper variation of Lemma 2.4 in and Proposition 2 in . In contrast to the existing results, we can use it to build networks with parameters bounded by one. It also provides an explicit bound on the approximation error.

this shows the claim for Rk+1R^{k+1} and completes the induction.

Let g(x)=x(1x)g(x)=x(1-x) as in the previous proof. To construct a network which returns approximately xyxy given input xx and yy, we use the polarization type identity

For any positive integer m,m, there exists a network MultmF(m+4,(2,6,6,,6,1)),\operatorname{Mult}_{m}\in\mathcal{F}(m+4,(2,6,6,\ldots,6,1)), such that Multm(x,y),\operatorname{Mult}_{m}(x,y)\in,

and Multm(0,y)=Multm(x,0)=0.\operatorname{Mult}_{m}(0,y)=\operatorname{Mult}_{m}(x,0)=0.

Write Tk(x)=(x/2)+(x212k)+=T+(x)Tk(x)T_{k}(x)=(x/2)_{+}-(x-2^{1-2k})_{+}=T_{+}(x)-T_{-}^{k}(x) with T+(x)=(x/2)+T_{+}(x)=(x/2)_{+} and Tk(x)=(x212k)+T_{-}^{k}(x)=(x-2^{1-2k})_{+} and let h:[0,)h:\rightarrow[0,\infty) be a non-negative function. In a first step we show that there is a network NmN_{m} with mm hidden layers and width vector (3,3,,3,1)(3,3,\ldots,3,1) that computes the function

for all u.u\in. The proof is given in Figure 2. Notice that all parameters in this networks are bounded by one. In a next step, we show that there is a network with m+3m+3 hidden layers that computes the function

Given input (x,y),(x,y), this network computes in the first layer

On the first three and the last three components, we apply the network Nm.N_{m}. This gives a network with m+1m+1 hidden layers and width vector (2,6,,6,2)(2,6,\ldots,6,2) that computes

Apply to the output the two hidden layer network (u,v)(1(1(uv))+)+=(uv)+1.(u,v)\mapsto(1-(1-(u-v))_{+})_{+}=(u-v)_{+}\wedge 1. The combined network Multm(x,y)\operatorname{Mult}_{m}(x,y) has thus m+4m+4 hidden layers and computes

This shows that the output is always in .. By (28) and Lemma A.1, Multm(x,y)xy2m.|\operatorname{Mult}_{m}(x,y)-xy|\leq 2^{-m}.

By elementary computations, one can check that R1((1u)/2)=R1((1+u)/2)R^{1}((1-u)/2)=R^{1}((1+u)/2) and R2((1+u)/2)=R2(u/2)R^{2}((1+u)/2)=R^{2}(u/2) for all 0u1.0\leq u\leq 1. Therefore, Rk((1u)/2)=Rk((1+u)/2)R^{k}((1-u)/2)=R^{k}((1+u)/2) for all k1k\geq 1 and Rk((1+u)/2)=Rk(u/2)R^{k}((1+u)/2)=R^{k}(u/2) for all k2.k\geq 2. This shows that the output (29) is zero for all inputs (0,y)(0,y) and (x,0).(x,0).

For any positive integer m,m, there exists a network

such that Multmr\operatorname{Mult}_{m}^{r}\in and

Moreover, Multmr(x)=0\operatorname{Mult}_{m}^{r}(\mathbf{x})=0 if one of the components of x\mathbf{x} is zero.

Let q:=log2(r)q:=\lceil\log_{2}(r)\rceil. Let us now describe the construction of the Multmr\operatorname{Mult}_{m}^{r} network. In the first hidden layer the network computes

If a,b,c,d,a,b,c,d\in, then by Lemma A.2 and triangle inequality, Multm(a,b)cd2m+ac+bd.|\operatorname{Mult}_{m}(a,b)-cd|\leq 2^{-m}+|a-c|+|b-d|. By induction on the number of iterated multiplications q,q, we therefore find that Multmr(x)i=1rxi3q12mr22m|\operatorname{Mult}_{m}^{r}(\mathbf{x})-\prod_{i=1}^{r}x_{i}|\leq 3^{q-1}2^{-m}\leq r^{2}2^{-m} since log2(3)1.58<2.\log_{2}(3)\approx 1.58<2.

From Lemma A.2 and the construction above, it follows that Multmr(x)=0\operatorname{Mult}_{m}^{r}(\mathbf{x})=0 if one of the components of x\mathbf{x} is zero. ∎

The number of monomials with degree α<γ|\bm{\alpha}|<\gamma is denoted by Cr,γ.C_{r,\gamma}. Obviously, Cr,γ(γ+1)rC_{r,\gamma}\leq(\gamma+1)^{r} since each αi\alpha_{i} has to take values in {0,1,,γ}.\{0,1,\ldots,\lfloor\gamma\rfloor\}.

For γ>0\gamma>0 and any positive integer m,m, there exists a network

such that Monm,γrCr,γ\operatorname{Mon}_{m,\gamma}^{r}\in^{C_{r,\gamma}} and

For α1,|\bm{\alpha}|\leq 1, the monomials are linear or constant functions and there exists a shallow network in the class F(1,(1,1,1))\mathcal{F}(1,(1,1,1)) with output exactly xα.\mathbf{x}^{\bm{\alpha}}.

By taking the multiplicity into account in (30), Lemma A.3 can be extended in a straightforward way to monomials. For α2,|\bm{\alpha}|\geq 2, this shows the existence of a network in the class F((m+5)log2α,(r,6α,,6α,1))\mathcal{F}((m+5)\lceil\log_{2}|\bm{\alpha}|\rceil,(r,6|\bm{\alpha}|,\ldots,6|\bm{\alpha}|,1)) taking values in $andapproximatingand approximating\mathbf{x}^{\bm{\alpha}}uptosupnormerrorup to sup-norm error|\bm{\alpha}|^{2}2^{-m}.$ Using the parallelization and depth synchronization properties in Section 7.1 yields then the claim of the lemma. ∎

Appendix B Proof of Theorem 5

We follow the classical idea of function approximation by local Taylor approximations that has previously been used for network approximations in . For a vector ar\mathbf{a}\in^{r} define

By Taylor’s theorem for multivariate functions, we have for a suitable ξ,\xi\in,

We have (xa)α=ixiaiαixaα.|(\mathbf{x}-\mathbf{a})^{\bm{\alpha}}|=\prod_{i}|x_{i}-a_{i}|^{\alpha_{i}}\leq|\mathbf{x}-\mathbf{a}|_{\infty}^{|\bm{\alpha}|}. Consequently, for fCrβ(r,K),f\in\mathcal{C}_{r}^{\beta}(^{r},K),

We may also write (31) as a linear combination of monomials

for suitable coefficients cγ.c_{\bm{\gamma}}. For convenience, we omit the dependency on a\mathbf{a} in cγ.c_{\bm{\gamma}}. Since γPaβf(x)x=0=γ!cγ,\partial^{\bm{\gamma}}P_{\mathbf{a}}^{\beta}f(\mathbf{x})\,|_{\mathbf{x}=0}=\bm{\gamma}!c_{\bm{\gamma}}, we must have

Notice that since ar\mathbf{a}\in^{r} and fCrβ(r,K),f\in\mathcal{C}_{r}^{\beta}(^{r},K),

If fCrβ(r,K),f\in\mathcal{C}_{r}^{\beta}(^{r},K), then \|P^{\beta}f-f\|_{L^{\infty}^{r}}\leq KM^{-\beta}.

Since for all x=(x1,,xr)r,\mathbf{x}=(x_{1},\ldots,x_{r})\in^{r},

In a next step, we describe how to build a network that approximates Pβf.P^{\beta}f.

For any positive integers M,m,M,m, there exists a network

with s49r2(M+1)r(1+(m+5)log2r),s\leq 49r^{2}(M+1)^{r}(1+(m+5)\lceil\log_{2}r\rceil), such that Hatr(M+1)r\operatorname{Hat}^{r}\in^{(M+1)^{r}} and for any x=(x1,,xr)r,\mathbf{x}=(x_{1},\ldots,x_{r})\in^{r},

By Lemma A.3 there exist Multmr\operatorname{Mult}_{m}^{r} networks in the class

All the constructed networks in this proof are of the form F(L,p,s)=F(L,p,s,)\mathcal{F}(L,\mathbf{p},s)=\mathcal{F}(L,\mathbf{p},s,\infty) with F=.F=\infty. Let MM be the largest integer such that (M+1)rN(M+1)^{r}\leq N and define L:=(m+5)log2(βr).L^{*}:=(m+5)\lceil\log_{2}(\beta\vee r)\rceil. Thanks to (34), (33) and Lemma A.4, we can add one hidden layer to the network Monm,βr\operatorname{Mon}_{m,\beta}^{r} to obtain a network

such that Q1(x)(M+1)rQ_{1}(\mathbf{x})\in^{(M+1)^{r}} and for any xr,\mathbf{x}\in^{r},

with B:=2Ker.B:=\lceil 2Ke^{r}\rceil. By (20), the number of non-zero parameters in the Q1Q_{1} network is bounded by 6r(β+1)Cr,β+42(β+1)2Cr,β2(L+1)+Cr,β(M+1)r.6r(\beta+1)C_{r,\beta}+42(\beta+1)^{2}C_{r,\beta}^{2}(L^{*}+1)+C_{r,\beta}(M+1)^{r}.

where for the last inequality, we used Cr,β(β+1)r,C_{r,\beta}\leq(\beta+1)^{r}, the definition of LL^{*} and that for any x1,x\geq 1, 1+log2(x)2+log2(x)2(1+log(x))2x.1+\lceil\log_{2}(x)\rceil\leq 2+\log_{2}(x)\leq 2(1+\log(x))\leq 2x.

To obtain a network reconstruction of the function ff, it remains to scale and shift the output entries. This is not entirely trivial because of the bounded parameter weights in the network. Recall that B=2Ker.B=\lceil 2Ke^{r}\rceil. The network xBMrxx\mapsto BM^{r}x is in the class F(3,(1,Mr,1,2Ker,1))\mathcal{F}(3,(1,M^{r},1,\lceil 2Ke^{r}\rceil,1)) with shift vectors vj\mathbf{v}_{j} are all equal to zero and weight matrices WjW_{j} having all entries equal to one. Because of N(K+1)er,N\geq(K+1)e^{r}, the number of parameters of this network is bounded by 2Mr+22Ker6N.2M^{r}+2\lceil 2Ke^{r}\rceil\leq 6N. This shows existence of a network in the class F(4,(1,2,2Mr,2,22Ker,1))\mathcal{F}(4,(1,2,2M^{r},2,2\lceil 2Ke^{r}\rceil,1)) computing aBMr(ac)a\mapsto BM^{r}(a-c) with c:=1/(2Mr).c:=1/(2M^{r}). This network computes in the first hidden layer (ac)+(a-c)_{+} and (ca)+(c-a)_{+} and then applies the network xBMrxx\mapsto BM^{r}x to both units. In the output layer the second value is subtracted from the first one. This requires at most 6+12N6+12N active parameters.

Because of (38) and (35), there exists a network Q3Q_{3} in

With (39), the number of non-zero parameters of Q3Q_{3} is bounded by

Observe that by construction (M+1)rN(M+2)r(3M)r(M+1)^{r}\leq N\leq(M+2)^{r}\leq(3M)^{r} and hence MβNβ/r3β.M^{-\beta}\leq N^{-\beta/r}3^{\beta}. Together with Lemma B.1, the result follows. ∎

Appendix C Proofs for Section 7.2

Throughout the proof we write E=Ef0.E=E_{f_{0}}. Define gn2:=1ni=1ng(Xi)2.\|g\|_{n}^{2}:=\frac{1}{n}\sum_{i=1}^{n}g(\mathbf{X}_{i})^{2}. For any estimator f~\widetilde{f}, we introduce \widehat{R}_{n}(\widetilde{f},f_{0}):=E\big{[}\|\widetilde{f}-f_{0}\|_{n}^{2}\big{]} for the empirical risk. In a first step, we show that we may restrict ourselves to the case logNnn.\log\mathcal{N}_{n}\leq n. Since R(f^,f0)4F2,R(\widehat{f},f_{0})\leq 4F^{2}, the upper bound trivially holds if logNnn.\log\mathcal{N}_{n}\geq n. To see that also the lower bound is trivial in this case, let f~argminfFi=1n(Yif(Xi))2\widetilde{f}\in\mathop{\rm arg\min}_{f\in\mathcal{F}}\sum_{i=1}^{n}(Y_{i}-f(\mathbf{X}_{i}))^{2} be a (global) empirical risk minimizer. Observe that

From this equation, it follows that Δn8F2\Delta_{n}\leq 8F^{2} and this implies the lower bound in the statement of the lemma for logNnn.\log\mathcal{N}_{n}\geq n. We may therefore assume logNnn.\log\mathcal{N}_{n}\leq n. The proof is divided into four parts which are denoted by (I)(IV).(I)-(IV).

(I): We relate the risk R(f^,f0)=E[(f^(X)f0(X))2]R(\widehat{f},f_{0})=E[(\widehat{f}(\mathbf{X})-f_{0}(\mathbf{X}))^{2}] to its empirical counterpart R^n(f^,f0)\widehat{R}_{n}(\widehat{f},f_{0}) via the inequalities

(II): For any estimator f~\widetilde{f} taking values in F,\mathcal{F},

Combining (I)(I) and (IV)(IV) gives the lower bound of the assertion since F1.F\geq 1. The upper bound follows from (I)(I) and (III).(III).

(I): Given a minimal δ\delta-covering of F,\mathcal{F}, denote the centers of the balls by fj.f_{j}. By construction there exists a (random) jj^{*} such that f^fjδ.\|\widehat{f}-f_{j^{*}}\|_{\infty}\leq\delta. Without loss of generality, we can assume that fjF.\|f_{j}\|_{\infty}\leq F. Generate i.i.d. random variables Xi,\mathbf{X}_{i}^{\prime}, i=1,,ni=1,\ldots,n with the same distribution as X\mathbf{X} and independent of (Xi)i=1,,n.(\mathbf{X}_{i})_{i=1,\ldots,n}. Using that fj,f0,δF,\|f_{j}\|_{\infty},\|f_{0}\|_{\infty},\delta\leq F,

with g_{j^{*}}(\mathbf{X}_{i},\mathbf{X}_{i}^{\prime}):=(f_{j^{*}}(\mathbf{X}_{i}^{\prime})-f_{0}(\mathbf{X}_{i}^{\prime})\big{)}^{2}-(f_{j^{*}}(\mathbf{X}_{i})-f_{0}(\mathbf{X}_{i}))^{2}. Define gjg_{j} in the same way with fjf_{j^{*}} replaced by fj.f_{j}. Similarly, set rj:=n1logNnE1/2[(fj(X)f0(X))2]r_{j}:=\sqrt{n^{-1}\log\mathcal{N}_{n}}\vee E^{1/2}[(f_{j}(\mathbf{X})-f_{0}(\mathbf{X}))^{2}] and define rr^{*} as rjr_{j} for j=j,j=j^{*}, which is the same as

where the last part follows from triangle inequality and fjf^δ.\|f_{j^{*}}-\widehat{f}\|_{\infty}\leq\delta.

For random variables U,T,U,T, Cauchy-Schwarz gives E[UT]E1/2[U2]E1/2[T2].E[UT]\leq E^{1/2}[U^{2}]E^{1/2}[T^{2}]. Choose U=E1/2[(f^(X)f0(X))2(Xi,Yi)i]U=E^{1/2}[(\widehat{f}(\mathbf{X})-f_{0}(\mathbf{X}))^{2}|(\mathbf{X}_{i},Y_{i})_{i}] and T:=maxji=1ngj(Xi,Xi)/(rjF).T:=\max_{j}|\sum_{i=1}^{n}g_{j}(\mathbf{X}_{i},\mathbf{X}_{i}^{\prime})/(r_{j}F)|. Using that E[U2]=R(f^,f0)E[U^{2}]=R(\widehat{f},f_{0})

Observe that E[gj(Xi,Xi)]=0,E[g_{j}(\mathbf{X}_{i},\mathbf{X}_{i}^{\prime})]=0, gj(Xi,Xi)4F2|g_{j}(\mathbf{X}_{i},\mathbf{X}_{i}^{\prime})|\leq 4F^{2} and

Bernstein’s inequality states that for independent and centered random variables U1,,Un,U_{1},\ldots,U_{n}, satisfying UiM,|U_{i}|\leq M, P(i=1nUit)2exp(t2/[2Mt/3+2i=1nVar(Ui)]),P(|\sum_{i=1}^{n}U_{i}|\geq t)\leq 2\exp(-t^{2}/[2Mt/3+2\sum_{i=1}^{n}\operatorname{Var}(U_{i})]), cf. . Combining Bernstein’s inequality with a union bound argument yields

The first term in the denominator of the exponent dominates for large t.t. Since rjn1logNn,r_{j}\geq\sqrt{n^{-1}\log\mathcal{N}_{n}}, we have P(Tt)2Nnexp(3tlogNn/(16n))P(T\geq t)\leq 2\mathcal{N}_{n}\exp(-3t\sqrt{\log\mathcal{N}_{n}}/(16\sqrt{n})) for all t6nlogNn.t\geq 6\sqrt{n\log\mathcal{N}_{n}}. We therefore find

By assumption Nn3\mathcal{N}_{n}\geq 3 and hence logNn1.\log\mathcal{N}_{n}\geq 1. We can argue in a similar way as for the upper bound of E[T]E[T] in order to find for the second moment

where the second inequality uses b2euadu=2bsesads=2(ba+1)eba/a2\int_{b^{2}}^{\infty}e^{-\sqrt{u}a}du=2\int_{b}^{\infty}se^{-sa}ds=2(ba+1)e^{-ba}/a^{2} which can be obtained from substitution and integration by parts. With (41) and 1logNnn,1\leq\log\mathcal{N}_{n}\leq n,

Let a,b,c,da,b,c,d be positive real numbers, such that ab2ac+d.|a-b|\leq 2\sqrt{a}c+d. Then, for any ε(0,1],\varepsilon\in(0,1],

To see this observe that ab2ac+d|a-b|\leq 2\sqrt{a}c+d implies aεa/(1+ε)+(1+ε)c2/ε+(b+d).a\leq\varepsilon a/(1+\varepsilon)+(1+\varepsilon)c^{2}/\varepsilon+(b+d). Rearranging the terms yields the upper bound. For the lower bound, we use the same argument and find aεa/(1ε)(1ε)c2/ε+(bd)a\geq-\varepsilon a/(1-\varepsilon)-(1-\varepsilon)c^{2}/\varepsilon+(b-d) which gives (43). With a=R(f^,f0),a=R(\widehat{f},f_{0}), b=R^n(f^,f0),b=\widehat{R}_{n}(\widehat{f},f_{0}), c=F\big{(}9n\log\mathcal{N}_{n}+64n\big{)}^{1/2}/n, d=F(6logNn+11)/n+26δFd=F(6\log\mathcal{N}_{n}+11)/n+26\delta F the asserted inequality of (I)(I) follows from (42). Notice that we have used 2(1+ε)/ε2\leq(1+\varepsilon)/\varepsilon for the upper bound.

(II): Given an estimator f~\widetilde{f} taking values in F,\mathcal{F}, let jj^{\prime} be such that f~fjδ.\|\widetilde{f}-f_{j^{\prime}}\|_{\infty}\leq\delta. We have E[i=1nεi(f~(Xi)fj(Xi))]δE[i=1nεi]nδ.|E[\sum_{i=1}^{n}\varepsilon_{i}(\widetilde{f}(\mathbf{X}_{i})-f_{j^{\prime}}(\mathbf{X}_{i}))]|\leq\delta E[\sum_{i=1}^{n}|\varepsilon_{i}|]\leq n\delta. Since E[εif0(Xi)]=E[E[εif0(Xi)Xi]]=0,E[\varepsilon_{i}f_{0}(\mathbf{X}_{i})]=E[E[\varepsilon_{i}f_{0}(\mathbf{X}_{i})\,|\,\mathbf{X}_{i}]]=0, we also find

Conditionally on (Xi)i,(\mathbf{X}_{i})_{i}, ξjN(0,1).\xi_{j}\sim\mathcal{N}(0,1). With Lemma C.1, we obtain E[ξj2]E[maxjξj2]3logNn+1.E[\xi_{j^{\prime}}^{2}]\leq E[\max_{j}\xi_{j}^{2}]\leq 3\log\mathcal{N}_{n}+1. Using Cauchy-Schwarz,

Because of logNnn,\log\mathcal{N}_{n}\leq n, we have 2n1/2δ3logNn+14δ.2n^{-1/2}\delta\sqrt{3\log\mathcal{N}_{n}+1}\leq 4\delta. Together with (44) and (45) the result follows.

(III): For any fixed fF,f\in\mathcal{F}, E[1ni=1n(Yif^(Xi))2]E[1ni=1n(Yif(Xi))2]+Δn.E[\tfrac{1}{n}\sum_{i=1}^{n}(Y_{i}-\widehat{f}(\mathbf{X}_{i}))^{2}]\leq E[\tfrac{1}{n}\sum_{i=1}^{n}(Y_{i}-f(\mathbf{X}_{i}))^{2}]+\Delta_{n}. Because of Xi=DX\mathbf{X}_{i}\stackrel{{\scriptstyle\mathcal{D}}}{{=}}\mathbf{X} and ff being deterministic, we have E[ff0n2]=E[(f(X)f0(X))2].E[\|f-f_{0}\|_{n}^{2}]=E[(f(\mathbf{X})-f_{0}(\mathbf{X}))^{2}]. Since also E[εif(Xi)]=E[E[εif(Xi)Xi]]=0,E[\varepsilon_{i}f(\mathbf{X}_{i})]=E[E[\varepsilon_{i}f(\mathbf{X}_{i})\,|\,\mathbf{X}_{i}]]=0,

using for the second inequality (II).(II). Applying (43) to a:=R^n(f^,f0),a:=\widehat{R}_{n}(\widehat{f},f_{0}), b:=0,b:=0, c:=(3logNn+1)/n,c:=\sqrt{(3\log\mathcal{N}_{n}+1)/n}, d:=E[(f(X)f0(X))2]+6δ+Δn,d:=E[(f(\mathbf{X})-f_{0}(\mathbf{X}))^{2}]+6\delta+\Delta_{n}, yields (III).(III).

(IV): Let f~argminfFi=1n(Yif(Xi))2\widetilde{f}\in\mathop{\rm arg\min}_{f\in\mathcal{F}}\sum_{i=1}^{n}(Y_{i}-f(\mathbf{X}_{i}))^{2} be an empirical risk minimizer. Using (40), (II)(II) and (1ε)/ε+1=1/ε,(1-\varepsilon)/\varepsilon+1=1/\varepsilon, we find

Rearranging of the terms leads then to the conclusion of (IV).(IV).

Let ηjN(0,1),\eta_{j}\sim\mathcal{N}(0,1), then E[maxj=1,,Mηj2]3logM+1.E[\max_{j=1,\ldots,M}\eta_{j}^{2}]\leq 3\log M+1.

Let Z=maxj=1,,Mηj2.Z=\max_{j=1,\ldots,M}\eta_{j}^{2}. Since Zjηj2,Z\leq\sum_{j}\eta_{j}^{2}, we have E[Z]M.E[Z]\leq M. For M{1,2,3}M\in\{1,2,3\} it can be checked that M3log(M)+1.M\leq 3\log(M)+1. It is therefore enough to consider M4.M\geq 4. Mill’s ratio gives P(η1t)=2P(η1t)2et/2/(2πt).P(|\eta_{1}|\geq\sqrt{t})=2P(\eta_{1}\geq\sqrt{t})\leq 2e^{-t/2}/(\sqrt{2\pi t}). For any T,T, we have using the union bound,

For T=2logMT=2\log M and M4,M\geq 4, we find E[Z]2logM+2/πlogM2logM+1.E[Z]\leq 2\log M+2/\sqrt{\pi\log M}\leq 2\log M+1.

References