Toward Deeper Understanding of Neural Networks: The Power of Initialization and a Dual View on Expressivity
Amit Daniely, Roy Frostig, Yoram Singer
Introduction
Neural network (NN) learning has underpinned state of the art empirical results in numerous applied machine learning tasks (see for instance ). Nonetheless, neural network learning remains rather poorly understood in several regards. Notably, it remains unclear why training algorithms find good weights, how learning is impacted by the network architecture and activations, what is the role of random weight initialization, and how to choose a concrete optimization procedure for a given architecture.
We start by analyzing the expressive power of NNs subsequent to the random weight initialization. The motivation is the empirical success of training algorithms despite inherent computational intractability, and the fact that they optimize highly non-convex objectives with potentially many local minima. Our key result shows that random initialization already positions learning algorithms at a good starting point. We define an object termed a computation skeleton that describes a distilled structure of feed-forward networks. A skeleton induces a family of network architectures along with a hypothesis class of functions obtained by certain non-linear compositions according to the skeleton’s structure. We show that the representation generated by random initialization is sufficiently rich to approximately express the functions in . Concretely, all functions in can be approximated by tuning the weights of the last layer, which is a convex optimization task.
In addition to explaining in part the success in finding good weights, our study provides an appealing perspective on neural network learning. We establish a tight connection between network architectures and their dual kernel spaces. This connection generalizes several previous constructions (see Sec 2). As we demonstrate, our dual view gives rise to design principles for NNs, supporting current practice and suggesting new ideas. We outline below a few points.
Duals of convolutional networks appear a more suitable fit for vision and acoustic tasks than those of fully connected networks.
Our framework surfaces a principled initialization scheme. It is very similar to common practice, but incorporates a small correction.
By modifying the activation functions, two consecutive fully connected layers can be replaced with one while preserving the network’s dual kernel.
The ReLU activation, i.e. , possesses favorable properties. Its dual kernel is expressive, and it can be well approximated by random initialization, even when the initialization’s scale is moderately changed.
As the number of layers in a fully connected network becomes very large, its dual kernel converges to a degenerate form for any non-linear activation.
Our result suggests that optimizing the weights of the last layer can serve as a convex proxy for choosing among different architectures prior to training. This idea was advocated and tested empirically in .
Related work
Understanding neural network learning, particularly its recent successes, commonly decomposes into the following research questions.
What functions can be efficiently expressed by neural networks?
When does a low empirical loss result in a low population loss?
Why and when do efficient algorithms, such as stochastic gradient, find good weights?
Though still far from being complete, previous work provides some understanding of questions (i) and (ii). Standard results from complexity theory imply that essentially all functions of interest (that is, any efficiently computable function) can be expressed by a network of moderate size. Biological phenomena show that many relevant functions can be expressed by even simpler networks, similar to convolutional neural networks that are dominant in ML tasks today. Barron’s theorem states that even two-layer networks can express a very rich set of functions. As for question (ii), both classical and more recent results from statistical learning theory show that as the number of examples grows in comparison to the size of the network the empirical loss must be close to the population loss. In contrast to the first two, question (iii) is rather poorly understood. While learning algorithms succeed in practice, theoretical analysis is overly pessimistic. Direct interpretation of theoretical results suggests that when going slightly deeper beyond single layer networks, e.g. to depth two networks with very few hidden units, it is hard to predict even marginally better than random . Finally, we note that the recent empirical successes of NNs have prompted a surge of theoretical work around NN learning .
Compositional kernels and connections to networks.
The idea of composing kernels has repeatedly appeared throughout the machine learning literature, for instance in early work by Schölkopf et al. , Grauman and Darrell . Inspired by deep networks’ success, researchers considered deep composition of kernels . For fully connected two-layer networks, the correspondence between kernels and neural networks with random weights has been examined in . Notably, Rahimi and Recht proved a formal connection (similar to ours) for the RBF kernel. Their work was extended to include polynomial kernels as well as other kernels . Several authors have further explored ways to extend this line of research to deeper, either fully-connected networks or convolutional networks . Our work sets a common foundation for and expands on these ideas. We extend the analysis from fully-connected and convolutional networks to a rather broad family of architectures. In addition, we prove approximation guarantees between a network and its corresponding kernel in our more general setting. We thus extend previous analyses that only applies to fully connected two-layer networks. Finally, we use the connection as an analytical tool to reason about architectural design choices.
Setting
Input space.
Supervised learning.
Neural network learning.
Kernel learning.
a convex objective that often can be efficiently minimized.
Computation skeletons
In this section we define a simple structure which we term a computation skeleton. The purpose of a computational skeleton is to compactly describe a feed-forward computation from an input to an output. A single skeleton encompasses a family of neural networks that share the same skeletal structure. Likewise, it defines a corresponding kernel space.
A computation skeleton is a DAG whose non-input nodes are labeled by activations.
Figure 1 shows four example skeletons, omitting the designation of the activation functions. The skeleton is rather basic as it aggregates all the inputs in a single step. Such topology can be useful in the absence of any prior knowledge of how the output label may be computed from an input example, and it is commonly used in natural language processing where the input is represented as a bag-of-words . The only structure in is a single fully connected layer:
An induced subgraph of a skeleton with nodes, , is called a fully connected layer if its edges are .
The skeleton is slightly more involved: it first processes consecutive (overlapping) parts of the input, and the next layer aggregates the partial results. Altogether, it corresponds to networks with a single one-dimensional convolutional layer, followed by a fully connected layer. The two-dimensional (and deeper) counterparts of such skeletons correspond to networks that are common in visual object recognition.
Let be positive integers and denote . A subgraph of a skeleton is a one dimensional convolution layer of width and stride if it has nodes, , and edges, , for .
The skeleton is a somewhat more sophisticated version of : the local computations are first aggregated, then reconsidered with the aggregate, and finally aggregated again. The last skeleton, , corresponds to the networks that arise in learning sequence-to-sequence mappings as used in translation, speech recognition, and OCR tasks (see for example Sutskever et al. ).
Note that the notion of the replication parameter corresponds, in the terminology of convolutional networks, to the number of channels taken in a convolutional layer and to the number of hidden units taken in a fully-connected layer.
Figure 2 illustrates a - and -realizations of a skeleton with coordinate dimension . The -realization is a network with a single (one dimensional) convolutional layer having channels, stride of , and width of , followed by three fully-connected layers. The global replication parameter in a realization is used for brevity; it is straightforward to extend results when the different nodes in are each replicated to a different extent.
We next define a scheme for random initialization of the weights of a neural network, that is similar to what is often done in practice. We employ the definition throughout the paper whenever we refer to random weights.
Architectures such as convolutional nets have weights that are shared across different edges. Again, it is straightforward to extend our results to these cases and for simplicity we assume no explicit weight sharing.
2 From computation skeletons to reproducing kernels
The following definition gives the kernel corresponding to a skeleton having normalized activations.For a skeleton with unnormalized activations, the corresponding kernel is the kernel of the skeleton obtained by normalizing the activations of .
The final kernel is , the kernel associated with the output node . The resulting Hilbert space and norm are denoted and respectively, and and denote the space and norm when formed at node .
As we show later, is indeed a (normalized) kernel for every skeleton . To understand the kernel in the context of learning, we need to examine which functions can be expressed as moderate norm functions in . As we show in section 7, these are the functions obtained by certain simple compositions according to the feed-forward structure of . For intuition, the following example contrasts two commonly used skeletons.
Consider a network whose activations are all ReLU, , and an input space . Say that is a skeleton comprising a single fully connected layer, and that is one comprising a convolutional layer of stride and width , followed by a single fully-connected layer. (The skeleton from Figure 1 is a concrete example of the convolutional skeleton with and .) The kernel takes the form . It is a symmetric kernel and therefore functions with small norm in are essentially low-degree polynomials. For instance, fix a bound on the norm of the functions. In this case, the space contains multiplication of one or two input coordinates. However, multiplication of or more coordinates are no-longer in . Moreover, this property holds true regardless of the choice of activation function. On the other hand, contains functions whose dependence on adjacent input coordinates is far more complex. It includes, for instance, any function that is symmetric (i.e. ) and that depends on adjacent coordinates . Furthermore, any sum of such functions is also in .
Main results
We review our main results. Let us fix a compositional kernel . There are a few upshots to underscore upfront. First, our analysis implies that a representation generated by a random initialization of approximates the kernel . The sense in which the result holds is twofold. First, with the proper rescaling we show that . Then, we also show that the functions obtained by composing bounded linear functions with are approximately the bounded-norm functions in . In other words, the functions expressed by under varying the weights of the last layer are approximately bounded-norm functions in . For simplicity, we restrict the analysis to the case . We also confine the analysis to either bounded activations, with bounded first and second derivatives, or the ReLU activation. Extending the results to a broader family of activations is left for future work. Through this and remaining sections we use to hide universal constants.
Note that many activations are -bounded for some constant . In particular, most of the popular sigmoid-like functions such as , , , , and satisfy the boundedness requirements. We next introduce terminology that parallels the representation layer of with a kernel space. Concretely, let be a network whose representation part has output neurons. Given weights , the normalized representation is obtained from the representation by dividing each output neuron by . The empirical kernel corresponding to is defined as . We also define the empirical kernel space corresponding to as . Concretely,
and the norm of is defined as . Our first result shows that the empirical kernel approximates the kernel .
Let be a skeleton with -bounded activations. Let be a random initialization of with
Then, for all , with probability of at least ,
We note that if we fix the activation and assume that the depth of is logarithmic, then the required bound on is polynomial. For the ReLU activation we get a stronger bound with only quadratic dependence on the depth. However, it requires that .
Let be a skeleton with ReLU activations. Let be a random initialization of with
Then, for all and , with probability of at least ,
Let be a skeleton with -bounded activations. Let be a random initialization of with
Then, with probability of at least over the choices of we have that -approximates and -approximates .
Let be a skeleton with ReLU activations and . Let be a random initialization of with
Then, with probability of at least over the choices of we have that -approximates and -approximates .
As in Theorems 2 and 3, for a fixed -bounded activation and logarithmically deep , the required bounds on are polynomial. Analogously, for the ReLU activation the bound is polynomial even without restricting the depth. However, the polynomial growth in Theorems 4 and 5 is rather large. Improving the bounds, or proving their optimality, is left to future work.
Our results can be extended to incorporate bias terms. Namely, in addition to the weights we can add a bias vector and let each neuron compute the function
To do so, we extend the definition of random initialization and compositional kernel as follows:
The final kernel is , the kernel associated with the output node .
Note that our original definitions correspond to . With the above definitions, Theorems 2, 3, 4 and 5 extend to the case when there exist bias terms. To simplify the notation, we focus on the case when there are no biases.
Mathematical background
The following theorem describes a tight connection between embeddings of into a Hilbert space and kernel spaces.
, where .
For every , .
Positive definite functions.
The norm of is defined as . We say that is normalized if
The restriction to the unit sphere of many of the kernels used in machine learning applications corresponds to positive definite functions. An example is the Gaussian kernel,
Indeed, note that for unit vectors we have
Another example is the Polynomial kernel .
Hermite polynomials.
The normalized Hermite polynomials is the sequence of orthonormal polynomials obtained by applying the Gram-Schmidt process to the sequence w.r.t. the inner-product . Recall that we define activations as square integrable functions w.r.t. the Gaussian measure. Thus, Hermite polynomials form an orthonormal basis to the space of activations. In particular, each activation can be uniquely described in the basis of Hermite polynomials,
Compositional kernel spaces
We now describe the details of compositional kernel spaces. Let be a skeleton with normalized activations and input nodes associated with the input’s coordinates. Throughout the rest of the section we study the functions in and their norm. In particular, we show that is indeed a normalized kernel. Recall that is defined inductively by the equation,
The recursion (7) describes a means for generating a kernel form another kernel. Since kernels correspond to kernel spaces, it also prescribes an operator that produces a kernel space from other kernel spaces. If is the space corresponding to , we denote this operator by
The function is indeed a kernel. Furthermore, the following properties hold.
If are normalized then so is .
(outline) The fact that is a kernel follows directly from the definition of a kernel and the fact that an average of PSD matrices is PSD. Also, it is straight forward to verify item 1. We now proceed to items 2 and 3. By Theorem 7 there are Hilbert spaces and mappings such that . Consider now the mapping
It holds that . Properties 2 and 3 now follow directly form Thm. 7 applied to . ∎
The extension of a kernel space.
Let be a normalized kernel space with a kernel . Let be a PSD function. As we will see shortly, a function is PSD if and only if it is a dual of an activation function. The extension of w.r.t. , denoted , is the kernel space corresponding to the kernel .
The function is indeed a kernel. Furthermore, the following properties hold.
is normalized if and only if is.
(outline) Let be a mapping from to the unit ball of a Hilbert space such that . Define
It is not difficult to verify that . Hence, by Thm. 7, is indeed a kernel. Verifying property 1 is a straightforward task. Properties 2 and 3 follow by applying Thm. 7 on the mapping . ∎
The dual activation function
The following lemma describes a few basic properties of the dual activation. These properties follow easily from the definition of the dual activation and equations (2), (4), and (5).
The following properties of the mapping hold:
If is the Hermite expansion of , then .
For every , is positive definite.
Every positive definite function is a dual of some activation.
The mapping preserves norms.
The mapping commutes with differentiation.
For every , is continuous in $(-1,1)$.
For every , is non-decreasing and convex in $$.
For every , the range of is .
We next discuss a few examples for activations and calculate their dual activation and kernel. Note that the dual of the exponential activation was calculated in and the duals of the step and the ReLU activations were calculated in . Here, our derivations are different and may prove useful for future calculations of duals for other activations.
Consider the activation function where is a normalization constant such that . The actual value of is but it will not be needed for the derivation below. From properties (e) and (f) of Lemma 11 we have that,
The the solution of ordinary differential equation is of the form . Since we have . We therefore obtain that the dual activation function is
Note that the kernel induced by is the RBF kernel, restricted to the -dimensional sphere,
The Sine activation and the Sinh kernel.
Consider the activation . We can write . We have
Similarly, since the variance of and is , we get
Hermite activations and polynomial kernels.
From Lemma 11 it follows that the dual activation of the Hermite polynomial is . Hence, the corresponding kernel is the polynomial kernel.
The normalized step activation.
To calculate we compute the Hermite expansion of . For we let
Since , , and , we get the corresponding coefficients,
For we write and note that
Here, the second equality follows from (4) and the third form (3). We therefore get
The second equality follows from (3) and the last equality follows from (6). Finally, from Lemma 11 we have that where
In particular, . Note that from the Taylor expansion of it follows that .
The normalized ReLU activation.
Consider the activation . We now write . The first coefficient is
To calculate the remaining coefficients we simply note that the derivative of the ReLU activation is the step activation and the mapping commutes with differentiation. Hence, from the calculation of the step activation we get,
In particular, . We see that the coefficients corresponding to the degrees , , and sum to . The sums up to degrees or are and respectively. That is, we get an excellent approximation of less than error with a dual activation of degree .
The collapsing tower of fully connected layers.
To conclude this section we discuss the case of very deep networks. The setting is taken for illustrative purposes but it might surface when building networks with numerous fully connected layers. Indeed, most deep architectures that we are aware of do not employ more than five consecutive fully connected layers.
Consider a skeleton consisting of fully connected layers, each layer associated with the same (normalized) activation . We would like to examine the form of the compositional kernel as the number of layers becomes very large. Due to the repeated structure and activation we have
Hence, the limiting properties of can be understood from the limit of . In the case that or , is the identity function. Therefore for all and is simply the linear kernel. Assume now that is neither the identity nor its negation. The following claim shows that has a point-wise limit corresponding to a degenerate kernel.
There exists a constant such that for all ,
Before proving the claim, we note that for , for all , and therefore . For , if is anti-symmetric then for all , and in particular . In any other case, our argument can show that .
Recall that where the ’s are non-negative numbers that sum to 1. By the assumption that is not the identity or its negation, . We first claim that there is a unique such that
To prove (9) it suffices to prove the following properties.
for
is non-decreasing and convex in $$
the graph of has at most a single intersection in with the graph of
If the above properties hold we can take to be the intersection point or if such a point does not exist. We first show (a). For we have that
Here, the third inequality follows form the fact that and for all , . Moreover since , one of these inequalities must be strict. Properties (b) and (c) follows from Lemma 11. Finally, to show (d), we note that the second derivative of is which is non-negative in . Hence, is convex in $x12\hat{\sigma}\hat{\sigma}(1)=112\rho=1[0,1)$.
Lastly, we derive the conclusion of the claim from equation (9). Fix . Assume first that . By (9), is a monotonically non-increasing sequence that is lower bounded by . Hence, it has a limit . Now, by the continuity of we have
Since the only solution to in is , we must have . We next deal with the case that . If for some , , the argument for shows that . If this is not the case, we have that for all , . As in the case of , this can be used to show that converges to . ∎
Proofs
Let be an activation. Define the following,
We underscore the following properties of the extension of a dual activation.
The restriction of the extended to the sphere agrees with the restricted definition.
The extended dual activation and kernel are defined for every activation such that for all , is square integrable with respect to the Gaussian measure.
A normalized activation is -decent for if the following conditions hold.
The dual activation is -Lipschitz in with respect to the -norm.
If are independent samples from for then
It is enough to show that the following properties hold.
The (extended) dual activation is -Lipschitz in w.r.t. the -norm.
If are independent samples from then
From the boundedness of it holds that . Hence, the second property follows directly from Hoeffding’s bound. We next prove the first part. Let and . Note that for we have
Let . Then, the first and second order partial derivatives of are
We conclude that is differentiable in with partial derivatives that are point-wise bounded by . Thus, is -Lipschitz in w.r.t. the -norm. ∎
We next show that the ReLU activation is decent.
The measure concentration property follows from standard concentration bounds for sub-exponential random variables (e.g. ). It remains to show that is -Lipschitz in . We first calculate an exact expression for . The expression was already calculated in , yet we give here a derivation for completeness.
The following equality holds for all ,
By the positive homogeneity of the ReLU activation we have
For brevity, we henceforth drop the argument from and use the abbreviation . In order to show that is -Lipschitz w.r.t. the -norm it is enough to show that for every we have,
First, Note that and have the same sign, hence,
We therefore get that the -norm of is,
The gradient of at is . Therefore, from the mean value theorem we get, . Furthermore, , and are bounded by in absolute value. Hence, we can write,
Finally, if we let , we can further simply the expression for ,
Finally, the proof is obtained from the fact that the function satisfies for every . Indeed, it is simple to verify that and . Hence, it suffices to show that is non-negative in $$ which is indeed the case since,
2 Proofs of Thms. 2 and 3
We start by an additional theorem which serves as a simple stepping stone for proving the aforementioned main theorems.
Let be a skeleton with -decent activations, , and . Let be a random initialization of the network with
Then, for every with probability of at least , it holds that
Before proving the theorem we show that together with Lemmas 12 and 13, Theorems 2 and 3 follow from Theorem 14. We restate them as corollaries, prove them, and then proceed to the proof of Theorem 14.
Let be a skeleton with -bounded activations. Let be a random initialization of with
Then, for every , w.p. ,
From Lemma 12, for all , each activation is -decent. By Theorem 14, it suffices to show that
Let be a skeleton with ReLU activations, and a random initialization of with . For all and , w.p. ,
Here, are universal constants.
This claim follows from the fact that as long as . Since we assume that , the expression is bounded by for sufficiently small . ∎
Note that . We say that a node , is well-initialized if
Here, we use the convention that . It is enough to show that with probability of at least all nodes are well-initialized. We first note that input nodes are well-initialized by construction since . Next, we show that given that all incoming nodes for a certain node are well-initialized, then w.h.p. the node is well-initialized as well.
It is easy to verify that is the empirical covariance matrix of independent variables distributed according to where . Given the assumption that all nodes incoming to are well-initialized, we have,
Further, since then . Using the fact that is -decent and that , we get that w.p. of at least ,
Finally, using (9.2) and (13) along with the fact that is -Lipschitz, we have
We are now ready to conclude the proof. Let be an ordered list of the nodes in in accordance to their depth, starting with the shallowest nodes, and ending with the output node. Denote by the event that are well-initialized. We need to show that . We do so using an induction on for the inequality . Indeed, for , is an input node and . Thus, the base of the induction hypothesis holds. Assume that . By Claim (3) we have that . Finally, from the induction hypothesis we have,
3 Proofs of Thms. 4 and 5
Theorems 4 and 5 follow from using the following lemma combined with Theorems 2 and 3. When we apply the lemma, we always focus on the special case where one of the kernels is constant w.p. .
For some , .
Then, w.p. over the choices of , for every there is such that .
To prove the above lemma, we state another lemma below followed by a basic measure concentration result.
Denote , , and . Suppose that we run stochastic gradient decent on the sample w.r.t. the loss , with learning rate , and with projections onto the ball of radius . Namely, we start with and at each iteration , we choose at random and perform the update,
After iterations the loss in expectation would be at most (see for instance Chapter 14 in ). In particular, there exists a sequence of at most gradient steps that attains a solution with . Each update adds or subtracts from the current solution. Hence can be written as a weighted sum of ’s where the sum of each coefficient is at most . ∎
In particular, w.p. (14) and (15) hold and therefore it suffices to prove the conclusion of the theorem under these conditions. Indeed, let be two mapping from to a Hilbert space so that . Let . By lemma 18 there are so that for the vector we have
Consider the function defined by . We note that
Discussion
Our results surface the question of the extent to which random initialization accounts for the success of neural networks. While we mostly leave this question for future research, we would like to point to empirical evidence supporting the important role of initialization. First, numerous researchers and practitioners demonstrated that random initialization, similar to the scheme we analyze, is crucial to the success of neural network learning (see for instance ). This suggests that starting from arbitrary weights is unlikely to lead to a good solution. Second, several studies show that the contribution of optimizing the representation layers is relatively small . For example, competitive accuracy on CIFAR-10, STL-10, MNIST and MONO datasets can be achieved by optimizing merely the last layer . Furthermore, Saxe et al. show that the performance of training the last layer is quite correlated with training the entire network. The effectiveness of optimizing solely the last layer is also manifested by the popularity of the random features paradigm . Finally, other studies show that the metrics induced by the initial and fully trained representations are not substantially different. Indeed, Giryes et al. demonstrated that for the MNIST and CIFAR-10 datasets the distances’ histogram of different examples barely changes when moving from the initial to the trained representation. For the ImageNet dataset the difference is more pronounced yet still moderate.
The role of architecture.
By using skeletons and compositional kernel spaces, we can reason about functions that the network can actually learn rather than merely express. This may explain in retrospect past architectural choices and potentially guide future choices. Let us consider for example the task of object recognition. It appears intuitive, and is supported by visual processing mechanisms in mammals, that in order to perform object recognition, the first processing stages are confined to local receptive fields. Then, the result of the local computations are applied to detect more complex shapes which are further combined towards a prediction. This processing scheme is naturally expressed by convolutional skeletons. A two dimensional version of Example 1 demonstrates the usefulness of convolutional networks for vision and speech applications.
The rationale we described above was pioneered by LeCun and colleagues . Alas, the mere fact that a network can express desired functions does not guarantee that it can actually learn them. Using for example Barron’s theorem , one may claim that vision-related functions are expressed by fully connected two layer networks, but such networks are inferior to convolutional networks in machine vision applications. Our result mitigates this gap. First, it enables use of the original intuition behind convolutional networks in order to design function spaces that are provably learnable. Second, as detailed in Example 1, it also explains why convolutional networks perform better than fully connected networks.
The role of other architectural choices.
In addition to the general topology of the network, our theory can be useful for understanding and guiding other architectural choices. We give two examples. First, suppose that a skeleton has a fully connected layer with the dual activation , followed by an additional fully connected layer with dual activation . It is straightforward to verify that if these two layers are replaced by a single layer with dual activation , the corresponding compositional kernel space remains the same. This simple observation can be useful in potentially saving a whole layer in the corresponding networks.
The second example is concerned with the ReLU activation, which is one of the most common activations used in practice. Our theory suggests a somewhat surprising explanation for its usefulness. First, the dual kernel of the ReLU activation enables expression of non-linear functions. However, this property holds true for many activations. Second, Theorem 3 shows that even for quite deep networks with ReLU activations, random initialization approximates the corresponding kernel. While we lack a proof at the time of writing, we conjecture that this property holds true for many other activations. What is then so special about the ReLU? Well, an additional property of the ReLU is being positive homogeneous, i.e. satisfying for all . This fact makes the ReLU activation robust to small perturbations in the distribution used for initialization. Concretely, if we multiply the variance of the random weights by a constant, the distribution of the generated representation and the space remain the same up to a scaling. Note moreover that training algorithms are sensitive to the initialization. Our initialization is very similar to approaches used in practice, but encompasses a small “correction”, in the form of a multiplication by a small constant which depends on the activation. For most activations, ignoring this correction, especially in deep networks, results in a large change in the generated representation. The ReLU activation is more robust to such changes. We note that similar reasoning applies to the max-pooling operation.
Future work.
Though our formalism is fairly general, we mostly analyzed fully connected and convolutional layers. Intriguing questions remain, such as the analysis of max-pooling and recursive neural network components from the dual perspective. On the algorithmic side, it is yet to be seen whether our framework can help in understanding procedures such as dropout and batch-normalization . Beside studying existing elements of neural network learning, it would be interesting to devise new architectural components inspired by duality. More concrete questions are concerned with quantitative improvements of the main results. In particular, it remains open whether the dependence on can be made polynomial and the quartic dependence on , , and can be improved. In addition to being interesting in their own right, improving the bounds may further underscore the effectiveness of random initialization as a way of generating low dimensional embeddings of compositional kernel spaces. Randomly generating such embeddings can be also considered on its own, and we are currently working on design and analysis of random features a la Rahimi and Recht .
Acknowledgments
We would like to thank Yossi Arjevani, Elad Eban, Moritz Hardt, Elad Hazan, Percy Liang, Nati Linial, Ben Recht, and Shai Shalev-Shwartz for fruitful discussions, comments, and suggestions.