DIMINISHING BATCH NORMALIZATION
Year: 2018

Abstract:
In this paper, we propose a generalization of the BN algorithm, diminishing batch normalization (DBN), where we update the BN parameters in a diminishing moving average way. Batch normalization (BN) is very effective in accelerating the convergence of a neural network training phase that it has become a common practice. Our proposed DBN algorithm remains the overall structure of the original BN algorithm while introduces a weighted averaging update to some trainable parameters. We provide an analysis of the convergence of the DBN algorithm that converges to a stationary point with respect to trainable parameters. Our analysis can be easily generalized for original BN algorithm by setting some parameters to constant. To the best knowledge of authors, this analysis is the first of its kind for convergence with Batch Normalization introduced. We analyze a two-layer model with arbitrary activation function. The primary challenge of the analysis is the fact that some parameters are updated by gradient while others are not. The convergence analysis applies to any activation function that satisfies our common assumptions. For the analysis, we also show the sufficient and necessary conditions for the stepsizes and diminishing weights to ensure the convergence. In the numerical experiments, we use more complex models with more layers and ReLU activation. We observe that DBN outperforms the original BN algorithm on Imagenet, MNIST, NI and CIFAR-10 datasets with reasonable complex FNN and CNN models.

heading: 1 INTRODUCTION
text: Deep neural networks (DNN) have shown unprecedented success in various applications such as object detection. However, it still takes a long time to train a DNN until it converges. Ioffe & Szegedy identified a critical problem involved in training deep networks, internal covariate shift, and then proposed batch normalization (BN) to decrease this phenomenon. BN addresses this problem by normalizing the distribution of every hidden layer’s input. In order to do so, it calculates the preactivation mean and standard deviation using mini-batch statistics at each iteration of training and uses these estimates to normalize the input to the next layer. The output of a layer is normalized by using the batch statistics, and two new trainable parameters per neuron are introduced that capture the inverse operation. It is now a standard practice Bottou et al. (2016); He et al. (2016). While this approach leads to a significant performance jump, to the best of our knowledge, there is no known theoretical guarantee for the convergence of an algorithm with BN. The difficulty of analyzing the convergence of the BN algorithm comes from the fact that not all of the BN parameters are updated by gradients. Thus, it invalidates most of the classical studies of convergence for gradient methods.
In this paper, we propose a generalization of the BN algorithm, diminishing batch normalization (DBN), where we update the BN parameters in a diminishing moving average way. It essentially means that the BN layer adjusts its output according to all past mini-batches instead of only the current one. It helps to reduce the problem of the original BN that the output of a BN layer on a particular training pattern depends on the other patterns in the current mini-batch, which is pointed out by Bottou et al.. By setting the layer parameter we introduce into DBN to a specific value, we recover the original BN algorithm.
We give a convergence analysis of the algorithm with a two-layer batch-normalized neural network and diminishing stepsizes. We assume two layers (the generalization to multiple layers can be made by using the same approach but substantially complicating the notation) and an arbitrary loss function. The convergence analysis applies to any activation function that follows our common
assumption. The main result shows that under diminishing stepsizes on gradient updates and updates on mini-batch statistics, and standard Lipschitz conditions on loss functions DBN converges to a stationary point. As already pointed out the primary challenge is the fact that some trainable parameters are updated by gradient while others are updated by a minor recalculation.
Contributions. The main contribution of this paper is in providing a general convergence guarantee for DBN. Specifically, we make the following contributions.
• In section 4, we show the sufficient and necessary conditions for the stepsizes and diminishing weights to ensure the convergence of BN parameters.
• We show that the algorithm converges to a stationary point under a general nonconvex objective function.
This paper is organized as follows. In Section 2, we review the related works and the development of the BN algorithm. We formally state our model and algorithm in Section 3. We present our main results in Sections 4. In Section 5, we numerically show that the DBN algorithm outperforms the original BN algorithm. Proofs for main steps are collected in the Appendix.

heading: 2 LITERATURE REVIEW
text: Before the introduction of BN, it has long been known in the deep learning community that input whitening and decorrelation help to speed up the training process. In fact, Orr & Müller show that preprocessing the data by subtracting the mean, normalizing the variance, and decorrelating the input has various beneficial effects for back-propagation. Krizhevsky et al. propose a method called local response normalization which is inspired by computational neuroscience and acts as a form of lateral inhibition, i.e., the capacity of an excited neuron to reduce the activity of its neighbors. Gülçehre & Bengio propose a standardization layer that bears significant resemblance to batch normalization, except that the two methods are motivated by very different goals and perform different tasks.
Inspired by BN, several new works are taking BN as a basis for further improvements. Layer normalization Ba et al. (2016) is much like the BN except that it uses all of the summed inputs to compute the mean and variance instead of the mini-batch statistics. Besides, unlike BN, layer normalization performs precisely the same computation at training and test times. Normalization propagation that Arpit et al. uses data-independent estimations for the mean and standard deviation in every layer to reduce the internal covariate shift and make the estimation more accurate for the validation phase. Weight normalization also removes the dependencies between the examples in a minibatch so that it can be applied to recurrent models, reinforcement learning or generative models Salimans & Kingma (2016). Cooijmans et al. propose a new way to apply batch normalization to RNN and LSTM models.
Given all these flavors, the original BN method is the most popular technique and for this reason our choice of the analysis. To the best of our knowledge, we are not aware of any prior analysis of BN.
BN has the gradient and non-gradient updates. Thus, nonconvex convergence results do not immediately transfer. Our analysis explicitly considers the workings of BN. However, nonconvex convergence proofs are relevant since some small portions of our analysis rely on known proofs and approaches.
Neural nets are not convex, even if the loss function is convex. For classical convergence results with a nonconvex objective function and diminishing learning rate, we refer to survey papers Bertsekas (2011); Bertsekas & Tsitsiklis (2000); Bottou et al. (2016). Bertsekas & Tsitsiklis provide a convergence result with the deterministic gradient with errors. Bottou et al. provide a convergence result with the stochastic gradient. The classic analyses showing the norm of gradients of the objective function going to zero date back to Grippo (1994); Polyak & Tsypkin (1973); Polyak (1987). For strongly convex objective functions with a diminishing learning rate, we learn the classic convergence results from Bottou et al..

heading: 3 MODEL AND ALGORITHM
text: The optimization problem for a network is an objective function consisting of a large number of component functions, that reads:
min f̄(θ, λ) = N∑ i=1 fi(Xi : θ, λ),
subject to θ ∈ P, λ ∈ Q,
(1)
where fi : Rn1 × Rn2 → R, i = 1, ..., N , are real-valued functions for any data record Xi. Index i associates with data recordXi and target response yi (hidden behind the dependency of f on i) in the training set. Parameters θ include the common parameters updated by gradients directly associated with the loss function, i.e., behind the part that we have a parametric model, while BN parameters λ are introduced by the BN algorithm and not updated by gradient methods but by the mini-batch statistics. We define that the derivative of fi is always taken with respect to θ:
∇fi(Xi : θ, λ) := ∇θfi(Xi : θ, λ). (2)
The deep network we analyze has 2 fully-connected layers with D1 neurons each. The techniques presented can be extended to more layers with additional notation. Each hidden layer computes y = a(Wu) with activation function a(·) and u is the input vector of the layer. We do not need to include an intercept term since the BN algorithm automatically adjusts for it. BN is applied to the output of the first hidden layer.
We next describe the computation in each layer to show how we obtain the output of the network. The notations introduced here is used in the analysis. Figure 1 shows the full structure of the network. The input data is vector X , which is one of {Xi}Ni=1. Vector λ = ( (µj) D j=1, (σj) D j=1 ) is the
set of all BN parameters and vector θ = ( W1,W2, (β (1) j ) D j=1, (γ (1) j ) D j=1 ) is the set of all trainable parameters which are updated by gradients.
Matrices W1,W2 are the actual model parameters and β, γ are introduced by BN. The value of jth neuron of the first hidden layer is
z (1) j (X : θ) = a(W1,j,·X), (3)
where W1,j,· denotes the weights of the linear transformations for the jth neuron.
The jth entry of batch-normalized output of the first layer is
y (1) j (X : θ, λ) = γ (1) j ( z (1) j (X : θ)− µj σj + B ) + β (1) j ,
where β(1)j and γ (1) j are trainable parameters updated by gradient and µj and σj are batch normalization parameters for z(1)j . Trainable parameter µj is the mini-batch mean of z (1) j and trainable parameter σj is the mini-batch sample deviation of z (1) j . Constant B keeps the denominator from zero. The output of jth entry of the output layer is:
z (2) j (X : θ) = a ( W2,j,· [ γ (1) j ( z (1) j (X : θ)− µj σj + B ) + β (1) j ]) (4)
The objective function for the ith sample is
fi(Xi : θ, λ) = li (( z (2) j (Xi : θ, λ) ) j ) , (5)
where li(·) is the loss function associated with the target response yi. For sample i, we have the following complete expression for the objective function:
fi(Xi : θ, λ) = li a( D∑ j=1 W2,k,j [ γ (1) j a(W1,j,·Xi − µj) σj + B + β (1) j ] )k  . (6) Function fi(Xi : θ, λ) is nonconvex with respect to θ and λ.

heading: 3.1 ALGORITHM
text: Algorithm 1 shows the algorithm studied herein. There are two deviations from the standard BN algorithm, one of them actually being a generalization. We use the full gradient instead of the more popular stochastic gradient (SG) method. It essentially means that each batch contains the entire training set instead of a randomly chosen subset of the training set. An analysis of SG is potential future research. Although the primary motivation for full gradient update is to reduce the burdensome in showing the convergence, the full gradient method is similar to SG in the sense that both of them go through the entire training set, while full gradient goes through it deterministically and the SG goes through it in expectation. Therefore, it is reasonable to speculate that the SG method has similar convergence property as the full algorithm studied herein.
Algorithm 1 DBN: Diminishing Batch-Normalized Network Update Algorithm 1: Initialize θ ∈ Rn1 and λ ∈ Rn2 2: for iteration m=1,2,... do 3: θ(m+1) := θ(m) − η(m) ∑N i=1∇fi(Xi : θ(m), λ(m))
4: for j=1,...,D1 do 5: µ(m+1)j := 1 N ∑N i=1 z (1) j (Xi : θ (m+1))
6: σ(m+1)j := √ 1 N ∑N i=1 ( z (1) j (Xi : θ (m+1))− µ(m+1)j )2
7: λ(m+1) := α(m+1) (
(µ (m+1) j ) D1 j=1, (σ (m+1) j ) D1 j=1
) + (1− α(m+1))λ(m)
The second difference is that we update the BN parameters (θ, λ) by their moving averages with respect to diminishing α(m). The original BN algorithm can be recovered by setting α(m) = 1 for every m. After introducing diminishing α(m), λ(m) and hence the output of the BN layer is determined by the history of all past data records, instead of those solely in the last batch. Thus, the output of the BN layer becomes more general that better reflects the distribution of the entire dataset. We use two strategies to decide the values of α(m). One is to use a constant smaller than 1 for all m, and the other one is to decay the α(m) gradually, such as α(m) = 1/m.
In our numerical experiment, we show that Algorithm 1 outperforms the original BN algorithm, where both are based on SG and non-linear activation functions with many layers FNN and CNN models.

heading: 4 GENERAL CASE
text: The main purpose of our work is to show that Algorithm 1 converges. In the general case, we focus on the nonconvex objective function.

heading: 4.1 ASSUMPTIONS
text: Here are the assumptions we used for the convergence analysis.
Assumption 1 (Lipschitz continuity on θ and λ). For every i we have
‖∇fi(X : θ̃, λ)−∇fi(X : θ̂, λ)‖2 ≤ L̄‖θ̃ − θ̂‖2,∀θ̃, θ̂, λ,X. (7)
‖∇W1,j,·fi(X : θ̃, λ)−∇W1,j,·fi(X : θ̂, λ)‖2 ≤ L̄‖W̃1,j,· − Ŵ1,j,·‖2,∀λ, θ̃, θ̂, X, j ∈ {1, ..., D1}.
(8)
‖∇fi(X : θ, λ̃)−∇fi(X : θ, λ̂)‖2 ≤ L̄‖λ̃− λ̂‖2, ∀θ, λ̃, λ̂,X, j ∈ {1, ..., D1}.
(9)
Noted that the Lipschitz constants associated with each of the above inequalities are not necessarily the same. Here L̄ is an upper bound for these Lipschitz constants for simplicity.
Assumption 2 (bounded parameters). Sets P and Q are compact set, where θ ∈ P and λ ∈ Q. Thus, there exists a constant M that weights W and parameters λ are bounded element-wise by this constant M .
‖W1‖ M and ‖W2‖ M and ‖λ‖ M.
This also implies that the updated θ, λ in Algorithm 1 remain in P and Q, respectively.
Assumption 3 (diminishing update on θ). The stepsizes of θ update satisfy ∞∑ m=1 η(m) =∞ and ∞∑ m=1 (η(m))2 <∞. (10)
This is a common assumption for diminishing stepsizes in optimization problems.
Assumption 4 (Lipschitz continuity of li(·)). Assume the loss functions li(·) for every i is continuously differentiable. It implies that there exists M̂ such that
‖li(x)− li(y)‖ ≤ M̂‖x− y‖,∀x, y.
Assumption 5 (existence of a stationary point). There exists a stationary point (θ∗, λ∗) such that ‖∇f̄(θ∗, λ∗)‖ = 0.
We note that all these are standard assumptions in convergence proofs. We also stress that Assumption 4 does not directly imply 1. Since we assume that P and Q are compact, then Assumptions 1, 4 and 5 hold for many standard loss function such as softmax and MSE.
Assumption 6 (Lipschitz at activation function). The activation function a(·) is Lipschitz with constant k:
|a(x)| ≤ k‖x‖ (11)
Since for all activation function there is a(0) = 0, the condition is equivalent to |a(x) − a(0)| ≤ k‖x − 0‖. We note that this assumption works for many popular choices of activation functions, such as ReLU and LeakyReLu.

heading: 4.2 CONVERGENCE ANALYSIS
text: We first have the following lemma specifying sufficient conditions for λ to converge. Proofs for main steps are given in the Appendix.
Theorem 7 Under Assumptions 1, 2, 3 and 6, if {α(m)} satisfies ∞∑ m=1 α(m) <∞ and ∞∑ m=1 m∑ n=1 α(m)η(n) <∞,
then sequence {λ(m)} converges to λ̄.
We give a discussion of the above conditions for α(m) and η(m) at the end of this section. With the help of Theorem 7, we can show the following convergence result.
Lemma 8 Under Assumptions 4, 5 and the assumptions of Theorem 7, when
∞∑ m=1 ∞∑ i=m i∑ n=1 α(i)η(n) <∞ and ∞∑ m=1 ∞∑ n=m α(n) <∞, (12)
we have
lim sup M→∞ M∑ m=1 η(m)‖∇f̄(θ(m), λ̄)‖22 <∞. (13)
This result is similar to the classical convergence rate analysis for the non-convex objective function with diminishing stepsizes, which can be found in Bottou et al. (2016).
Lemma 9 Under the assumptions of Lemma 8, we have
lim inf m→∞
‖∇f̄(θ(m), λ̄)‖22 = 0. (14)
This theorem states that for the full gradient method with diminishing stepsizes the gradient norms cannot stay bounded away from zero. The following result characterizes more precisely the convergence property of Algorithm 1.
Lemma 10 Under the assumptions stated in Lemma 8, we have
lim m→∞
‖∇f̄(θ(m), λ̄)‖22 = 0. (15)
Our main result is listed next.
Theorem 11 Under the assumptions stated in Lemma 8, we have
lim m→∞
‖∇f̄(θ(m), λ(m))‖22 = 0. (16)
We cannot show that {θ(m)}’s converges (standard convergence proofs are also unable to show such a stronger statement). For this reason, Theorem 11 does not immediately follow from Lemma 10 together with Theorem 7. The statement of Theorem 11 would easily follow from Lemma 10 if the convergence of {θ(m)} is established and the gradient being continuous.
Considering the cases η(m) = O( 1 mk ) and α(m) = O( 1 mh ). We show in the Appendix that the set of sufficient and necessary conditions to satisfy the assumptions of Theorem 7 are h > 1 and k ≥ 1. The set of sufficient and necessary conditions to satisfy the assumptions of Lemma 8 are h > 2 and k ≥ 1. For example, we can pick η(m) = O( 1m ) and α
(m) = O( 1m2.001 ) to achieve the above convergence result in Theorem 11.

heading: 5 COMPUTATIONAL EXPERIMENTS
text: We conduct the computational experiments with Theano and Lasagne on a Linux server with a Nvidia Titan-X GPU. We use MNIST LeCun et al. (1998), CIFAR-10 Krizhevsky & Hinton (2009) and Network Intrusion (NI) kdd (1999) datasets to compare the performance between DBN and the original BN algorithm. For the MNIST dataset, we use a four-layer fully connected FNN (784 × 300× 300× 10) with the ReLU activation function and for the NI dataset, we use a four-layer fully connected FNN (784×50×50×10) with the ReLU activation function. For the CIFAR-10 dataset, we use a reasonably complex CNN network that has a structure of (Conv-Conv-MaxPool-DropoutConv-Conv-MaxPool-Dropout-FC-Dropout-FC), where all four convolution layers and the first fully connected layers are batch normalized. We use the softmax loss function and l2 regularization with for all three models. All the trainable parameters are randomly initialized before training. For all 3 datasets, we use the standard epoch/minibatch setting with the minibatch size of 100, i.e., we do not compute the full gradient and the statistics are over the minibatch. We use AdaGrad Duchi, John and Hazan, Elad and Singer (2011) to update the learning rates η(m) for trainable parameters, starting from 0.01.
We use two different strategies to decide the values of α(m) in DBN: constant values of α(m) and diminishing α(m) where α(m) = 1/m and α(m) = 1/m2. We test the choices of constant α(m) ∈ {1, 0.75, 0.5, 0.25, 0.1, 0.01, 0.001, 0}.
We test all the choices of α(m) with the performances presented in Figure 2. Figure 2 shows that all the non-zero choices of α(m) converge properly. The algorithms converge without much difference even when α(m) in DBN is very small, e.g., 1/m2. However, if we select α(m) = 0, the algorithm is erratic. Besides, we observe that all the non-zero choices of α(m) converge at a similar rate. The fact that DBN keeps the batch normalization layer stable with a very small α(m) suggests that the BN parameters do not have to be depended on the latest minibatch, i.e., the original BN.
We compare a selected set of the most efficient choices of α(m) in Figures 3 and 4. They show that DBN with α(m) < 1 is more stable than the original BN algorithm. The variances with respect to epochs of the DBN algorithm are smaller than those of the original BN algorithms in each figure.
Table 1 shows the best result obtained from each choice of α(m). Most importantly, it suggests that the choices of α(m) = 1/m and 1/m2 perform better than the original BN algorithm. Besides, all the constant less-than-one choices of α(m) perform better than the original BN, showing the importance of considering the mini-batch history for the update of the BN parameters. The BN algorithm in each figure converges to similar error rates on test datasets with different choices of α(m) except for the α(m) = 0 case. Among all the models we tested, α(m) = 0.25 is the only one that performs top 3 for all three datasets, thus the most robust choice.
To summarize, our numerical experiments show that the DBN algorithm outperforms the original BN algorithm on the MNIST, NI and CIFAT-10 datasets with typical deep FNN and CNN models.
Future Directions. On the analytical side, we believe an extension to more than 2 layers is doable with significant augmentations of the notation. A stochastic gradient version is likely to be much more challenging to analyze. A second open question concerns more general activation functions. It would be interesting to analyze other activation functions, such as Sigmoid, that do not apply to our current assumptions.

heading: 6 APPENDIX: PROOFS

heading: 6.1 PRELIMINARY RESULTS
text: The following proofs are shortened to corporate with AAAI submission page limit.
Proposition 12 There exists a constant M such that, for any θ and fixed λ, we have ‖∇f̄(θ, λ)‖22 ≤M. (17)
Proof. By Assumption 5, we know there exists (θ∗, λ∗) such that ‖∇f̄(θ∗, λ∗)‖2 = 0. Then we have ‖∇f̄(θ, λ)‖2
=‖∇f̄(θ, λ)‖2 − ‖∇f̄(θ, λ∗)‖2 + ‖∇f̄(θ, λ∗)‖2 − ‖∇f̄(θ∗, λ∗)‖2 ≤‖∇f̄(θ, λ)−∇f̄(θ, λ∗)‖2 + ‖∇f̄(θ, λ∗)−∇f̄(θ∗, λ∗)‖2 ≤ N∑ i=1 ‖∇fi(Xi : θ, λ)−∇fi(Xi : θ, λ∗)‖2
+ N∑ i=1 ‖∇fi(Xi : θ, λ∗)−∇fi(Xi : θ∗, λ∗)‖2
≤NL̄(‖λ− λ∗‖2 + ‖θ − θ∗‖2),
(18)
where the last inequality is by Assumption 1. We then have ‖∇f̄(θ, λ)‖22 ≤ N2L̄2(‖λ− λ∗‖2 + ‖θ − θ∗‖2)2 ≤M, (19) because sets P and Q are compact by Assumption 2.
Proposition 13 We have
fi(X : θ̃, λ) ≤ fi(X : θ̂, λ) +∇fi(X : θ̂, λ)T (θ̃ − θ̂) + 1
2 L̄‖θ̃ − θ̂‖22, ∀θ̃, θ̂, X. (20)
Proof. This is a known result of the Lipschitz-continuous condition that can be found in Bottou et al. (2016). We have this result together with Assumption 1.

heading: 6.2 PROOF OF THEOREM 7
text: Lemma 14 When ∑∞
m=1 α (m) <∞ and ∑∞ m=1 ∑m n=1 α (m)η(n) <∞,
µ̃ (m) j :=
µ (m) j
(1− α(1))(1− α(2))...(1− α(m)) is a Cauchy series.
Proof. By Algorithm 1, we have
µ (m) j = α
(m) 1
N N∑ i=1 a(W (m) 1,j,·Xi) + (1− α (m))µ (m−1) j . (21)
We define α̃(m) := α(m)
(1− α(1))(1− α(2))...(1− α(m)) and ∆W (m)1,j,· := W (m) 1,j,· −W (m−1) 1,j,· . After dividing
equation 21 by (1− α(1))(1− α(2))...(1− α(m)), we obtain
µ̃ (m) j = α̃
(m) 1
N N∑ i=1 a(W (m) 1,j,·Xi) + µ̃ (m−1) j . (22)
Then we have
|µ̃(m)j − µ̃ (m−1) j | ≤ α̃ (m)|k| 1 N N∑ i=1 | m∑ n=1 ∆W (n) 1,j,·Xi|
= α̃(m)|k| 1 N N∑ i=1 ∣∣∣∣∣ m∑ n=1 ( η(n) N∑ l=1 ∇W1,j,·fl(Xl : θ (n), λ(n)) ) ·Xi ∣∣∣∣∣ = α̃(m)|k| 1
N N∑ i=1 m∑ n=1
( η(n) ∣∣∣∣∣ ( N∑ l=1 ∇W1,j,·fl(Xl : θ (n), λ(n)) ) ·Xi ∣∣∣∣∣ )
≤ α̃(m)|k| 1 N N∑ i=1 m∑ n=1
( η(n)‖
N∑ l=1 ∇W1,j,·fl(Xl : θ (n), λ(n))‖ · ‖Xi‖
) (23)
≤ α̃(m)|k| N∑ i=1 m∑ n=1 η(n) ( L̄ · (‖W (n)1,j,· −W ∗ 1,j,·‖2 + ‖λ (n) j,· − λ ∗ j,·‖2) · ‖Xi‖2 ) (24)
≤ α̃(m) m∑
n=1
( η(n) ) |k| N∑ i=1 ( 2L̄M‖Xi‖2 ) (25)
≤ α̃(m) m∑
n=1
η(n)M̃L̄,M . (26)
Equation equation 6.2 is due to W (m)1,i,j = ∑m n=1 ∆W (n) 1,i,j .
Therefore,
|µ̃(p)j − µ̃ (q) j | ≤ M̃L̄,M · q∑ m=p m∑ n=1 α̃(m)η(n). (27)
It remains to show that ∞∑
m=1
α(m) <∞, (28)
∞∑ m=1 m∑ n=1 α(m)η(n) <∞, (29)
implies the convergence of {µ̃(m)}. By (28), we have Π∞m=1(1 − α(m)) > 0, since ln(Π∞m=1(1 − α(m))) >∑∞ m=1−α (m) > −∞.
It is also easy to show that there exists C and Mc such that for all m ≥Mc, we have
(1− α(1))(1− α(2)) . . . (1− α(m)) ≥ C. (30)
Therefore, lim m→∞
(1− α(1))(1− α(2)) . . . (1− α(m)) ≥ C.
Thus the following holds:
α̃(m) ≤ 1 C α(m) (31)
and q∑
m=p m∑ n=1 α̃(m)η(n) ≤ 1 C q∑ m=p m∑ n=1 α(m)η(n). (32)
From equation 29 and equation 32 it follows that the sequence {µ̃(m)j } is a Cauchy series.
Lemma 15 Since {µ̃(m)j } is a Cauchy series, {µ (m) j } is a Cauchy series.
Proof. We know that µ(m)j = µ̃ (m) j (1−α (1))...(1−α(m)). Since lim m→∞ µ̃ (m) j → µ̃j and lim m→∞ (1−α(1))...(1− α(m))→ C̃, we have lim m→∞ µ (m) j → µ̃j · C̃. Thus µ (m) j is a Cauchy series. Lemma 16 If ∑∞
m=1 α (m) <∞ and ∑∞ m=1 ∑m n=1 α (m)η(n) <∞, {σ(m)j } is a Cauchy series.
Proof. We define σ(m)j := σ̃ (m) j (1− α (1))...(1− α(m)). Then we have
|σ̃(m+1)j − σ̃ (m) j | = α̃ (m) √√√√ 1 N N∑ i=1 ( a(W (m) 1,j,·Xi)− µ (m) j )2
= α̃(m) |k|√ N √√√√ N∑ i=1 ( a(W (m) 1,j,·Xi) k − µ (m) j k )2 . (33)
Since {µ(m)j } is convergent, there exists c1, c2 and N1 such that for any m > N1, −∞ < c1 < µ (m) j < c2 < ∞. For any C̄ ∈ {c1 k , c2 k } , we have
|σ̃(m+1)j − σ̃ (m) j | ≤ α̃ (m) |k|√ N · √√√√ N∑ i=1 ( a(W (m) 1,j,·Xi) k − C̄ )2 (34)
≤ α̃(m) |k|√ N · √√√√ N∑ i=1 ( | a(W (m) 1,j,·Xi) k |+ |C̄| )2 (35)
≤ α̃(m) |k|√ N · √√√√ N∑ i=1 ( m∑ n=1 η(n) ( 2NL̄M‖Xi‖2 ) + |C̄| )2 (36)
≤ α̃(m) |k|√ N · √√√√N ·(M̃L̄,M m∑ n=1 η(n) + |C̄| )2 (37)
= α̃(m)|k| · ( M̃L̄,M
m∑ n=1 η(n) + |C̄|
) . (38)
Inequality equation 35 is by the following fact:√√√√ n∑ i=1 (ai − c)2 ≤ √√√√ n∑ i=1 (|ai|+ |c|)2, (39)
where b and ai for every i are arbitrary real scalars. Besides, equation 39 is due to −2aic ≤ max{−2|ai|c, 2|ai|c}.
Inequality equation 36 follow from the square function being increasing for nonnegative numbers. Besides these facts, equation 36 is also by the same techniques we used in equation 23-equation 25 where we bound the derivatives with the Lipschitz continuity in the following inequality:
‖ N∑ l=1 ∇W1,j,·fl(Xl : θ (n), λ(n))‖ ≤ 2NL̄M. (40)
Inequality equation 37 is by collecting the bounded terms into a single bound M̃L̄,M . Therefore,
|σ̃(q)j − σ̃ (p) j | ≤ q−1∑ m=p α̃(m)|k| ·
( M̃L̄,M
m∑ n=1 η(n) + |C̄|
) . (41)
Using the similar methods in deriving equation 28 and equation 29, it can be seen that a set of sufficient conditions ensuring the convergence for {σ̃(m)j } is: ∑∞ m=1 α (m) <∞ and ∑∞ m=1 ∑m n=1 α (m)η(n) <∞.
Therefore, the convergence conditions for {σ(m)j } are the same as for {µ (m) j }.
It is clear that these lemmas establish the proof of Theorem 7.

heading: 6.3 CONSEQUENCES OF THEOREM 7
text: Proposition 17 Under the assumptions of Theorem 7, we have |λ(m) − λ̄|∞ ≤ am, where
am = M1 ∞∑ i=m i∑ j=1 α(i)η(j) +M2 ∞∑ i=m α(i). (42)
M1 and M2 are constants.
Proof. For the upper bound of σ(m)j , by equation 38, we have
|σ̃(q)j − σ̃ (p) j | ≤ q−1∑ m=p α̃(m)|k|
( M̃L̄,M
m∑ n=1 η(n) + |C̄|
) .
We define σ̃j := σ̄j
(1− α(1))...(1− α(u))... . Therefore,
|σ̃j − σ̃(m)j | ≤ ∞∑
i=m
α̃(i)|k| ( M̃L̄,M
i∑ j=1 η(j) + |C̄|
)
≤ |k| C ∞∑ i=m
α(i) ( M̃L̄,M
i∑ j=1 η(j) + |C̄|
) .
(43)
The first inequality comes by substituting p by m and by taking lim as q → ∞ in equation 41. The second inequality comes from equation 30. We then obtain,∣∣∣σ(m)j − σ̄j ∣∣∣ ≤ ∣∣∣σ̃(m)j − σ̃(∞)j ∣∣∣+ ∣∣∣∣ σ̄j(1− α(1))...(1− α(m)) − σ̃(∞)j ∣∣∣∣ = ∣∣∣σ̃(m)j − σ̃(∞)j ∣∣∣+ ∣∣∣∣ σ̄j(1− α(1))...(1− α(m)) − σ̄j(1− α(1))...(1− α(u))...
∣∣∣∣ = ∣∣∣σ̃(m)j − σ̃(∞)j ∣∣∣+ σ̄j ∣∣∣∣ (1− α(m+1))...(1− α(u))...− 1(1− α(1))...(1− α(u))...
∣∣∣∣ ≤ ∣∣∣σ̃(m)j − σ̃(∞)j ∣∣∣+ σ̄jC |1− (1− α(m+1))...(1− α(u))...| ≤ ∣∣∣σ̃(m)j − σ̃(∞)j ∣∣∣+ σ̄jC ∞∑ n=m+1 α(n).
(44)
The second inequality is by (1 − α(1))...(1 − α(m)) < 1, the third inequality is by equation 30 and the last inequality can be easily seen by induction. By equation 44, we obtain
|σ̄j − σ(m)j | = lim M→∞ |σ(M)j − σ (m) j | ≤ |σ̃j − σ̃ (m) j |+ σ̄j C ∞∑ n=m+1 α(n). (45)
Therefore, we have
|σ̄j − σ(m)j | ≤ |σ̃j − σ̃ (m) j |+ σ̄j C ∞∑ n=m+1 α(n)
≤ ∞∑
i=m
α̃(i)|k| · ( M̃L̄,M
i∑ j=1 η(j) + |C̄| ) + σ̄j C ∞∑ i=m+1 α(i)
≤ ∞∑
i=m
1 C α(i)|k| ·
( M̃L̄,M
i∑ j=1 η(j) + |C̄| ) + σ̄j C ∞∑ i=m+1 α(i)
≤ M̃L̄,M |k|
C
∞∑ i=m i∑ j=1 α(i)η(j) + ( σ̄j C + |k||C̄| C ) ∞∑ i=m α(i).
(46)
The first inequality is by equation 45, the second inequality is by equation 41, the third inequality is by equation 31 and the fourth inequality is by adding the nonnegative term σ̄j C α(m) to the right-hand side.
For the upper bound of µ(m)j , we have∣∣∣µ(m)j − µ̄j ∣∣∣ ≤ ∣∣∣µ̃(m) − µ̃(∞)∣∣∣+ ∣∣∣∣ µ̄j(1− α(1))...(1− α(m)) − µ̃(∞) ∣∣∣∣ . (47)
Let us define Am := ∣∣∣µ̃(m) − µ̃(∞)∣∣∣ and Bm := ∣∣∣∣ µ̄j(1− α(1))...(1− α(m)) − µ̃(∞) ∣∣∣∣. Recall from Theorem 7 that {µ(m)j } is a Cauchy series, by equation 27, |µ̃ (p) j − µ̃ (q) j | ≤ M̄L̄,M · ∑q m=p ∑m n=1 α
(m)η(n). Therefore, the first term in equation 47 is bounded by
|µ̃(m)j − µ̃ ∞ j | ≤ M̃L̄,M · ∞∑ i=m i∑ n=1 α(i)η(n) <∞. (48)
For the second term in equation 47, recall that C := (1 − α(1))...(1 − α(u)).... Then we have C · ∣∣∣∣ µ̄j(1− α(1))...(1− α(m)) − µ̃(∞) ∣∣∣∣ ≤ µ̄j ∞∑ i=m+1 α(i), where the inequality can be easily seen by induction. Therefore, the second term in equation 47 is bounded by∣∣∣∣ µ̄j(1− α(1))...(1− α(m)) − µ̃(∞) ∣∣∣∣ ≤ µ̄jC ∞∑ i=m+1 α(i). (49) From these we obtain ∣∣∣µ(m)j − µ̄j ∣∣∣ ≤ M̃L̄,M ∞∑ i=m i∑ n=1 α(i)η(n) + µ̄j C ∞∑ i=m+1 α(i). (50) The first inequality is by equation 47 and the second inequality is by equation 48 and equation 49. Combining equation 46 and equation 50, we have that
|λ(m) − λ̄|∞ ≤M1 ∞∑
i=m i∑ j=1 α(i)η(j) +M2 ∞∑ i=m α(i),
where M1 and M2 are constants defined as M1 = max( M̃L̄,M |k|
C , M̄L̄,M ) and M2 =
max( σ̄j + |k||C̄| C , µ̄j C ).
Proposition 18 Under the assumptions of Theorem 7, −∇f̄(θ(m), λ̄)T · ∇f̄(θ(m), λ(m)) ≤ −‖∇f̄(θ(m), λ̄)‖2 + L̄M √ n2am, where am is defined in Proposition 17.
Proof. For simplicity of the proof, let us define x(m) := ∇f̄(θ(m), λ̄), y(m) := ∇f̄(θ(m), λ(m)). We have |x(m) − y(m)|∞ ≤ L̄ √ n2‖λ(m) − λ̄‖∞ ≤ L̄ √ n2am, (51) where √ n2 is the dimension of λ. The second inequality is by Assumption 1 and the fourth inequality is by Proposition 17. Inequality equation 51 implies that for all m and i, we have |x(m)i − y (m) i | ≤ L̄ √ n2am. It remains to show − ∑ i y (m) i x (m) i ≤ − ∑ i x (m) i 2 + L̄M √ n2am, ∀i,m. (52) This is established by the following four cases.
1) If x(m)i ≥ 0, x (m) i −y (m) i ≥ 0, then x (m) i ≤ L̄ √ n2am +y (m) i . Thus−x (m) i y (m) i ≤ −x (m) i 2 + L̄M √ n2am by Proposition 12.
2) If x(m)i ≥ 0, x (m) i − y (m) i ≤ 0, then x (m) i ≤ y (m) i , x (m) i 2 ≤ x(m)i · y (m) i and −x (m) i y (m) i ≤ −x (m) i 2 .
3) If x(m)i < 0, x (m) i − y (m) i ≥ 0, then x (m) i ≥ y (m) i , x (m) i 2 ≤ x(m)i · y (m) i and −x (m) i y (m) i ≤ −x (m) i 2 .
4) If x(m)i < 0, x (m) i − y (m) i ≤ 0, then y (m) i − x (m) i ≤ L̄ √ n2am, y (m) i x (m) i − x (m) i 2 ≥ L̄√n2amx(m)i and −y(m)i x (m) i ≤ −x (m) i 2 − L̄√n2amx(m)i ≤ −x (m) i 2 + L̄M √ n2am. The last inequality is by Proposition 12.
All these four cases yield equation 52.
Proposition 19 Under the assumptions of Theorem 7, we have
f̄(θ(m+1), λ̄) ≤f̄(θ(m), λ̄)− η(m)‖∇f̄(θ(m), λ̄)‖22
+ η(m)L̄M √ n2am + 1
2 (η(m))2 ·NL̄M,
(53)
where M is a constant and am is defined in Proposition 17.
Proof. By Proposition 13,
fi(Xi : θ̃, λ) ≤ fi(Xi : θ̂, λ) +∇fi(Xi : θ̂, λ)T (θ̃ − θ̂) + 1
2 L̄‖θ̃ − θ̂‖22.
Therefore, we can sum it over the entire training set from i = 1 to N to obtain
f̄(θ̃, λ) ≤ f̄(θ̂, λ) +∇f̄(θ̂, λ)T (θ̃ − θ̂) + N 2 L̄‖θ̃ − θ̂‖22. (54)
In Algorithm 1, we define the update of θ in the following full gradient way:
θ(m+1) := θ(m) − η(m) · N∑ i=1 ·∇fi(Xi : θ(m), λ(m)), (55)
which implies θ(m+1) − θ(m) = −η(m) · ∇f̄(θ(m), λ(m)). (56)
By equation 56 we have θ̃ − θ̂ = θ(m+1) − θ(m) = −η(m)∇f̄(θ(m), λ(m)). We now substitute θ̃ := θ(m+1), θ̂ := θ(m) and λ := λ̄ into equation 54 to obtain
f̄(θ(m+1), λ̄)
≤f̄(θ(m), λ̄)− η(m)∇f̄(θ(m), λ̄)T∇f̄(θ(m), λ(m)) + (η(m))2 · NL̄M 2 ≤f̄(θ(m), λ̄)− η(m)‖∇f̄(θ(m), λ̄)‖22 + η(m)L̄M √ n2am
+ 1
2 (η(m))2 ·NL̄M.
(57)
The first inequality is by plugging equation 56 into equation 54, the second inequality comes from Proposition 12 and the third inequality comes from Proposition 18.

heading: 6.4 PROOF OF THEOREM 11
text: Here we show Theorem 11 as the consequence of Theorem 7 and Lemmas 8, 9 and 10.

heading: 6.4.1 PROOF OF LEMMA 8
text: Here we show Lemma 8 as the consequence of Lemmas 20, 21 and 22. Lemma 20 ∑∞
m=1 ∑∞ i=m ∑i n=1 α (i)η(n) <∞ and ∑∞ m=1 ∑∞ n=m α
(n) <∞ is a set of sufficient condition to ensure
∞∑ m=1 |σ̄j − σ(m)j | <∞,∀j. (58)
Proof. By plugging equation 45 and equation 43 into equation 58, we have the following for all j: ∞∑
m=1 ∣∣∣σ̄j − σ(m)j ∣∣∣ ≤ ∞∑ m=1 ( |σ̃j − σ̃(m)j |+ σ̄j C ∞∑ n=m+1 α(n) )
≤ |k| · M̃L̄,M
C
∞∑ m=1 ∞∑ i=m α(i) i∑ j=1 η(j) + σ̄j + |k||C̄| C ∞∑ m=1 ∞∑ n=m+1 α(n).
(59)
It is easy to see that the the following conditions are sufficient for right-hand side of equation 59 to be finite:∑∞ m=1 ∑∞ i=m ∑i n=1 α (i)η(n) <∞ and ∑∞ m=1 ∑∞ n=m α (n) <∞. Therefore, we obtain ∞∑
m=1
|σ̄j − σ(m)j | <∞, ∀j.
Lemma 21 Under Assumption 4, ∞∑
m=1 ∞∑ i=m i∑ n=1 α(i)η(n) <∞ and ∞∑ m=1 ∞∑ n=m α(n) <∞
is a set of sufficient conditions to ensure
lim sup M→∞ M∑ m=1 ∣∣∣f̄(θ(m), λ(m))− f̄(θ(m), λ̄)∣∣∣ <∞. Proof. By Assumption 4, we have
‖li(x)− li(y)‖ ≤ M̂‖x− y‖ ≤ M̂ D∑ i=1 |xi − yi|. (60)
By the definition of fi(·), we then have ∞∑
m=1 ∣∣∣f̄(θ(m), λ(m))− f̄(θ(m), λ̄)∣∣∣ (61) ≤ ∞∑
m=1 N∑ i=1 ∣∣∣(li(Xi : θ(m), λ(m))− li(Xi : θ(m), λ̄))∣∣∣ (62) ≤M2
∞∑ m=1 D∑ j=1 N∑ i=1 ∣∣∣∣∣a(W (m) 1,j,·Xi)− µ (m) j σ (m) j + B − a(W (m) 1,j,·Xi)− µ̄j σ̄j + B ∣∣∣∣∣ (63) ≤M3
∞∑ m=1 D∑ j=1 ( N∑ i=1 |k||W (m)1,j,·Xi| ∣∣∣∣∣ σ̄j − σ (m) j 2B ∣∣∣∣∣+N ∣∣∣∣∣ µ̄jσ̄j + B − µ (m) j σ (m) j + B ∣∣∣∣∣ ) . (64)
The first inequality is by the Cauchy-Schwarz inequality, and the second one is by equation 60. To show the finiteness of equation 64, we only need to show the following two statements:
∞∑ m=1 N∑ i=1 |k||W (m)1,j,·Xi| ∣∣∣∣∣ σ̄j − σ (m) j 2B ∣∣∣∣∣ <∞,∀j (65) and
∞∑ m=1 ∣∣∣∣∣ µ̄jσ̄j + B − µ (m) j σ (m) j + B ∣∣∣∣∣ <∞, ∀j. (66) Proof of equation 65: For all j we have
∞∑ m=1 N∑ i=1 |k||W (m)1,j,·Xi| ∣∣∣∣∣ σ̄j − σ (m) j 2B ∣∣∣∣∣ ≤ ∞∑
m=1
|k|NDM max i ‖Xi‖
1 2B ∣∣∣σ̄j − σ(m)j ∣∣∣ =|k|NDM max
i ‖Xi‖
1 2B ∞∑ m=1
∣∣∣σ̄j − σ(m)j ∣∣∣ . (67)
The inequality comes from |W (m)1,j,·Xi| ≤ DM‖Xi‖2, where D is the dimension of Xi and M is the elementwise upper bound for W (m)1,j,· in Assumption 2.
Finally, we invoke Lemma 14 to assert that ∑∞
m=1 ∣∣∣σ̄j − σ(m)j ∣∣∣ is finite. Proof of equation 66: For all j we have
∞∑ m=1 ∣∣∣∣∣ µ̄jσ̄j + B − µ (m) j σ (m) j + B ∣∣∣∣∣ ≤ ∞∑
m=1
∣∣∣∣∣ µ̄jσ̄j + B − µ (m) j σ̄j + B ∣∣∣∣∣+ ∞∑
m=1
∣∣∣∣∣ µ (m) j σ̄j + B − µ (m) j
σ (m) j + B
∣∣∣∣∣ . (68)
The first term in equation 68 is finite since {µ(m)j } is a Cauchy series. For the second term, we know that there exists a constant M such that for all m ≥ M , µ(m)j ≤ µ̄+ 1. This is also by the fact that {µ (m) j } is a Cauchy series and it converges to µ̄. Therefore, the second term in equation 68 becomes M−1∑ m=1 ∣∣∣∣∣ µ (m) j σ̄j + B − µ (m) j σ (m) j + B ∣∣∣∣∣+ ∞∑ m=M ∣∣∣∣∣ µ (m) j σ̄j + B − µ (m) j σ (m) j + B
∣∣∣∣∣ ≤
M−1∑ m=1 ∣∣∣∣∣ µ (m) j σ̄j + B − µ (m) j σ (m) j + B ∣∣∣∣∣+ ∞∑ m=M (µ̄+ 1) ∣∣∣∣∣ 1σ̄j + B − 1σ(m)j + B ∣∣∣∣∣ .
(69)
Noted that function f(σ) = 1
σ + B is Lipschitz continuous since its gradient is bounded by
1 2B . Therefore
we can choose 1
2B as the Lipschitz constant for f(σ). We then have the following inequality:
∣∣∣∣∣ 1σ̄j + B − 1σ(m)j + B ∣∣∣∣∣ ≤ 1 2B |σ̄j − σ(m)j |. (70)
Plugging equation 70 into equation 69, we obtain M−1∑ m=1 ∣∣∣∣∣ µ (m) j σ̄j + B − µ (m) j σ (m) j + B ∣∣∣∣∣+ ∞∑ m=M (µ̄+ 1) ∣∣∣∣∣ 1σ̄j + B − 1σ(m)j + B ∣∣∣∣∣
≤ M−1∑ m=1 ∣∣∣∣∣ µ (m) j σ̄j + B − µ (m) j σ (m) j + B ∣∣∣∣∣+ ∞∑ m=M (µ̄+ 1) 2B |σ̄j − σ(m)j |,
(71)
where the first term is finite by the fact that M is a finite constant. We have shown the condition for the second term to be finite in Lemma 20. Therefore,
∞∑ m=1 ∣∣∣∣∣ µ̄jσ̄j + B − µ (m) j σ (m) j + B ∣∣∣∣∣ <∞, ∀j. By equation 65 and equation 66, we have that the right-hand side of equation 64 is finite. It means that the left-hand side of equation 64 is finite. Thus,
∞∑ m=1 ∣∣∣f̄(θ(m), λ(m))− f̄(θ(m), λ̄)∣∣∣ <∞. Lemma 22 If
∞∑ m=1 ∞∑ i=m i∑ n=1 α(i)η(n) <∞ and ∞∑ m=1 ∞∑ n=m α(n) <∞,
then
lim sup M→∞ M∑ m=1 η(m)‖∇f̄(θ(m), λ̄)‖22 <∞.
Proof. For simplicity of the proof, we define
T (M) := M∑ m=1 η(m)‖∇f̄(θ(m), λ̄)‖22,
O(m) := f̄(θ(m+1), λ(m+1))− f̄(θ(m), λ(m)),
∆ (m+1) 1 := f̄(θ (m+1), λ(m+1))− f̄(θ(m+1), λ̄),
∆ (m) 2 := f̄(θ (m+1), λ̄)− f̄(θ(m), λ̄),
where λ̄ is the converged value of λ in Theorem 7. Therefore,
O(m) = ∆ (m+1) 1 + ∆ (m) 1 + ∆ (m) 2 ≤ |∆ (m+1) 1 |+ |∆ (m) 1 |+ ∆ (m) 2 . (72)
By Proposition 19,
∆ (m) 2 ≤ −η (m)‖∇f̄(θ(m), λ̄)‖22 + η(m)L̄M √ n2am + 1
2 (η(m))2 ·NL̄M. (73)
We sum the inequality equation 72 from 1 to K with respect to m and plug equation 73 into it to obtain K∑
m=1
O(m) ≤ K∑
m=1
|∆(m+1)1 |+ K∑
m=1
|∆(m)1 | − K∑
m=1
{η(m)‖∇f̄(θ(m), λ̄)‖22}
+ K∑ m=1 η(m)L̄M √ n2am + K∑ m=1 {1 2 (η(m))2NL̄M}
= K∑ m=1 |∆(m+1)1 |+ K∑ m=1 |∆(m)1 | − T (K)
+ L̄2 √ n2 · K∑ m=1 η(m)am + K∑ m=1 {1 2 (η(m))2NL̄M}.
(74)
From this, we have:
lim sup K→∞ T (K) ≤ lim sup K→∞ −1 c1 (f̄(θ(K), λ(K))− f̄(θ(1), λ(1)))
+ lim sup K→∞
1
c1 K∑ m=1 (|∆(m+1)1 |+ |∆ (m) 1 |)
+ lim sup K→∞
L̄2 √ n2 K∑ m=1 η(m)am
+ lim sup K→∞
NL̄K
2c1 K∑ m=1 η(m) 2 .
(75)
Next we show that each of the four terms in the right-hand side of equation 75 is finite, respectively. For the first term,
lim sup K→∞ −1 c1 (f̄(θ(K), λ(K))− f̄(θ(1), λ(1))) <∞ (76)
is by the fact that the parameters {θ, λ} are in compact sets, which implies that the image of fi(·) is in a bounded set.
For the second term, we showed its finiteness in Lemma 21.
For the third term, by equation 42, we have
lim sup K→∞ K∑ m=1 η(m)am
= lim sup K→∞ K∑ m=1
η(m) ( K1
∞∑ i=m i∑ j=1 α(i)η(j) +K2 ∞∑ i=m
α(i) )
=K1 lim sup K→∞ K∑ m=1
η(m) ( ∞∑
i=m i∑ j=1
α(i)η(j) )
+K2 lim sup K→∞ K∑ m=1 η(m) ∞∑ i=m α(i).
(77)
The right-hand side of equation 77 is finite because
∞∑ m=1
η(m) ( ∞∑
i=m i∑ j=1
α(i)η(j) ) <
∞∑ m=1
( ∞∑
i=m i∑ j=1
α(i)η(j) ) <∞ (78)
and ∞∑
m=1
η(m) ∞∑
i=m
α(i) < ∞∑ m=1 ∞∑ i=m α(i) <∞. (79)
The second inequalities in equation 78 and equation 79 come from the stated assumptions of this lemma.
For the fourth term,
lim sup K→∞
NL̄M
2c K∑ m=1 η(m) 2 <∞ (80)
holds, because we have ∑∞
m=1(η (m))2 < ∞ in Assumption 3. Therefore, T (∞) =∑∞
m=1 η (m)‖∇f̄(θ(m), λ̄)‖22 <∞ holds.
In Lemmas 20, 21 and 22, we show that {σ(m)} and {µ(m)} are Cauchy series, hence Lemma 8 holds.

heading: 6.4.2 PROOF OF LEMMA 9
text: This proof is similar to the the proof by Bertsekas & Tsitsiklis (2000).
Proof. By Theorem 8, we have
lim sup M→∞ M∑ m=1 η(m)‖∇f̄(θ(m), λ̄)‖22 <∞. (81)
If there exists a > 0 and an integer m̄ such that
‖∇f̄(θ(m), λ̄)‖2 ≥
for all m ≥ m̄, we would have
lim inf M→∞ M∑ m=m̄ η(m)‖∇f̄(θ(m), λ̄)‖22 ≥ lim inf M→∞ 2 M∑ m=m̄ η(m) =∞ (82)
which contradicts equation 81. Therefore, lim inf m→∞ ‖∇f̄(θ(m), λ̄)‖2 = 0.

heading: 6.4.3 PROOF OF LEMMA 10
text: Lemma 23 Let Yt,W, t and Zt be three sequences such that Wt is nonnegative for all t. Assume that Yt+1 ≤ Yt −Wt + Zt, t = 0, 1, ..., (83)
and that the series ∑T
t=0 Zt converges as T →∞. Then either Yt →∞ or else Yt converges to a finite value and ∑∞ t=0 Wt <∞.
This lemma has been proven by Bertsekas & Tsitsiklis (2000).
Lemma 24 When ∞∑
m=1 ∞∑ i=m i∑ n=1 α(i)η(n) <∞ and ∞∑ m=1 ∞∑ n=m α(n) <∞, (84)
it follows that f̄(θ(m), λ̄) converge to a finite value.
Proof. By Proposition 19, we have
f̄(θ(m+1), λ̄) ≤f̄(θ(m), λ̄)− η(m)‖∇f̄(θ(m), λ̄)‖22
+ η(m)L̄M √ n2am + 1
2 (η(m))2 ·NL̄M.
(85)
Let Y (m) := f̄(θ(m), λ̄),W (m) := η(m)‖∇f̄(θ(m), λ̄)‖22 andZ(m) := η(m)L̄M √ n2am+ 1
2 (η(m))2 ·NL̄M . By equation 10 and equation 77- equation 79, it is easy to see that ∑M
m=0 Z (m) converges as M → ∞.
Therefore, by Lemma 23, Y (m) converges to a finite value. The infinite case can not occur in our setting due to Assumptions 1 and 2.
Lemma 25 If∑∞ m=1 ∑∞ i=m ∑i n=1 α (i)η(n) <∞ and ∑∞ m=1 ∑∞ n=m α (n) <∞,
then lim m→∞ ‖∇f̄(θ(m), λ̄)‖2 = 0.
Proof. To show that lim m→∞ ‖∇f̄(θ(m), λ̄)‖2 = 0, assume the contrary; that is,
lim sup m→∞
‖∇f̄(θ(m), λ̄)‖2 > 0.
Then there exists an > 0 such that ‖∇f̄(θ(m), λ̄)‖ < /2 for infinitely manym and also ‖∇f̄(θ(m), λ̄)‖ > for infinitely many m. Therefore, there is an infinite subset of integers M, such that for each m ∈ M, there exists an integer q(m) > m such that
‖∇f̄(θ(m), λ̄)‖ < /2,
‖∇f̄(θ(i(m)), λ̄)‖ > ,
/2 ≤ ‖∇f̄(θ(i), λ̄)‖ ≤ , if m < i < q(m).
(86)
From ‖∇f̄(θ(m+1), λ̄)‖ − ‖∇f̄(θ(m), λ̄)‖ ≤ L̄η(m)‖∇f̄(θ(m), λ(m))‖, it follows that for allm ∈ M that are sufficiently large so that L̄η(m) < /4, we have
/4 ≤ ‖∇f̄(θ(m), λ(m))‖. (87)
Otherwise the condition /2 ≤ ‖∇f̄(θ(m+1), λ̄)‖ would be violated. Without loss of generality, we assume that the above relations as well as equation 57 hold for all m ∈ M. With the above observations, we have for all m ∈ M,
2 ≤ ‖∇f̄(θq(m), λ̄)‖ − ‖∇f̄(θ(m), λ̄)‖ ≤ L̄‖θq(m) − θ(m)‖
≤ L̄ q(m)−1∑ i=m η(i)(‖∇f̄(θ(i), λ̄)‖+ ‖∇f̄(θ(i), λ(i))−∇f̄(θ(i), λ̄)‖)
= L̄ q(m)−1∑ i=m η(i) + L̄2 √ n2M1 q(m)−1∑ i=m η(i) ∞∑ j=m j∑ k=1 α(j)η(k)
+ L̄2 √ n2M2 q(m)−1∑ i=m η(i) ∞∑ j=m α(j)
(88)
The first inequality is by equation 86 and the third one is by the Lipschitz condition assumption. The seventh one is by equation 51. By equation 12, we have for all m ∈ M,
q(m)−1∑ i=m η(i) ∞∑ j=m j∑ k=1 α(j)η(k) < ∞∑ i=1 ∞∑ j=i j∑ k=1 α(j)η(k) <∞ (89)
and q(m)−1∑ i=m η(i) ∞∑ j=m α(j) < ∞∑ i=1 ∞∑ j=i α(j) <∞. (90)
It is easy to see that for any sequence {αi} with ∑∞ i=1 αi <∞, if follows that lim infM→∞ ∑∞ i=M αi = 0. Therefore, lim inf m→∞ ∑q(m)−1 i=m η (i)∑∞ j=m ∑j k=1 α (j)η(k) = 0 and lim inf m→∞ ∑q(m)−1 i=m η (i)∑∞ j=m α (j) = 0. From this it follows that
lim inf m→∞ q(m)−1∑ i=m η(i) ≥ 1 2L̄ . (91)
By equation 51 and equation 87, if we pick m ∈ M such that L√n2am ≤
8 , we have ‖∇f̄(θ(m), λ̄)‖ ≥ 8 .
Using equation 57, we observe that f̄(θq(m), λ̄)
≤ f̄(θ(m), λ̄)− c1 (
8 )2 q(m)−1∑ i=m η(i) + 1 2 ·NL̄M q(m)−1∑ i=m (η(i))2, ∀m ∈ M, (92)
where the second inequality is by equation 87. By Lemma 24, f̄(θq(m), λ̄) and f̄(θ(m), λ̄) converge to the same finite value. Using this convergence result and the assumption ∑∞ m=0(η
(m))2 <∞, this relation implies that
lim sup m→∞,m∈M
∑q(m)−1 i=m η (i) = 0 and contradicts equation 91.
By Lemmas 23, 24 and 25, we show that Theorem 11 holds.

heading: 6.5 DISCUSSIONS OF CONDITIONS FOR STEPSIZES
text: Here we discuss the actual conditions for η(m) and α(m) to satisfy the assumptions of Theorem 7 and Lemma 8. We only consider the cases η(m) = 1
mk and α(m) = 1 mh , but the same analysis applies to the cases
η(m) = O( 1 mk ) and α(m) = O( 1 mh ).

heading: 6.6 ASSUMPTIONS OF THEOREM 7
text: For the assumptions of Theorem 7, the first condition ∑∞
m=1 α (m) < ∞ requires h > 1. Besides, the second
condition ∞∑
m=1 m∑ n=1 α(m)η(n) ≈ 1 h− 1 ∞∑ n=1 η(n) 1 nh−1 = 1 h− 1 ∞∑ n=1
1
nk+h−1 <∞ (93)
requires k + h > 2. The approximation comes from the fact that for every p > 1, we have ∞∑
k=n
k−p ≈ ∫ ∞ k=n k−pdx = 1 1− px 1−p ∣∣∣∣∞ n = 1 p− 1 1 np−1 . (94)
Since k ≥ 1 due to Assumption 3, we conclude that k + h > 2. Therefore, the conditions for η(m) and α(m) to satisfy the assumptions of Theorem 7 are h > 1 and k ≥ 1.

heading: 6.7 ASSUMPTIONS OF LEMMA 8
text: For the assumptions of Theorem 7, the first condition ∞∑
m=1 ∞∑ n=m α(n) ≈ ∞∑ m=1 1 mh−1 <∞ (95)
requires h > 2.
Besides, the second condition is ∞∑
m=1 ∞∑ i=m i∑ n=1 α(i)η(n) = ∞∑ m=1 ∞∑ i=m α(i) i∑ n=1 η(n) ≤ C ∞∑ m=1 ∞∑ i=m α(i) <∞. (96)
The inequality holds because for any p > 1, we have n∑
k=1
k−p ≈ ∫ n k=1 k−pdk = 1 1− pk 1−p ∣∣∣∣n 1 = 1 p− 1(1− n 1−p) ≤ C (97)
Therefore, the conditions for η(m) and α(m) to satisfy the assumptions of Lemma 8 are h > 2 and k ≥ 1.

Review summaries:
The authors propose a momentum based approach for batch normalization and provide an asymptotic convergence analysis of the objective in terms of the first order criterion. To my understanding, the main effort in the analysis is to show that the sequences of interest are Cauchy. Some numerical results are reported to demonstrate that the proposed variant of BN slightly outperforms BN with careful adjustment of some hyper parameter. The proposed approach is incremental, and the theoretical results are somewhat weak.

In this work, the authors propose a generalization of the batch normalization (BN) technique often used in training neural networks, and analyzed this convergence. In particular, a one hidden layer and one BN hidden layer fully connected network is considered, and a deterministic gradient descent algorithm with certain kind of BN has been considered in this work. The proposed “generalized” BN strategy is devised on the deterministic setting, but it is a slight generalization of the original BN by introducing a moving average operation. Classical results of Bertsekas is leveraged to show the asymptotic convergence of the algorithm.