Title: THE LOSS SURFACE OF RESIDUAL NETWORKS: ENSEMBLES & THE ROLE OF BATCH NORMALIZATION

Abstract: Deep Residual Networks present a premium in performance in comparison to conventional networks of the same depth and are trainable at extreme depths. It has recently been shown that Residual Networks behave like ensembles of relatively shallow networks. We show that these ensembles are dynamic: while initially the virtual ensemble is mostly at depths lower than half the network's depth, as training progresses, it becomes deeper and deeper. The main mechanism that controls the dynamic ensemble behavior is the scaling introduced, e.g., by the Batch Normalization technique. We explain this behavior and demonstrate the driving force behind it. As a main tool in our analysis, we employ generalized spin glass models, which we also use in order to study the number of critical points in the optimization of Residual Networks.

1. INTRODUCTION:

Residual Networks (He et al., 2015) (ResNets) are neural networks with skip connections. These networks, which are a specific case of Highway Networks (Srivastava et al., 2015) , present state of the art results in the most competitive computer vision tasks including image classification and object detection.

The success of residual networks was attributed to the ability to train very deep networks when employing skip connections (He et al., 2016) . A complementary view is presented by Veit et al. (2016) , who attribute it to the power of ensembles and present an unraveled view of ResNets that depicts ResNets as an ensemble of networks that share weights, with a binomial depth distribution around half depth. They also present experimental evidence that short paths of lengths shorter than half-depth dominate the ResNet gradient during training.

The analysis presented here shows that ResNets are ensembles with a dynamic depth behavior. When starting the training process, the ensemble is dominated by shallow networks, with depths lower than half-depth. As training progresses, the effective depth of the ensemble increases. This increase in depth allows the ResNet to increase its effective capacity as the network becomes more and more accurate.

Our analysis reveals the mechanism for this dynamic behavior and explains the driving force behind it. This mechanism remarkably takes place within the parameters of Batch Normalization (Ioffe & Szegedy, 2015) , which is mostly considered as a normalization and a fine-grained whitening mechanism that addresses the problem of internal covariate shift and allows for faster learning rates.

We show that the scaling introduced by batch normalization determines the depth distribution in the virtual ensemble of the ResNet. These scales dynamically grow as training progresses, shifting the effective ensemble distribution to bigger depths.

The main tool we employ in our analysis is spin glass models. Choromanska et al. (2015a) have created a link between conventional networks and such models, which leads to a comprehensive study of the critical points of neural networks based on the spin glass analysis of Auffinger et al. (2013) . In our work, we generalize these results and link ResNets to generalized spin glass models. These models allow us to analyze the dynamic behavior presented above. Finally, we apply the results of Auffinger & Arous (2013) in order to study the loss surface of ResNets.

2. A RECAP OF CHOROMANSKA ET AL. (2015A):

We briefly summarize Choromanska et al. (2015a) , which connects the loss function of multilayer networks with the hamiltonian of the p spherical spin glass model, and state their main contributions and results. The notations of our paper are summarized in Appendix A and slightly differ from those in Choromanska et al. (2015a) .

A simple feed forward fully connected network N , with p layers and a single output unit is considered. Let n i be the number of units in layer i, such that n 0 is the dimension of the input, and n p = 1. It is further assumed that the ReLU activation functions denoted by R() are used. The output Y of the network given an input vector x ∈ R d can be expressed as

EQUATION

where the first summation is over the network inputs x 1 ...x d , and the second is over all paths from input to output. There are γ = p i=1 n i such paths and ∀i, x i1 = x i2 = ...x iγ . The variable A ij ∈ {0, 1} denotes whether the path is active, i.e., whether all of the ReLU units along this path are producing positive activations, and the product

p k=1 w (k) ij represents the specific weight configuration w 1 ij ...w k ij multiplying x i given path j.

It is assumed throughout the paper that the input variables are sampled i.i.d from a normal Gaussian distribution. Definition 1. The mass of the network N is defined as ψ = p i=0 n i .

The variables A ij are modeled as independent Bernoulli random variables with a success probability ρ, i.e., each path is equally likely to be active. Therefore,

E A [Y ] = d i=1 γ j=1 x ij ρ p k=1 w (k) ij .

(2)

The task of binary classification using the network N with parameters w is considered, using either the hinge loss L h N or the absolute loss L a N :

EQUATION

where Y x is a random variable corresponding to the true label of sample x. In order to equate either loss with the hamiltonian of the p-spherical spin glass model, a few key approximations are made:

A1 Variable independence -The inputs x ij are modeled as independent normal Gaussian random variables.

A2 Redundancy in network parameterization -It is assumed the set of all the network weights [w 1 , w 2 ...w N ] contains only Λ unique weights such that Λ < N .

A3 Uniformity -It is assumed that all unique weights are close to being evenly distributed on the graph of connections defining the network N . Practically, this means that we assume every node is adjacent to an edge with any one of the Λ unique weights.

A4 Spherical constraint -The following is assumed:

1 Λ Λ i=1 w 2 i = C 2 (4)

for some constant C > 0.

These assumptions are made for the sake of analysis, and do not necessarily hold. The validity of these assumption was posed as an open problem in Choromanska et al. (2015b) , where a different degree of plausibility was assigned to each. Specifically, A1, as well as the independence assumption of A ij , were deemed unrealistic, and A2 -A4 as plausible. For example, A1 does not hold since each input x i is associated with many different paths and x i1 = x i2 = ...x iγ . See Choromanska et al. (2015a) for further justification of these approximations.

Under A1-A4, the loss takes the form of a centered Gaussian process on the sphere S Λ-1 ( √ Λ). Specifically, it is shown to resemble the hamiltonian of the a spherical p-spin glass model given by:

H p,Λ ( w) = 1 Λ p-1 2 Λ i1...ip x i1...ip r k=1 wi k (5) with spherical constraint 1 Λ Λ i=1 w2 i = 1 (6)

where x i1...ip are independent normal Gaussian variables.

In Auffinger et al. (2013) , the asymptotic complexity of spherical p spin glass model is analyzed based on random matrix theory. In Choromanska et al. (2015a) these results are used in order to shed light on the optimization process of neural networks. For example, the asymptotic complexity of spherical spin glasses reveals a layered structure of low-index critical points near the global optimum. These findings are then given as a possible explanation to several central phenomena found in neural networks optimization, such as similar performance of large nets, and the improbability of getting stuck in a "bad" local minima.

As part of our work, we follow a similar path. First, a link is formed between residual networks and the hamiltonian of a general multi-interaction spherical spin glass model as given by:

EQUATION

where 1 ... p are positive constants. Then, using Auffinger & Arous (2013) , we obtain insights on residual networks. The other part of our work studies the dynamic behavior of residual networks, where we relax the assumptions made for the spin glass model.

3. RESIDUAL NETS AND GENERAL SPIN GLASS MODELS:

We begin by establishing a connection between the loss function of deep residual networks and the hamiltonian of the general spherical spin glass model. We consider a simple feed forward fully connected network N , with ReLU activation functions and residual connections. For simplicity of notations without the loss of generality, we assume n 1 = ... = n p = n. n 0 = d as before. In our ResNet model, there exist p -1 identity connections skipping a single layer each, starting from the first hidden layer. The output of layer l > 1 is given by:

EQUATION

where W l denotes the weight matrix connecting layer l -1 with layer l. Notice that the first hidden layer has no parallel skip connection, and so N 1 (x) = R(W 1 x). Without loss of generality, the scalar output of the network is the sum of the outputs of the output layer p and is expressed as

EQUATION

where A (r) ij ∈ {0, 1} denotes whether path j of length r is open, and ∀j, j , r, r x r ij = x r ij . The residual connections in N imply that the output Y is now the sum of products of different lengths, indexed by r. Since our ResNet model attaches a skip connection to every layer except the first, 1 ≤ r ≤ p. See Sec. 6 regarding models with less frequent skip connections.

Each path of length r includes r -1 non-skip connections (those involving the first term in Eq. 8 and not the second, identity term) out of layers l = 2..p. Therefore, γ r = p-1 r-1 n r . We define the following measure on the network: Definition 2. The mass of a depth r subnetwork in N is defined as ψ r = dγ r .

The properties of redundancy in network parameters and their uniform distribution, as described in Sec. 2, allow us to re-index Eq. 9.

Lemma 1. Assuming assumptions A2 -A4 hold, and n Λ ∈ Z, then the output can be expressed after reindexing as:

Y = p r=1 Λ i1,i2...ir=1 ψr Λ r j=1 x (j) i1,i2...ir A (j) i1,i2...ir r k=1 w i k .

(10)

All proofs can be found in Appendix B.

Making the modeling assumption that the ReLU gates are independent Bernoulli random variables with probability ρ, we obtain that for every path of length r, E A

EQUATION

In order to connect ResNets to generalized spherical spin glass models, we denote the variables:

ξ i1,i2...ir = ψr Λ r j=1 x j i1,i2...ir , xi1,i2...ir = ξ i1,i2...ir E x [ξ 2 i1,i2...ir ] 1 2 (12)

Note that since the input variables x 1 ...x d are sampled from a centered Gaussian distribution (dependent or not), then the set of variables xi1,i2...ir are dependent normal Gaussian variables. Lemma 2. Assuming A2 -A3 hold, and n Λ ∈ N then ∀ r,i1...ir the following holds:

EQUATION

)

We approximate the expected output E A (Y ) with Ŷ by assuming the minimal value in 13 holds such that ∀ r,i1...ir E[ξ 2 i1,i2...ir ] = 1 d ( ψr Λ r ) 2 . This approximation holds exactly when Λ = n, since all weight configurations of a particular length in Eq. 10 will appear the same number of times. When Λ = n, the uniformity assumption dictates that each configuration of weights would appear approximately equally regardless of the inputs, and the expectation values would be very close to the lower bound. The following expression for Ŷ is thus obtained:

EQUATION

The independence assumption A1 was not assumed yet, and 14 holds regardless. Assuming A4 and denoting the scaled weights wi = 1 C w i , we can link the distribution of Ŷ to the distribution on x:

Ŷ = p r=1 ψ r √ d ( ρC Λ ) r Λ i1,i2...ir=1 xi1,i2...ir r k=1 wi k = z p r=2 r Λ r-1 2 Λ i1,i2...ir=1 xi1,i2...ir r k=1 wi k (15) where r = r = 1 z p-1 r-1 ( ρnC √ Λ ) r and z is a normalization factor such that p r=1 2 r = 1.

The following lemma gives a generalized expression for the binary and hinge losses of the network. Lemma 3 ( Choromanska et al. (2015a) ). Assuming assumptions A2 -A4 hold, then both the losses L h N (x) and L a N (x) can be generalized to a distribution of the form:

EQUATION

)

where C 1 , C 2 are positive constants that do not affect the optimization process.

The model in Eq. 16 has the form of a spin glass model, except for the dependency between the variables xi1,i2...ir . We later use an assumption similar to A1 of independence between these variables in order to link the two binary classification losses and the general spherical spin glass model. However, for the results in this section, this is not necessary.

We denote the important quantities:

EQUATION

The series ( r ) p r=1 determines the weight of interactions of a specific length in the loss surface. Notice that for constant depth p and large enough β, arg max r ( r ) = p. Therefore, for wide networks, where n and, therefore, β are large, interactions of order p dominate the loss surface, and the effect of the residual connections diminishes. Conversely, for constant β and a large enough p (deep networks), we have that arg max r ( r ) < p, and can expect interactions of order r < p to dominate the loss. The asymptotic behavior of is captured by the following lemma:

Theorem 1. Assuming β 1+β p ∈ N, we have that:

lim p→∞ 1 p arg max r ( r ) = β 1 + β (18)

As the next theorem shows, the epsilons are concentrated in a narrow band near the maximal value.

Theorem 2. For any α 1 < β 1+β < α 2 , and assuming α 1 p, α 2 p, β 1+β p ∈ N, it holds that:

EQUATION

Thm. 2 implies that for deep residual networks, the contribution of weight products of order far away from the maximum β 1+β p is negligible. The loss is, therefor, similar in complexity to that of an ensemble of potentially shallow conventional nets. The next Lemma shows that we can shift the effective depth to any value by simply controlling C. Lemma 4. For any integer 1 ≤ k ≤ p there exists a global scaling parameter C such that arg max r ( r (C)) = k.

A simple global scaling of the weights is, therefore, enough to change the loss surface, from an ensemble of shallow conventional nets, to an ensemble of deep nets. This is illustrated in Fig. 1(a-c ) for various values of β. In a common weight initialization scheme for neural networks, C = 1 √ n (Orr & Müller, 2003; Glorot & Bengio, 2010) . With this initialization and Λ = n, β = ρ and the maximal weight is obtained at less than half the network's depth lim p→∞ arg max r ( r ) < p 2 . Therefore, at the initialization, the loss function is primarily influenced by interactions of considerably lower order than the depth p, which facilitates easier optimization.

4. DYNAMIC BEHAVIOR OF RESIDUAL NETS:

The expression for the output of a residual net in Eq. 15 provides valuable insights into the machinery at work when optimizing such models. Thm. 1 and 2 imply that the loss surface resembles that of an ensemble of shallow nets (although not a real ensemble due to obvious dependencies), with various depths concentrated in a narrow band. As noticed in Veit et al. (2016) , viewing ResNets as ensembles of relatively shallow networks helps in explaining some of the apparent advantages of these models, particularly the apparent ease of optimization of extremely deep models, since deep paths barely affect the overall loss of the network. However, this alone does not explain the increase in accuracy of deep residual nets over actual ensembles of standard networks. In order to explain the improved performance of ResNets, we make the following claims:

1. The distribution of the depths of the networks within the ensemble is controlled by the scaling parameter C.

2.:

During training, C changes and causes a shift of focus from a shallow ensemble to deeper and deeper ensembles, which leads to an additional capacity.

3. In networks that employ batch normalization, C is directly embodied as the scale parameter λ. The starting condition of λ = 1 offers a good starting condition that involves extremely shallow nets.

For the remainder of Sec.4, we relax all assumptions, and assume that at some point in time

1 Λ Λ i=1 w 2 i = C 2

, and Λ = N . Using Eq. 9 for the output of the network Ŷ in Lemma. 3, the loss can be expressed:

L N (x, w) = C 1 + C 2 p r=1 d i=1 γr j=1 x (r) ij A (r) ij r k=1 w (r)(k) ij (20)

where C 1 , C 2 are some constants that do not affect the optimization process. In order to gain additional insight into this dynamic mechanism, we investigate the derivative of the loss with respect to the scale parameter C. Using Eq. 9 for the output, we obtain:

EQUATION

Notice that the addition of a multiplier r indicates that the derivative is increasingly influenced by deeper networks.

4.1. BATCH NORMALIZATION:

Batch normalization has shown to be a crucial factor in the successful training of deep residual networks. As we will show, batch normalization layers offer an easy starting condition for the network, such that the gradients from early in the training process will originate from extremely shallow paths.

We consider a simple batch normalization procedure, which ignores the additive terms, has the output of each ReLU unit in layer l normalized by a factor σ l and then is multiplied by some parameter λ l . The output of layer l > 1 is therefore:

EQUATION

where σ l is the mean of the estimated standard deviations of various elements in the vector R(W l N l-1 (x)). Furthermore, a typical initialization of batch normalization parameters is to set ∀ l , λ l = 1. In this case, providing that units in the same layer have equal variance σ l , the recursive relation

E[N l+1 (x) 2 j ] = 1 + E[N l (x) 2 j ]

holds for any unit j in layer l. This, in turn, implies that the output of the ReLU units should have increasing variance σ 2 l as a function of depth. Multiplying the weight parameters in deep layers with an increasingly small scaling factor 1 σ l , effectively reduces the influence of deeper paths, so that extremely short paths will dominate the early stages of optimization. We next analyze how the weight scaling, as introduced by batch normalization, provides a driving force for the effective ensemble to become deeper as training progresses.

4.2. THE DRIVING FORCE BEHIND THE SCALE INCREASE:

The analysis below focuses on a single residual connection skipping a block of one or more layers. Since it holds for each block individually, it holds also for a residual network of multiple skipped blocks of arbitrary depth.

We consider a simple network of depth p, with a single residual connection skipping p -m layers. We further assume that batch normalization is applied at the output of each ReLU unit as described in Eq. 22. We denote by l 1 ...l m the indices of layers that are not skipped by the residual connection, and λm = m i=1 λ l i σ l i , λp = p i=1 λi σi . Since every path of length m is multiplied by λm , and every path of length p is multiplied by λp , the expression for the loss can be expressed using Eq. 20 and ignoring constant terms:

L N (x, w) = λm d i=1 γm j=1 x (m) ij A (m) ij r k=1 w (m)(k) ij + λp d i=1 γp j=1 x (m) ij A (p) ij p k=1 w (p)(k) ij = L m (x, w) + L p (x, w) (23)

We denote by ∇ w the derivative operator with respect to the parameters w, and the gradient g = ∇ w L N (x, w) = g m + g p evaluated at point w.

Theorem 3. Considering the loss in 23, and assuming ∂L N (x,w) ∂λ l = 0, then for a small learning rate 0 < µ << 1 the following hold:

1. For any λ l∈l1...lm then:

λ l -µ ∂L N (x, w -µg) ∂λ l > |λ l | (24)

2. For any λ l ∈l1...lm , if g p 2 2 + g p g m > 0 then: This steady increase of |λ l |, as predicted in our theoretical analysis, is also backed in experimental results, as depicted in Fig. 1(d ). Note that the first layer, which cannot be skipped, behaves differently than the other layers. More experiments can be found in Appendix C.

EQUATION

It is worth noting that the mechanism for this dynamic property of residual networks can also be observed without the use of batch normalization, as a steady increase in the L2 norm of the weights, as shown in Fig. 1(e) . In order to model this, consider the residual network as discussed above, without batch normalization layers. Recalling, w 2 = C √ Λ, w = w C , the loss of this network is expressed as:

L N (x, w) = C m d i=1 γm j=1 x (m) ij A (m) ij r k=1 w(m)(k) ij + C p d i=1 γp j=1 x (m) ij A (p) ij p k=1 w(p)(k) ij = L m (x, w) + L p (x, w) (26)

Theorem 4. Considering the loss in 26, and assuming ∂L N (x,w) ∂C = 0, then for a small learning rate 0 < µ << 1 the following hold:

EQUATION

Thm. 4 indicates that if either g p 2 or g m 2 is dominant (for example, near local minimas of the shallow network, or at the start of training), the scaling of the weights C will increase. This expansion will, in turn, emphasize the contribution of deeper paths over shallow paths, and increase the overall capacity of the residual network. This dynamic behavior of the effective depth of residual networks is of key importance in understanding the effectiveness of these models. While optimization starts off rather easily with gradients largely originating from shallow paths, the overall advantage of depth is still maintained by the dynamic increase of the effective depth.

5. THE LOSS SURFACE OF ENSEMBLES:

We now present the results of Auffinger & Arous (2013) regarding the asymptotic complexity in the case of lim Λ→∞ of the multi-spherical spin glass model given by:

H ,Λ = - ∞ r=2 r Λ r-1 2 Λ i1,...ir=1

J r i1...ir wi2 ... wir (28)

where J r i1...ir are independent centered standard Gaussian variables, and = ( r ) r>2 are positive real numbers such that ∞ r=2 r 2 r < ∞. A configuration w of the spin spherical spin-glass model is a vector in R Λ satisfying the spherical constraint:

1 Λ Λ i=1 w 2 i = 1, ∞ r=2 2 r = 1 (29)

Note that the variance of the process is independent of :

E[H 2 ,Λ ] = ∞ r=2 Λ 1-r 2 r ( Λ i=1 w 2 i ) r = Λ ∞ r=1 2 r = Λ (30)

Definition 3. We define the following:

v = ∞ r=2 2 r r, v = ∞ r=2 2 r r(r -1), α 2 = v + v -v 2 (31)

Note that for the single interaction spherical spin model α 2 = 0. The index of a critical point of H ,Λ is defined as the number of negative eigenvalues in the hessian ∇ 2 H ,Λ evaluated at the critical point w. Definition 4. For any 0 ≤ k < Λ and u ∈ R, we denote the random number Crt λ,k (u, ) as the number of critical points of the hamiltonian in the set BX = {ΛX|X ∈ (-∞, u)} with index k.

That is: Auffinger & Arous (2013) states that for any k > 0:

Crt Λ,k (u, ) = w:∇H ,Λ =0 1 {H ,Λ ∈ Λu} 1 i(∇ 2 H ,Λ ) = k (32) Furthermore, define θ k (u, ) = lim Λ→∞ 1 Λ log E[Crt Λ,k (u )]. Corollary 1.1 of

θ k (R, ) = 1 2 log( v v ) - v -v v + v (33)

Eq. 33 provides the asymptotic mean total number of critical points with non-diverging index k. It is presumed that the SGD algorithm will easily avoid critical points with a high index that have many descent directions, and maneuver towards low index critical points. We, therefore, investigate how the mean total number of low index critical points vary as the ensemble distribution embodied in ( r ) r>2 changes its shape by a steady increase in β.

Fig. 1 (f) shows that as the ensemble progresses towards deeper networks, the mean amount of low index critical points increases, which might cause the SGD optimizer to get stuck in local minima. This is, however, resolved by the the fact that by the time the ensemble becomes deep enough, the loss function has already reached a point of low energy as shallower ensembles were more dominant earlier in the training. In the following theorem, we assume a finite ensemble such that ∞ r=p+1 r 2 r ≈ 0. Theorem 5. For any k ∈ N, p > 1, we denote the solution to the following constrained optimization problems:

EQUATION

Thm. 5 implies that any heterogeneous mixture of spin glasses contains fewer critical points of a finite index, than a mixture in which only p interactions are considered. Therefore, for any distribution of that is attainable during the training of a ResNet of depth p, the number of critical points is lower than the number of critical points for a conventional network of depth p.

6. DISCUSSION:

In this work, we use spin glass analysis in order to understand the dynamic behavior ResNets display during training and to study their loss surface. In particular, we use at one point or another the assumptions of redundancy in network parameters, near uniform distribution of network weights, independence between the inputs and the paths and independence between the different copies of the input as described in Choromanska et al. (2015a) . The last two assumptions, i.e., the two independence assumptions, are deemed in Choromanska et al. (2015b) as unrealistic, while the remaining are considered plausible.

Our analysis of critical points in ensembles (Sec. 5) requires all of the above assumptions. However, Thm. 1 and 2, as well as Lemma. 4, do not assume the last assumption, i.e., the independence between the different copies of the input. Moreover, the analysis of the dynamic behavior of residual nets (Sec. 4) does not assume any of the above assumptions.

Our results are well aligned with some of the results shown in Larsson et al. (2016) , where it is noted empirically that the deepest column trains last. This is reminiscent of our claim that the deeper networks of the ensemble become more prominent as training progresses. The authors of Larsson et al. (2016) hypothesize that this is a result of the shallower columns being stabilized at a certain point of the training process. In our work, we discover the exact driving force that comes into play.

In addition, our work offers an insight into the mechanics of the recently proposed densely connected networks (Huang et al., 2016) . Following the analysis we provide in Sec. 3, the additional shortcut paths decrease the initial capacity of the network by offering many more short paths from input to output, thereby contributing to the ease of optimization when training starts. The driving force mechanism described in Sec. 4.2 will then cause the effective capacity of the network to increase.

Note that the analysis presented in Sec. 3 can be generalized to architectures with arbitrary skip connections, including dense nets. This is done directly by including all of the induced sub networks in Eq. 9. The reformulation of Eq. 10 would still holds, given that Ψ r is modified accordingly.

7. CONCLUSION:

Ensembles are a powerful model for ResNets, which unravels some of the key questions that have surrounded ResNets since their introduction. Here, we show that ResNets display a dynamic ensemble behavior, which explains the ease of training such networks even at very large depths, while still maintaining the advantage of depth. As far as we know, the dynamic behavior of the effective capacity is unlike anything documented in the deep learning literature. Surprisingly, the dynamic mechanism typically takes place within the outer multiplicative factor of the batch normalization module. The number of units in layers l > 0 Λ

The number of unique weights in the network N

The total number of weights in the network N W l

The weight matrix connecting layer l -1 to layer l in N .

H p,λ:

The hamiltonian of the p interaction spherical spin glass model. The mean of the estimated standard deviation various elements in R(W l N l-1 (x)).

B PROOFS:

Proof of Lemma 1. There are a total of ψ r paths of length r from input to output, and a total of Λ r unique r length configurations of weights. The uniformity assumption then implies that each configuration of weights is repeated ψr Λ r times. By summing over the unique configurations, and re indexing the input we arrive at Eq. 10.

Proof of Lemma 2. From 12, we have that ξ i1,i2...ir is defined as a sum of ψr Λ r inputs. Since there are only p distinct inputs, it holds that for each ξ i1,i2...ir there exists a sequence α = (α i ) p i=1 ∈ N such that d i=1 α i = ψr Λ r , and ξ i1,i2...ir = d i=1 α i x i . We, therefore, have that E[ξ 2 i1,i2...ir ] = α 2 2 . Note that the minimum value of E[ξ 2 i1,i2...ir ] is a solution to the following:

EQUATION

which achieves its minimal value at ∀ i , α i = 1 d ψr Λ r . Similarly, the maximum value is achieved at α i = ψr Λ r δ i for some index i.

Proof of Thm 4. Notice that ∂L N (x,w) ∂C = ∂L N (x,w) ∂w ∂w ∂ w 2 √ Λ = g w = 0, and hence the gradient is orthogonal to the weights. We have that ∂L N (x,w) ∂C = 1 C (mL m (x, w) + pL p (x, w)). Using taylor series expansion we have:

EQUATION

For the last term we have:

EQUATION

where the last step stems from the fact that w g = 0. Substituting ∇ w ∂L N (x,w) ∂C

= 1 C (mg m +pg p ) in 45 we have:

∂L N (x, w -µg w ) ∂C ≈ 0 -µ 1 C (mg m + pg p ) (g m + g p ) = -µ 1 C (m g p 2 2 + p g p 2 2 + (m + p)g p g m ) (47)

Proof of Thm 5. Inserting Eq. 31 into Eq. 33 we have that:

EQUATION

We denote the matrices V and V such that V ij = rδ ij and V ij = r(r -1)δ ij . We then have:

EQUATION

C ADDITIONAL EXPERIMENTS Fig. 1(d ) and 1(e) report the experimental results of a straightforward setting, in which the task is to classify a mixture of 10 multivariate Gaussians in 50D. The input is therefore of size 50. The loss employed is the cross entropy loss of ten classes. The network has 10 blocks, each containing 20 hidden neurons, a batch normalization layer, and a skip connection. Training was performed on 10,000 samples, using SGD with minibatches of 50 samples.

Next, we provide additional experiments performed on the public CIFAR-10 and CIFAR-100 data sets (Krizhevsky, 2009) . The public ResNet code of https://github.com/facebook/fb. resnet.torch is used for networks of depth 32.

As noted in Sec. 4.2, the dynamic behavior can be present in the Batch Normalization multiplicative coefficient or in the weight matrices themselves. In the following experiments, it seems that until the learning rate is reduced, the dynamic behavior is manifested in the Batch Normalization multiplicative coefficients and then it moves to the convolution layers themselves. We therefore absorb the BN coefficients into the convolutional layer using the public code of https: //github.com/e-lab/torch-toolbox/tree/master/BN-absorber. Note that the multiplicative coefficient of Batch Normalization is typically refereed to as γ. However, throughout our paper, since we follow the notation of Choromanska et al. (2015a) , γ refers to the number of paths. The multiplicative factor of Batch normalization appears as λ in Sec. 4. As can be seen, the dynamic phenomenon we describe is very prominent in the public ResNet implementation when applied to these conventional datasets: the dominance of paths with fewer skip connections increases over time. Moreover, once the learning rate is reduced in epoch 81 the phenomenon we describe speeds up.

In Fig. 3 we present the multiplicative coefficient of the Batch Normalization when not absorbed.

As future work, we would like to better understand why these coefficients start to decrease once the learning rate is reduced. As shown above, taking the magnitude of the convolutions into account, the dynamic phenomenon we study becomes even more prominent at this point. The change of location from the multiplicative coefficient of the Batch Normalization layers to the convolutions themselves might indicate that Batch Normalization is no longer required at this point. Indeed, Batch Normalization enables larger training rates and this shift happens exactly when the training rate is reduced. A complete analysis is left for future work. 

:



Figures and Tables:

Figure 1: (a) A histogram of r (β), r = 1..p, for β = 0.1 and p = 100 . (b) Same for β = 0.5 (c) Same for β = 2. (d) Values (y-axis) of the batch normalization parameters λ l (x-axis) for 10 layers ResNet trained to discriminate between 50 multivariate Gaussians (see Appendix C for more details). Higher plot lines indicate later stages of training. (e) The norm of the weights of a residual network, which does not employ batch normalization, as a function of the iteration. (f) The asymptotic of the mean number of critical points of a finite index as a function of β.

Fig.2depicts the results. There are two types of plots: Fig.2(a,c) presents for CIFAR-10 and CIFAR-100 respectively the magnitude of the various convolutional layers for multiple epochs (similar in type to Fig.1(d) in the paper). Fig.2(b,d) depict for the two datasets the mean of these norms over all convolutional layers as a function of epoch (similar to Fig.1(e)).

Figure 2: (a,c) The Norm of the convolutional layers once the factors of the subsequent Batch Normalization layers are absorbed, shown for CIFAR-10 and CIFAR-100 respectively. Each graph is a different epoch, see legend. Waving is due to the interleaving architecture of the convolutional layers. (b,d) Respectively for CIFAR-10 and CIFAR-100, the mean of the norm of the convolutional layers' weights per epoch.

25)Thm. 3 suggests that |λ l | will increase for layers l that do not have skip-connections. Conversely, if layer l has a parallel skip connection, then |λ

Table content: <table/>



Table content: <table><tr><td>: Notations</td></tr></table>

Peer review (iclr2017):
Summary: In this paper, the authors study ResNets through a theoretical formulation of a spin glass model. The conclusions are that ResNets behave as an ensemble of shallow networks at the start of training (by examining the magnitude of the weights for paths of a specific length) but this changes through training, through which the scaling parameter C (from assumption A4) increases, causing it to behave as an ensemble of deeper and deeper networks. Clarity: This paper was somewhat difficult to follow, being heavy in notation, with perhaps some notation overloading. A summary of some of the proofs in the main text might have been helpful. Specific Comments: - In the proof of Lemma 2, I*m not sure where the sequence beta comes from (I don*t see how it follows from 11?) - The ResNet structure used in the paper is somewhat different from normal with multiple layers being skipped? (Can the same analysis be used if only one layer is skipped? It seems like the skipping mostly affects the number of paths there are of a certain length?) - The new experiments supporting the scale increase in practice are interesting! I*m not sure about Theorems 3, 4 necessarily proving this link theoretically however, particularly given the simplifying assumption at the start of Section 4.2?