Title: TRAINING DEEP NEURAL NETWORKS WITH PAR-TIALLY ADAPTIVE MOMENTUM

Abstract: Adaptive gradient methods, which adopt historical gradient information to automatically adjust the learning rate, despite the nice property of fast convergence, have been observed to generalize worse than stochastic gradient descent (SGD) with momentum in training deep neural networks. This leaves how to close the generalization gap of adaptive gradient methods an open problem. In this work, we show that adaptive gradient methods such as Adam, Amsgrad, are sometimes "over adapted". We design a new algorithm, called Partially adaptive momentum estimation method, which unifies the Adam/Amsgrad with SGD by introducing a partial adaptive parameter p, to achieve the best from both worlds. We also prove the convergence rate of our proposed algorithm to a stationary point in the stochastic nonconvex optimization setting. Experiments on standard benchmarks show that our proposed algorithm can maintain fast convergence rate as Adam/Amsgrad while generalizing as well as SGD in training deep neural networks. These results would suggest practitioners pick up adaptive gradient methods once again for faster training of deep neural networks.

1. INTRODUCTION:

Stochastic gradient descent (SGD) is now one of the most dominant approaches for training deep neural networks (Goodfellow et al., 2016) . In each iteration, SGD only performs one parameter update on a mini-batch of training examples. SGD is simple and has been proved to be efficient, especially for tasks on large datasets. In recent years, adaptive variants of SGD have emerged and shown their successes for their convenient automatic learning rate adjustment mechanism. Adagrad (Duchi et al., 2011) is probably the first along this line of research, and significantly outperforms vanilla SGD in the sparse gradient scenario. Despite the first success, Adagrad was later found to demonstrate degraded performance especially in cases where the loss function is nonconvex or the gradient is dense. Many variants of Adagrad, such as RMSprop (Hinton et al., 2012) , Adam (Kingma & Ba, 2015) , Adadelta (Zeiler, 2012) , Nadam (Dozat, 2016) , were then proposed to address these challenges by adopting exponential moving average rather than the arithmetic average used in Adagrad. This change largely mitigates the rapid decay of learning rate in Adagrad and hence makes this family of algorithms, especially Adam, particularly popular on various tasks. Recently, it has also been observed (Reddi et al., 2018) that Adam does not converge in some settings where rarely encountered large gradient information quickly dies out due to the "short momery" problem of exponential moving average. To address this issue, Amsgrad (Reddi et al., 2018) has been proposed to keep an extra "long term memory" variable to preserve the past gradient information and to correct the potential convergence issue in Adam. There are also some other variants of adaptive gradient method such as SC-Adagrad / SC-RMSprop (Mukkamala & Hein, 2017) , which derives logarithmic regret bounds for strongly convex functions.

On the other hand, people recently found that for largely over-parameterized neural networks, e.g., more complex modern convolutional neural network (CNN) architectures such as VGGNet (He et al., 2016) , ResNet (He et al., 2016) , Wide ResNet (Zagoruyko & Komodakis, 2016) , DenseNet (Huang et al., 2017) , training with Adam or its variants typically generalizes worse than SGD with momentum, even when the training performance is better. In particular, people found that carefully-tuned SGD, combined with proper momentum, weight decay and appropriate learning rate decay schedules, can significantly outperform adaptive gradient algorithms eventually (Wilson et al., 2017) . As a result, even though adaptive gradient methods are relatively easy to tune and converge faster at the early stage, recent advances in designing neural network structures are all reporting their performances by training their models with SGD-momentum (He et al., 2016; Zagoruyko & Komodakis, 2016; Huang et al., 2017; Simonyan & Zisserman, 2014; Ren et al., 2015; Xie et al., 2017; Howard et al., 2017) . Different from SGD, which adopts a universal learning rate for all coordinates, the effective learning rate of adaptive gradient methods, i.e., the universal base learning rate divided by the second order moment term, is different for different coordinates. Due to the normalization of the second order moment, some coordinates will have very large effective learning rates. To alleviate this problem, one usually chooses a smaller base learning rate for adaptive gradient methods than SGD with momentum. This makes the learning rate decay schedule less effective when applied to adaptive gradient methods, since a much smaller base learning rate will soon diminish after several rounds of decay. We refer to the above phenomenon as the "small learning rate dilemma" (see more details in Section 3).

With all these observations, a natural question is:

Can we take the best from both Adam and SGD-momentum, i.e., design an algorithm that not only enjoys the fast convergence rate as Adam, but also generalizes as well as SGD-momentum?

In this paper, we answer this question affirmatively. We close the generalization gap of adaptive gradient methods by proposing a new algorithm, called partially adaptive momentum estimation (Padam) method, which unifies Adam/Amsgrad with SGD-momentum to achieve the best of both worlds, by a partially adaptive parameter. The intuition behind our algorithm is: by controlling the degree of adaptiveness, the base learning rate in Padam does not need to be as small as other adaptive gradient methods. Therefore, it can maintain a larger learning rate while preventing the gradient explosion. We note that there exist several studies (Zaheer et al., 2018; Loshchilov & Hutter, 2019; Luo et al., 2019) that also attempted to address the same research question. In detail, Yogi (Zaheer et al., 2018) studied the effect of adaptive denominator constant and minibatch size in the convergence of adaptive gradient methods. AdamW (Loshchilov & Hutter, 2019) proposed to fix the weight decay regularization in Adam by decoupling the weight decay from the gradient update and this improves the generalization performance of Adam. AdaBound (Luo et al., 2019) applies dynamic bound of learning rate on Adam and make them smoothly converge to a constant final step size as in SGD. Our algorithm is very different from Yogi, AdamW and AdaBound. Padam is built upon a simple modification of Adam without extra complicated algorithmic design and it comes with a rigorous convergence guarantee in the nonconvex stochastic optimization setting.

We highlight the main contributions of our work as follows:

• We propose a novel and simple algorithm Padam with a partially adaptive parameter, which resolves the "small learning rate dilemma" for adaptive gradient methods and allows for faster convergence, hence closing the gap of generalization.

• We provide a convergence guarantee for Padam in nonconvex optimization. Specifically, we prove that the convergence rate of Padam to a stationary point for stochastic nonconvex optimization is

O d i=1 g 1:T,i 2 1/2 T 3/4 + d T , (1.1)

where g 1 , . . . , g T are the stochastic gradients and g 1:

T,i = [g 1,i , g 2,i , . . . , g T,i

] . When the stochastic gradients are ∞ -bounded, (1.1) matches the convergence rate of SGD in terms of the rate of T .

• We also provide thorough experiments about our proposed Padam method on training modern deep neural architectures. We empirically show that Padam achieves the fastest convergence speed while generalizing as well as SGD with momentum. These results suggest that practitioners should pick up adaptive gradient methods once again for faster training of deep neural networks.

• Last but not least, compared with the recent work on adaptive gradient methods, such as Yogi (Zaheer et al., 2018) , AdamW (Loshchilov & Hutter, 2019) , AdaBound (Luo et al., 2019) , our proposed Padam achieves better generalization performance than these methods in our experiments.

1.1. ADDITIONAL RELATED WORK:

Here we review additional related work that is not covered before. Zhang et al. (2017) proposed a normalized direction-preserving Adam (ND-Adam), which changes the adaptive terms from individual dimensions to the whole gradient vector. Keskar & Socher (2017) proposed to improve the generalization performance by switching from Adam to SGD. On the other hand, despite the great successes of adaptive gradient methods for training deep neural networks, the convergence guarantees for these algorithms are still understudied. Most convergence analyses of adaptive gradient methods are restricted to online convex optimization (Duchi et al., 2011; Kingma & Ba, 2015; Mukkamala & Hein, 2017; Reddi et al., 2018) . A few recent attempts have been made to analyze adaptive gradient methods for stochastic nonconvex optimization. More specifically, Basu et al. (2018) proved the convergence rate of RMSProp and Adam when using deterministic gradient rather than stochastic gradient. Li & Orabona (2018) analyzed convergence rate of AdaGrad under both convex and nonconvex settings but did not consider more complicated Adam-type algorithms. Ward et al. (2018) also proved the convergence rate of AdaGrad under both convex and nonconvex settings without considering the effect of stochastic momentum. 

x 2 = d i=1 x 2 i , the ∞ norm of x by x ∞ = max d i=1 |x i |.

For a sequence of vectors {x j } t j=1 , we denote by x j,i the i-th element in x j . We also denote x 1:t,i = [x 1,i , . . . , x t,i ] . With slightly abuse of notation, for two vectors a and b, we denote a 2 as the element-wise square, a p as the element-wise power operation, a/b as the element-wise division and max(a, b) as the elementwise maximum. We denote by diag(a) a diagonal matrix with diagonal entries a 1 , . . . , a d . Given two sequences {a n } and {b n }, we write

a n = O(b n ) if there exists a positive constant C such that a n ≤ Cb n and a n = o(b n ) if a n /b n → 0 as n → ∞. Notation O(•) hides logarithmic factors.

2. REVIEW OF ADAPTIVE GRADIENT METHODS:

Various adaptive gradient methods have been proposed in order to achieve better performance on various stochastic optimization tasks. Adagrad (Duchi et al., 2011) is among the first methods with adaptive learning rate for each individual dimension, which motivates the research on adaptive gradient methods in the machine learning community. In detail, Adagrad1 adopts the following update form:

EQUATION

where g t stands for the stochastic gradient ∇f t (x t ), and α t = α/ √ t is the step size. In this paper, we call α t base learning rate, which is the same for all coordinates of x t , and we call α t / √ v t,i effective learning rate for the i-th coordinate of x t , which varies across the coordinates. Adagrad is proved to enjoy a huge gain in terms of convergence especially in sparse gradient situations. Empirical studies also show a performance gain even for non-sparse gradient settings. RMSprop (Hinton et al., 2012) follows the idea of adaptive learning rate and it changes the arithmetic averages used for v t in Adagrad to exponential moving averages. Even though RMSprop is an empirical method with no theoretical guarantee, the outstanding empirical performance of RMSprop raised people's interests in exponential moving average variants of Adagrad. Adam (Kingma & Ba, 2015) 2 is the most popular exponential moving average variant of Adagrad. It combines the idea of RMSprop and momentum acceleration, and takes the following update form:

x t+1 = x t -α t m t √ v t where m t = β 1 m t-1 + (1 -β 1 )g t , v t = β 2 v t-1 + (1 -β 2 )g 2 t . (Adam)

Adam also requires α t = α/ √ t for the sake of convergence analysis. In practice, any decaying step size or even constant step size works well for Adam. Note that if we choose β 1 = 0, Adam basically reduces to RMSprop. Reddi et al. (2018) identified a non-convergence issue in Adam. Specifically, Adam does not collect long-term memory of past gradients and therefore the effective learning rate could be increasing in some cases. They proposed a modified algorithm namely Amsgrad. More specifically, Amsgrad adopts an additional step to ensure the decay of the effective learning rate α t / √ v t , and its key update formula is as follows:

EQUATION

m t and v t are the same as Adam. By introducing the v t term, Reddi et al. (2018) corrected some mistakes in the original proof of Adam and proved an O(1/ √ T ) convergence rate of Amsgrad for convex optimization. Note that all the theoretical guarantees on adaptive gradient methods (Adagrad, Adam, Amsgrad) are only proved for convex functions.

3. THE PROPOSED ALGORITHM:

In this section, we propose a new algorithm for bridging the generalization gap for adaptive gradient methods. Specifically, we introduce a partial adaptive parameter p to control the level of adaptiveness of the optimization procedure. The proposed algorithm is displayed in Algorithm 1.

Algorithm 1 Partially adaptive momentum estimation method (Padam) input: initial point x 1 ∈ X ; step sizes {α t }; adaptive parameters

β 1 , β 2 , p ∈ (0, 1/2] set m 0 = 0, v 0 = 0, v 0 = 0 for t = 1, . . . , T do g t = ∇f (x t , ξ t ) m t = β 1 m t-1 + (1 -β 1 )g t v t = β 2 v t-1 + (1 -β 2 )g 2 t v t = max( v t-1 , v t ) x t+1 = x t -α t • m t / v p t end for Output: Choose x out from {x t }, 2 ≤ t ≤ T with probability α t-1 / T -1 i=1 α i )

In Algorithm 1, g t denotes the stochastic gradient and v t can be seen as a moving average over the second order moment of the stochastic gradients. As we can see from Algorithm 1, the key difference between Padam and Amsgrad (Reddi et al., 2018) is that: while m t is still the momentum as in Adam/Amsgrad, it is now "partially adapted" by the second order moment, i.e.,

EQUATION

We call p ∈ [0, 1/2] the partially adaptive parameter. Note that 1/2 is the largest possible value for p and a larger p will result in non-convergence in the proof (see the proof details in the supplementary materials). When p → 0, Algorithm 1 reduces to SGD with momentum3 and when p = 1/2, Algorithm 1 is exactly Amsgrad. Therefore, Padam indeed unifies Amsgrad and SGD with momentum.

With the notations defined above, we are able to formally explain the "small learning rate dilemma". In order to make things clear, we first emphasize the relationship between adaptiveness and learning rate decay. We refer the actual learning rate applied to m t as the effective learning rate, i.e., α t / v p t . Now suppose that a learning rate decay schedule is applied to α t . If p is large, then at early stages, the effective learning rate α t / v p t,i could be fairly large for certain coordinates with small v t,i value4 . To prevent those coordinates from overshooting we need to enforce a smaller α t , and therefore the base learning rate must be set small (Keskar & Socher, 2017; Wilson et al., 2017) . As a result, after several rounds of decaying, the learning rates of the adaptive gradient methods are too small to make any significant progress in the training process5 . We call this phenomenon "small learning rate dilemma". It is also easy to see that the larger p is, the more severe "small learning rate dilemma" is. This suggests that intuitively, we should consider using Padam with a proper adaptive parameter p, and choosing p < 1/2 can potentially make Padam suffer less from the "small learning rate dilemma" than Amsgrad, which justifies the range of p in Algorithm 1. We will show in our experiments (Section 5) that Padam with p < 1/2 can adopt an equally large base learning rate as SGD with momentum.

Note that even though in Algorithm 1, the choice of α t covers different choices of learning rate decay schedule, the main focus of this paper is not about finding the best learning rate decay schedule, but designing a new algorithm to control the adaptiveness for better empirical generalization result. In other words, our focus is not on α t , but on v t . For this reason, we simply fix the learning rate decay schedule for all methods in the experiments to provide a fair comparison for different methods. Figure 1 shows the comparison of test error performances under the different partial adaptive parameter p for ResNet on both CIFAR-10 and CIFAR-100 datasets. We can observe that a larger p will lead to fast convergence at early stages and worse generalization performance later, while a smaller p behaves more like SGD with momentum: slow in early stages but finally catch up. With a proper choice of p (e.g., 1/8 in this case), Padam can obtain the best of both worlds. Note that besides Algorithm 1, our partially adaptive idea can also be applied to other adaptive gradient methods such as Adagrad, Adadelta, RMSprop, AdaMax (Kingma & Ba, 2015) . For the sake of conciseness, we do not list the partially adaptive versions for other adaptive gradient methods here. We also would like to comment that Padam is totally different from the p-norm generalized version of Adam in Kingma & Ba (2015) , which induces AdaMax method when p → ∞. In their case, p-norm is used to generalize 2-norm of their current and past gradients while keeping the scale of adaptation unchanged. In sharp contrast, we intentionally change (reduce) the scale of the adaptive term in Padam to get better generalization performance. Finally, note that in Algorithm 1 we remove the bias correction step used in the original Adam paper following Reddi et al. (2018) . Nevertheless, our arguments and theory are applicable to the bias correction version as well.

4. CONVERGENCE ANALYSIS OF THE PROPOSED ALGORITHM:

In this section, we establish the convergence theory of Algorithm 1 in the stochastic nonconvex optimization setting, i.e., we aim at solving the following stochastic nonconvex optimization problem

min x∈R d f (x) := E ξ f (x; ξ) ,

where ξ is a random variable satisfying certain distribution, f (x; ξ) : R d → R is a L-smooth nonconvex function. In the stochastic setting, one cannot directly access the full gradient of f (x). Instead, one can only get unbiased estimators of the gradient of f (x), which is ∇f (x; ξ). This setting has been studied in Ghadimi & Lan (2013; 2016) . We first introduce the following assumptions.

Assumption 4.1 (Bounded Gradient). f (x) = E ξ f (x; ξ) has G ∞ -bounded stochastic gradient. That is, for any ξ, we assume that ∇f (x; ξ) ∞ ≤ G ∞ .

It is worth mentioning that Assumption 4.1 is slightly weaker than the 2 -boundedness assumption ∇f (x; ξ) 2 ≤ G 2 used in Reddi et al. ( 2016); Chen et al. (2018) 

. Since ∇f (x; ξ) ∞ ≤ ∇f (x; ξ) 2 ≤ √ d ∇f (x; ξ) ∞ , the 2 -boundedness assumption implies Assumption 4.1 with G ∞ = G 2 . Meanwhile, G ∞ will be tighter than G 2 by a factor of √ d when each coordinate of ∇f (x; ξ) almost equals to each other. Assumption 4.2 (L-smooth). f (x) = E ξ f (x; ξ) is L-smooth: for any x, y ∈ R d , it satisfied that f (x) -f (y) -∇f (y), x -y ≤ L 2 x -y 2 2 .

Assumption 4.2 is frequently used in analysis of gradient-based algorithms. It is equivalent to the L-gradient Lipschitz condition, which is often written as ∇f (x) -∇f (y) 2 ≤ L xy 2 . Next we provide the main convergence rate result for our proposed algorithm.

Theorem 4.3 (Padam). In Algorithm 1, suppose that p ∈ [0, 1/2], β 1 < β 2p 2 and α t = α for t = 1, . . . , T , under Assumptions 4.1 and 4.2, let ∆f = f (x 1 ) -inf x f (x), for any q ∈ [max{0, 4p -1}, 1], the output x out of Algorithm 1 satisfies that

E ∇f (xout) 2 2 ≤ M1 T α + M2d T + M3αd q T (1-q)/2 E d i=1 g1:T,i 2 1-q , (4.1)

where

M1 = 2G 2p ∞ ∆f, M2 = 4G 2+2p ∞ E v -p 1 1 d(1 -β1) + 4G 2 ∞ , M3 = 4LG 1+q-2p ∞ (1 -β2) 2p + 8LG 1+q-2p ∞ (1 -β1) (1 -β2) 2p (1 -β1/β 2p 2 ) β1 1 -β1 2 .

Remark 4.4. From Theorem 4.3, we can see that M 1 and M 3 are independent of the number of iterations T and dimension d. 1), it is easy to see that M 2 also has an upper bound that is independent of T and d.

In addition, if v -1 1 ∞ = O(

The following corollary is a special case of Theorem 4.3 when p ∈ [0, 1/4] and q = 0. Corollary 4.5. Under the same conditions in Theorem 4.

3, if p ∈ [0, 1/4], Padam's output satisfies E ∇f (xout) 2 2 ≤ M1 T α + M2d T + M 3 α √ T E d i=1 g1:T,i 2 , (4.2)

where M 1 and M 2 and ∆f are the same as in Theorem 4.3, and M 3 is defined as follows:

M 3 = 4LG 1-2p ∞ (1 -β2) 2p + 8LG 1-2p ∞ (1 -β1) (1 -β2) 2p (1 -β1/β 2p 2 ) β1 1 -β1 2 .

Remark 4.6. Corollary 4.5 simplifies the result of Theorem 4.3 by choosing q = 0 under the condition p ∈ [0, 1/4]. We remark that this choice of q is optimal in an important special case studied in Duchi et al. (2011); Reddi et al. (2018) : when the gradient vectors are sparse, we assume that d i=1 g 1:T,i 2

√:

dT . Then for q > 0, it follows that

d i=1 g1:T,i 2 T d q d i=1 g1:T,i 2 1-q T 1-q/2 . (4.3) (4.

3) implies that the upper bound provided by (4.2) is strictly better than (4.1) with q > 0. Therefore when the gradient vectors are sparse, Padam achieves faster convergence when p ∈ [0, 1/4]. Remark 4.7. We show the convergence rate under different choices of step size α. If

α = Θ T 1/4 d i=1 g1:T,i 2 1/2 -1 ,

then by (4.2), we have

E ∇f (xout) 2 2 = O d i=1 g1:T,i 2 1/2 T 3/4 + d T . (4.4)

Note that the convergence rate given by (4.4) is related to the sum of gradient norms d i=1 g 1:T,i 2 . As we mentioned in Remark 4.6, when the stochastic gradients g 1:T,i , i = 1, . . . , d are sparse, we have d i=1 g 1:T,i 2

dT (Duchi et al. (2011) ). More specifically, suppose 

d i=1 g 1:T,i 2 = O(d s √ T ) for some 0 ≤ s ≤ 1/2. We have E[ ∇f (x out ) 2 2 ] = O d s/2 /T 1/2 + d/T . When s = 1/2, we have E[ ∇f (x out ) 2 2 ] = O d 1/4 / √ T + d/T ,

2 2 = O 1 √ T + d T + 1 T d i=1 g1:T,i 2 .

When d i=1 g 1:T,i 2 √ dT (Duchi et al., 2011; Reddi et al., 2018) , we further have

E ∇f (xout) 2 2 = O 1 √ T + d T + d T ,

which matches the convergence result in nonconvex SGD (Ghadimi & Lan, 2016) considering the dependence of T .

5. EXPERIMENTS:

In this section, we empirically evaluate our proposed algorithm for training various modern deep learning models and test them on several standard benchmarks6 . We show that for nonconvex loss functions in deep learning, our proposed algorithm still enjoys a fast convergence rate, while its generalization performance is as good as SGD with momentum and much better than existing adaptive gradient methods such as Adam and Amsgrad. All experiments are conducted on Amazon AWS p3.8xlarge servers which come with Intel Xeon E5 CPU and 4 NVIDIA Tesla V100 GPUs (16G RAM per GPU). All experiments are implemented in Pytorch version 0.4.1 within Python 3.6.4. 4) AdamW (Loshchilov & Hutter, 2019) (5) Yogi (Zaheer et al., 2018) and ( 6) AdaBound (Luo et al., 2019) . We use several popular datasets for image classifications and language modeling: CIFAR-10 ( Krizhevsky & Hinton, 2009) , CIFAR-100 (Krizhevsky & Hinton, 2009) , ImageNet dataset (ILSVRC2012) (Deng et al., 2009) and Penn Treebank dataset (Marcus et al., 1993) . We adopt three popular CNN architectures to evaluate the performance of our proposed algorithms on image classification task: VGGNet-16 (Simonyan & Zisserman, 2014) , Residual Neural Network (ResNet-18) (He et al., 2016) , Wide Residual Network (WRN-16-4) (Zagoruyko & Komodakis, 2016) . We test the language modeling task using 2-layer and 3-layer Long Short-Term Memory (LSTM) network (Hochreiter & Schmidhuber, 1997) . We test CIFAR-10, CIFAR-100 and Penn Treebank tasks for 200 epochs. For all experiments, we use a fixed multi-stage learning rate decaying scheme: we decay the learning rate by 0.1 at the 100th and 150th epochs. We test ImageNet tasks for 100 epochs with similar multi-stage learning rate decaying scheme at the 30th, 60th and 80th epochs. We perform grid search on validation set to choose the best hyper-parameters for each algorithm. Details about the datasets, CNN architectures and all the specific model parameters used in the following experiments can be found in the supplementary materials. SGD with momentum makes a huge drop while fully adaptive gradient methods (Adam and Amsgrad) start to generalize badly. Padam, on the other hand, maintains relatively good generalization performance and also holds the lead over other algorithms. After the second decaying at the 150-th epoch, Adam and Amsgrad basically lose all the power of traversing through the parameter space due to the "small learning dilemma", while the performance of SGD with momentum finally catches up with Padam. AdamW, Yogi and Ad-aBound indeed improve the performance compared with original Adam but the performance is still worse than Padam. Overall we can see that Padam achieves the best of both worlds (i.e., Adam and SGD with momentum): it maintains faster convergence rate while also generalizing as well as SGD with momentum in the end. Figure 3 plots the Top-1 and Top-5 error against training epochs on the ImageNet dataset for both VGGNet and ResNet. We can see that on ImageNet, all methods behave similarly as in our CIFAR-10 experiments. Padam method again obtains the best from both worlds by achieving the fastest convergence while generalizing as well as SGD with momentum. Even though methods such as AdamW, Yogi and AdaBound have better performance than standard Adam, they still suffer from a big generalization gap on ImageNet dataset. Note that we did not conduct WideResNet experiment on Imagenet dataset due to GPU memory limits.

5.1. EXPERIMENTAL RESULTS:

We also perform experiments on the language modeling tasks to test our proposed algorithm on Long Short-Term Memory (LSTM) network (Hochreiter & Schmidhuber, 1997) , where adaptive gradient methods such as Adam are currently the mainstream optimizers for these tasks. Figure 4 plots the test perplexity against training epochs on the Penn Treebank dataset (Marcus et al., 1993) for both 2-layer LSTM and 3-layer LSTM models. We can observe that the differences on simpler 2-layer LSTM model is not very obvious but on more complicated 3-layer LSTM model, different algorithms have quite different optimizing behaviors. Even though Adam, Amsgrad and AdamW have faster convergence in the early stages, Padam achieves the best final test perplexity on this language modeling task for both of our experiments.

6. CONCLUSIONS AND FUTURE WORK:

In this paper, we proposed Padam, which unifies Adam/Amsgrad with SGD-momentum. With an appropriate choice of the partially adaptive parameter, we show that Padam can achieve the best from both worlds, i.e., maintaining fast convergence rate while closing the generalization gap. We also provide a theoretical analysis towards the convergence rate of Padam to a stationary point for stochastic nonconvex optimization.

While the empirical generalization performances achieved by Padam backup our hypothesis of "small learning rate dilemma", it is still unclear how learning rate decay affects the performance of adaptive gradient methods in theory. We leave it as an important future direction.

Lemma A.4. Let v t and m t be as defined in Algorithm 1. Then under Assumption 4.1, we have .5 . Suppose that f has G ∞ -bounded stochastic gradient. Let β 1 , β 2 be the weight parameters, α t , t = 1, . . . , T be the step sizes in Algorithm 1 and q ∈ [max{4p -1, 0}, 1]. We denote γ = β 1 /β 2p 2 . Suppose that α t = α and γ ≤ 1, then under Assumption 4.1, we have the following two results:

∇f (x) ∞ ≤ G ∞ , v t ∞ ≤ G 2 ∞ and m t ∞ ≤ G ∞ . Lemma A

T t=1 α 2 t E V -p t m t 2 2 ≤ T (1+q)/2 d q α 2 (1 -β 1 )G (1+q-4p) ∞ (1 -β 2 ) 2p (1 -γ) E d i=1 g 1:T,i 2 1-q , and T t=1 α 2 t E V -p t g t 2 2 ≤ T (1+q)/2 d q α 2 G (1+q-4p) ∞ (1 -β 2 ) 2p E d i=1 g 1:T,i 2 1-q

. Now we are ready to prove Theorem 4.3.

Proof of Theorem 4.3. Since f is L-smooth, we have:

f (z t+1 ) ≤ f (z t ) + ∇f (z t ) (z t+1 -z t ) + L 2 z t+1 -z t 2 2 = f (z t ) + ∇f (x t ) (z t+1 -z t ) I1 + (∇f (z t ) -∇f (x t )) (z t+1 -z t ) I2 + L 2 z t+1 -z t 2 2 I3 (A.5)

In the following, we bound I 1 , I 2 and I 3 separately.

Bounding term I 1 : When t = 1, we have

∇f (x 1 ) (z 2 -z 1 ) = -∇f (x 1 ) α 1 V -p t g 1 . (A.6) For t ≥ 2, we have ∇f (x t ) (z t+1 -z t ) = ∇f (x t ) β 1 1 -β 1 α t-1 V -p t-1 -α t V -p t m t-1 -α t V -p t g t = β 1 1 -β 1 ∇f (x t ) α t-1 V -p t-1 -α t V -p t m t-1 -∇f (x t ) α t V -p t g t , (A.7)

where the first equality holds due to (A.3) in Lemma A.1. For ∇f

(x t ) (α t-1 V -p t-1 -α t V -p t )m t-1 in (A.7), we have ∇f (x t ) (α t-1 V -p t-1 -α t V -p t )m t-1 ≤ ∇f (x t ) ∞ • α t-1 V -p t-1 -α t V -p t 1,1 • m t-1 ∞ ≤ G 2 ∞ α t-1 V -p t-1 1,1 -α t V -p t 1,1 = G 2 ∞ α t-1 v -p t-1 1 -α t v -p t 1 . (A.8)

The first inequality holds because for a positive diagonal matrix A, we have

x Ay ≤ x ∞ • A 1,1 • y ∞ . The second inequality holds due to α t-1 V -p t-1 α t V -p t 0. Next we bound -∇f (x t ) α t V -p t g t . We have -∇f (x t ) α t V -p t g t = -∇f (x t ) α t-1 V -p t-1 g t -∇f (x t ) α t V -p t -α t-1 V -p t-1 g t ≤ -∇f (x t ) α t-1 V -p t-1 g t + ∇f (x t ) ∞ • α t V -p t -α t-1 V -p t-1 1,1 • g t ∞ ≤ -∇f (x t ) α t-1 V -p t-1 g t + G 2 ∞ α t-1 V -p t-1 1,1 -α t V -p t 1,1 = -∇f (x t ) α t-1 V -p t-1 g t + G 2 ∞ α t-1 v -p t-1 1 -α t v -p t 1 . (A.9)

The first inequality holds because for a positive diagonal matrix A, we have

x Ay ≤ x ∞ • A 1,1 • y ∞ .

The second inequality holds due to α t-1 V -p t-1 α t V -p t 0. Substituting (A.8) and (A.9) into (A.7), we have

∇f (x t ) (z t+1 -z t ) ≤ -∇f (x t ) α t-1 V -p t-1 g t + 1 1 -β 1 G 2 ∞ α t-1 v -p t-1 1 -α t v -p t 1 . (A.10) Bounding term I 2 : For t ≥ 1, we have ∇f (z t ) -∇f (x t ) (z t+1 -z t ) ≤ ∇f (z t ) -∇f (x t ) 2 • z t+1 -z t 2 ≤ α t V -p t g t 2 + β 1 1 -β 1 x t-1 -x t 2 • β 1 1 -β 1 • L x t -x t-1 2 = L β 1 1 -β 1 α t V -p t g t 2 • x t -x t-1 2 + L β 1 1 -β 1 2 x t -x t-1 2 2 ≤ L α t V -p t g t 2 2 + 2L β 1 1 -β 1 2 x t -x t-1 2 2 , (A.11)

where the second inequality holds because of Lemma A.1 and Lemma A.2, the last inequality holds due to Young's inequality.

Bounding term I 3 : For t ≥ 1, we have L 2 z t+1 -z t 2 2 ≤ L 2 α t V -p t g t 2 + β 1 1 -β 1 x t-1 -x t 2 2 ≤ L α t V -p t g t 2 2 + 2L β 1 1 -β 1 2 x t-1 -x t 2 2 .

(A.12)

The first inequality is obtained by introducing Lemma A.1.

For t = 1, substituting (A.6), (A.11) and (A.12) into (A.5), taking expectation and rearranging terms, we have A.13) where the last inequality holds because

EQUATION

-∇f (x 1 ) V -p 1 g 1 ≤ d • ∇f (x 1 ) ∞ • V -p 1 g 1 ∞ ≤ d • G ∞ • G 1-2p ∞ = dG 2-2p ∞ .

For t ≥ 2, substituting (A.10), (A.11) and (A.12) into (A.5), taking expectation and rearranging terms, we have

E f (z t+1 ) + G 2 ∞ α t v -p t 1 1 -β 1 -f (z t ) + G 2 ∞ α t-1 v -p t-1 1 1 -β 1 ≤ E -∇f (x t ) α t-1 V -p t-1 g t + 2L α t V -p t g t 2 2 + 4L β 1 1 -β 1 2 x t -x t-1 2 2 = E -∇f (x t ) α t-1 V -p t-1 ∇f (x t ) + 2L α t V -p t g t 2 2 + 4L β 1 1 -β 1 2 α t-1 V -p t-1 m t-1 2 2 ≤ E -α t-1 ∇f (x t ) 2 2 (G 2p ∞ ) -1 + 2L α t V -p t g t 2 2 + 4L β 1 1 -β 1 2 α t-1 V -p t-1 m t-1 2 2 , (A.14)

where the equality holds because E[g t ] = ∇f (x t ) conditioned on ∇f (x t ) and V -p t-1 , the second inequality holds because of Lemma A.4. Telescoping (A.14) for t = 2 to T and adding with (A.13), we have

(G 2p ∞ ) -1 T t=2 α t-1 E ∇f (x t ) 2 2 ≤ E f (z 1 ) + G 2 ∞ α 1 v -p 1 1 1 -β 1 + dα 1 G 2-2p ∞ -f (z T +1 ) + G 2 ∞ α T v -p T 1 1 -β 1 + 2L T t=1 E α t V -p t g t 2 2 + 4L β 1 1 -β 1 2 T t=2 E α t-1 V -p t-1 m t-1 2 2 ≤ E ∆f + G 2 ∞ α 1 v -p 1 1 1 -β 1 + dα 1 G 2-2p ∞ + 2L T t=1 E α t V -p t g t 2 2 + 4L β 1 1 -β 1 2 T t=1 E α t V -p t m t 2 2 . (A.15)

By Lemma A.5, we have .16) where γ = β 1 /β 2p 2 . We also have

EQUATION

T t=1 α 2 t E V -p t g t 2 2 ≤ T (1+q)/2 d q α 2 G (1+q-4p) ∞ (1 -β 2 ) 2p E d i=1 g 1:T,i 2 1-q . (A.17) = 1 T t=2 α t-1 T t=2 α t-1 E ∇f (x t ) 2 2 ≤ G 2p ∞ T t=2 α t-1 E ∆f + G 2 ∞ α 1 v -p 1 1 1 -β 1 + dα 1 G 2-2p ∞ + 2LG 2p ∞ T t=2 α t-1 T (1+q)/2 d q α 2 G (1+q-4p) ∞ (1 -β 2 ) 2p E d i=1 g 1:T,i 2 1-q + 4LG 2p ∞ T t=2 α t-1 β 1 1 -β 1 2 T (1+q)/2 d q α 2 (1 -β 1 )G (1+q-4p) ∞ (1 -β 2 ) 2p (1 -γ) E d i=1 g 1:T,i 2 1-q ≤ 1 T α 2G 2p ∞ ∆f + 4 T G 2+2p ∞ E v -p 1 1 1 -β 1 + dG 2 ∞ + d q α T (1-q)/2 E d i=1 g 1:T,i 2 1-q 4LG 1+q-2p ∞ (1 -β 2 ) 2p + 8LG 1+q-2p ∞ (1 -β 1 ) (1 -β 2 ) 2p (1 -γ) β 1 1 -β 1 2 , (A.18)

where the second inequality holds because α t = α. Rearranging (A.18), we obtain

E ∇f (x out ) 2 2 ≤ M 1 T α + M 2 d T + αd q M 3 T (1-q)/2 E d i=1 g 1:T,i 2 1-q

, where {M i } 3 i=1 are defined in Theorem 4.3. This completes the proof.

A.2 PROOF OF COROLLARY 4.5

Proof of Corollary 4.5. From Theorem 4.3, let p ∈ [0, 1/4], we have q ∈ [0, 1]. Setting q = 0, we have

E ∇f (x out ) 2 2 ≤ M 1 T α + M 2 • d T + M 3 α √ T E d i=1 g 1:T,i 2 ,

where M 1 and M 2 are defined in Theorem 4.3 and M 3 is defined in Corollary 4.5. This completes the proof.

B PROOF OF TECHNICAL LEMMAS:

B.1 PROOF OF LEMMA A.1

Proof. By definition, we have

z t+1 = x t+1 + β 1 1 -β 1 (x t+1 -x t ) = 1 1 -β 1 x t+1 - β 1 1 -β 1 x t .

Then we have

z t+1 -z t = 1 1 -β 1 (x t+1 -x t ) - β 1 1 -β 1 (x t -x t-1 ) = 1 1 -β 1 -α t V -p t m t + β 1 1 -β 1 α t-1 V -p t-1 m t-1 .

The equities above are based on definition. Then we have

z t+1 -z t = -α t V -p t 1 -β 1 β 1 m t-1 + (1 -β 1 )g t + β 1 1 -β 1 α t-1 V -p t-1 m t-1 = β 1 1 -β 1 m t-1 α t-1 V -p t-1 -α t V -p t -α t V -p t g t = β 1 1 -β 1 α t-1 V -p t-1 m t-1 I -α t V -p t α t-1 V -p t-1 -1 -α t V -p t g t = β 1 1 -β 1 I -α t V -p t α t-1 V -p t-1 -1 (x t-1 -x t ) -α t V -p t g t .

The equalities above follow by combining the like terms.

B.2 PROOF OF LEMMA A.2

Proof. By Lemma A.1, we have

z t+1 -z t 2 = β 1 1 -β 1 I -(α t V -p t )(α t-1 V -p t-1 ) -1 (x t-1 -x t ) -α t V -p t g t 2 ≤ β 1 1 -β 1 I -(α t V -p t )(α t-1 V -p t-1 ) -1 ∞,∞ • x t-1 -x t 2 + α V -p t g t 2 ,

where the inequality holds because the term β 1 /(1 -β 1 ) is positive, and triangle inequality. Con-

sidering that α t v -p t,j ≤ α t-1 v -p t-1,j , when p > 0, we have I -(α t V -p t )(α t-1 V -p t-1 ) -1 ∞,∞ ≤ 1.

With that fact, the term above can be bound as:

z t+1 -z t 2 ≤ α V -p t g t 2 + β 1 1 -β 1 x t-1 -x t 2 .

This completes the proof.

B.3 PROOF OF LEMMA A.3:

Proof. For term ∇f (z t ) -∇f (x t ) 2 , we have:

∇f (z t ) -∇f (x t ) 2 ≤ L z t -x t 2 ≤ L β 1 1 -β 1 (x t -x t-1 ) 2 ≤ L β 1 1 -β 1 • x t -x t-1 2 ,

where the last inequality holds because the term β 1 /(1 -β 1 ) is positive.

B.4 PROOF OF LEMMA A.4:

Proof of Lemma A.4. Since f has G ∞ -bounded stochastic gradient, for any x and ξ, ∇f (x; ξ) ∞ ≤ G ∞ . Thus, we have

∇f (x) ∞ = E ξ ∇f (x; ξ) ∞ ≤ E ξ ∇f (x; ξ) ∞ ≤ G ∞ .

Next we bound m t ∞ . We have m 0 = 0 ≤ G ∞ . Suppose that m t ∞ ≤ G ∞ , then for m t+1 , we have

m t+1 ∞ = β 1 m t + (1 -β 1 )g t+1 ∞ ≤ β 1 m t ∞ + (1 -β 1 ) g t+1 ∞ ≤ β 1 G ∞ + (1 -β 1 )G ∞ = G ∞ .

Thus, for any t ≥ 0, we have

m t ∞ ≤ G ∞ . Finally we bound v t ∞ . First we have v 0 ∞ = v 0 ∞ = 0 ≤ G 2 ∞ . Suppose that v t ∞ ≤ G 2 ∞ and v t ∞ ≤ G 2 ∞ . Note that we have v t+1 ∞ = β 2 v t + (1 -β 2 )g 2 t+1 ∞ ≤ β 2 v t ∞ + (1 -β 2 ) g 2 t+1 ∞ ≤ β 2 G 2 ∞ + (1 -β 2 )G 2 ∞ = G 2

∞ , and by definition, we have

v t+1 ∞ = max{ v t ∞ , v t+1 ∞ } ≤ G 2 ∞ . Thus, for any t ≥ 0, we have v t ∞ ≤ G 2 ∞ .

B.5 PROOF OF LEMMA A.5

Proof. Recall that v t,j , m t,j , g t,j denote the j-th coordinate of v t , m t and g t . We have

α 2 t E V -p t m t 2 2 = α 2 t E d i=1 m 2 t,i v 2p t,i ≤ α 2 t E d i=1 m 2 t,i v 2p t,i = α 2 t E d i=1 ( t j=1 (1 -β 1 )β t-j 1 g j,i ) 2 ( t j=1 (1 -β 2 )β t-j 2 g 2 j,i ) 2p , (B.1)

where the first inequality holds because v t,i ≥ v t,i . Next we have

α 2 t E d i=1 ( t j=1 (1 -β 1 )β t-j 1 g j,i ) 2 ( t j=1 (1 -β 2 )β t-j 2 g 2 j,i ) 2p ≤ α 2 t (1 -β 1 ) 2 (1 -β 2 ) 2p E d i=1 ( t j=1 β t-j 1 |g j,i | (1+q-4p) )( t j=1 β t-j 1 |g j,i | (1-q+4p) ) ( t j=1 β t-j 2 g 2 j,i ) 2p ≤ α 2 t (1 -β 1 ) 2 (1 -β 2 ) 2p E d i=1 ( t j=1 β t-j 1 G (1+q-4p) ∞ )( t j=1 β t-j 1 |g j,i | (1-q+4p) ) ( t j=1 β t-j 2 g 2 j,i ) 2p ≤ α 2 t (1 -β 1 )G (1+q-4p) ∞ (1 -β 2 ) 2p E d i=1 t j=1 β t-j 1 |g j,i | (1-q+4p) ( t j=1 β t-j 2 g 2 j,i ) 2p , (B.2)

where the first inequality holds due to Cauchy inequality, the second inequality holds because |g j,i | ≤ G ∞ , the last inequality holds because

t j=1 β t-j 1 ≤ (1 -β 1 ) -1 . Note that d i=1 t j=1 β t-j 1 |g j,i | (1-q+4p) ( t j=1 β t-j 2 g 2 j,i ) 2p ≤ d i=1 t j=1 β t-j 1 |g j,i | (1-q+4p) (β t-j 2 g 2 j,i ) 2p = d i=1 t j=1 γ t-j |g j,i | 1-q , (B.3)

where the equality holds due to the definition of γ. Substituting (B.2) and (B.3) into (B.1), we have

α 2 t E V -p t m t 2 2 ≤ α 2 t (1 -β 1 )G (1+q-4p) ∞ (1 -β 2 ) 2p E d i=1 t j=1 γ t-j |g j,i | 1-q . (B.4) Telescoping (B.4) for t = 1 to T , we have T t=1 α 2 t E V -p t m t 2 2 ≤ α 2 (1 -β 1 )G (1+q-4p) ∞ (1 -β 2 ) 2p E T t=1 d i=1 t j=1 γ t-j |g j,i | 1-q = α 2 (1 -β 1 )G (1+q-4p) ∞ (1 -β 2 ) 2p E d i=1 T j=1 |g j,i | 1-q T t=j γ t-j ≤ α 2 (1 -β 1 )G (1+q-4p) ∞ (1 -β 2 ) 2p (1 -γ) E d i=1 T j=1 |g j,i | 1-q . (B.5) Finally, we have d i=1 T j=1 |g j,i | 1-q ≤ d i=1 T j=1 g 2 j,i (1-q)/2 • T (1+q)/2 = T (1+q)/2 d i=1 g 1:T,i 1-q 2 ≤ T (1+q)/2 d q d i=1 g 1:T,i 2 1-q , (B.6)

where the first and second inequalities hold due to Hölder's inequality. Substituting (B.6) into (B.5), we have

T t=1 α 2 t E V -p t m t 2 2 ≤ T (1+q)/2 d q α 2 (1 -β 1 )G (1+q-4p) ∞ (1 -β 2 ) 2p (1 -γ) E d i=1 g 1:T,i 2 1-q .

Specifically, taking β 1 = 0, we have m t = g t , then

T t=1 α 2 t E V -p t g t 2 2 ≤ T (1+q)/2 d q α 2 G (1+q-4p) ∞ (1 -β 2 ) 2p E d i=1 g 1:T,i 2 1-q . C EXPERIMENT DETAILS C.1 DATASETS

We use several popular datasets for image classifications.

• CIFAR-10 ( Krizhevsky & Hinton, 2009) : it consists of a training set of 50, 000 32 × 32 color images from 10 classes, and also 10, 000 test images.

• CIFAR-100 (Krizhevsky & Hinton, 2009) : it is similar to CIFAR-10 but contains 100 image classes with 600 images for each.

• ImageNet dataset (ILSVRC2012) (Deng et al., 2009) : ILSVRC2012 contains 1.28 million training images, and 50k validation images over 1000 classes.

In addition, we adopt Penn Treebank (PTB) dataset (Marcus et al., 1993) 

C.3 PARAMETER SETTINGS:

We perform grid searches to choose the best hyper-parameters for all algorithms in both image classification and language modeling tasks. For the base learning rate, we do grid search over {0.0001, 0.001, 0.01, 0.1, 1, 10, 100} for all algorithms, and choose the partial adaptive parameter p from {2/5, 1/4, 1/5, 1/8, 1/16} and the second order moment parameter β 2 from {0.9, 0.99, 0.999}. Specifically, in terms of image classification experiments, for SGD with momentum, the base learning rate is set to be 0.1 with a momentum parameter of 0.9. For all adaptive gradient methods except Padam, we set the base learning rate as 0.001. For AdaBound, the final learning rate is set to be 0.1. For Padam, the base learning rate is set to be 0.1 and the partially adaptive parameter p is set to be 1/8 due to its best empirical performance. For Adam, Amsgrad, the momentum parameters are set to be β 1 = 0.9, β 2 = 0.99. And for other adaptive gradient methods including Padam, we set β 1 = 0.9, β 2 = 0.999. The weight decay factor is set to 5 × 10 -4 for all methods except AdamW which adopts a different weight decay scheme. For AdamW, the normalized weight decay factor is set to 2.5 × 10 -2 for CIFAR and 5 × 10 -2 for ImageNet. For Yogi, is set as 10 -3 as suggested in the original paper. For AdaBound, the final learning rate is set as 0.1 as suggested in Luo et al. (2019) . The minibatch sizes for CIFAR-10 and CIFAR-100 are set to be 128 and for ImageNet dataset we set it to be 256. Regarding the LSTM experiments, for SGD with momentum, the base learning rate is set to be 1 for 2-layer LSTM model and 10 for 3-layer LSTM. The momentum parameter is set to be 0.9 for both models. For all adaptive gradient methods except Padam and Yogi, we set the base learning rate as 0.001. For Yogi, we set the base learning rate as 0.01 for 2-layer LSTM model and 0.1 for 3-layer LSTM model. For Padam, we set the base learning rate as 0.01 for 2-layer LSTM model and 1 for 3-layer LSTM model. For all adaptive gradient methods, we set β 1 = 0.9, β 2 = 0.999. In terms of algorithm specific parameters, for Padam, we set the partially adaptive parameter p as 0.4 for 2-layer LSTM model and 0.2 for 3-layer LSTM model. For AdaBound, we set the final learning rate as 10 for 2-layer LSTM model and 100 for 3-layer LSTM model. For Yogi, is set as 10 -3 as suggested in the original paper. The weight decay factor is set to 1.2 × 10 -6 for all methods except AdamW which adopts a different weight decay scheme. For AdamW, the normalized weight decay factor is set to 4 × 10 -4 . The minibatch size is set to be 20 for all LSTM experiments.

D ADDITIONAL EXPERIMENTS:

As a sanity check, we first plot the max and min of v t across all the coordinates in Figure 5 . In detail, the maximum value of v t is around 0.04 in the end and the minimum value is around 3 × 10 -8 , 

:

The formula is equivalent to the one from the original paper(Duchi et al.,

2011) after simple manipulations.2 Here for simplicity and consistency, we ignore the bias correction step in the original paper of Adam. Yet adding the bias correction step will not affect the argument in the paper.

The only difference between Padam with p = 0 and SGD-Momentum is an extra constant factor (1 -β1), which can be moved into the learning rate such that the update rules for these two algorithms are identical.

The coordinate vt,i's are much less than 1 for most commonly used network architectures. See Figure

in the supplementary materials.

This does not mean the learning rate decay schedule weakens adaptive gradient method. On the contrary, applying the learning rate decay schedule still gives performance boost to the adaptive gradient methods in general but this performance boost is not as significant as SGD + momentum.

The code is available in the anonymous link

For CIFAR experiments, we change the ending two fully-connected layers from 2048 nodes to 512 nodes. For ImageNet experiments, we use batch normalized version (vgg16 bn) provided in Pytorch official package

Figures and Tables:

Figure 1: Performance comparison of Padam with different choices of p for training ResNet on CIFAR-10 / CIFAR-100.

Figure 2: Train loss and test error (top-1) plots of three CNN architectures on CIFAR-10 dataset.

Figure 4: Test perplexity for 2-layer and 3-layer LSTM model on Penn Treebank dataset.

Scalars are denoted by lower case letters, vectors by lower case bold face letters, and matrices by upper case bold face letters. For a vector x ∈ R d , we denote the 2 norm of x by

Table content: <table><tr><td>Chen et al. (2018) provided a convergence</td></tr><tr><td>analysis for a class of Adam-type algorithms for nonconvex optimization. Zou &amp; Shen (2018) ana-</td></tr><tr><td>lyzed the convergence rate of AdaHB and AdaNAG, two modified version of AdaGrad with the use</td></tr><tr><td>of momentum. However, none of these results are directly applicable to Padam. Our convergence</td></tr><tr><td>analysis in Section 4 is quite general and implies the convergence rate of Adam/AMSGrad for non-</td></tr><tr><td>convex optimization. In terms of learning rate decay schedule, Wu et al. (2018) studied the learning</td></tr><tr><td>rate schedule via short-horizon bias. Xu et al. (2016); Davis et al. (2019) analyzed the convergence</td></tr><tr><td>of stochastic algorithms with geometric learning rate decay. Ge et al. (2019) studied the learning</td></tr><tr><td>rate schedule for quadratic functions.</td></tr><tr><td>Notation:</td></tr></table>

) convergence rate under Assumption 4.1, thus the convergence rate of our algorithm is strictly better than that of SGD by a factor of d 1/4 when T is large.

Table content: <table><tr><td>Remark 4.8. If we set α = 1/</td><td>√</td><td>T which is not related to</td><td>d i=1 g 1:T,i 2 , then (4.2) suggests that</td></tr></table>

, which is widely used in Natural Language Processing (NLP) research. Note that word-level PTB dataset does not contain capital letters, numbers, and punctuation. Models are evaluated based on the perplexity metric (lower is better). Plot of max and min values across all coordinates of v t against the iteration number for ResNet model on CIFAR10 dataset.Wide ResNet(Zagoruyko & Komodakis, 2016): Wide Residual Network further exploits the "skip connections" used in ResNet and in the meanwhile increases the width of residual networks. In detail, we use the 16 layer Wide ResNet with 4 multipliers (WRN-16-4) in our experiments.LSTM(Hochreiter & Schmidhuber, 1997): Long Short-Term Memory (LSTM) network is a special kind of Recurrent Neural Network (RNN), capable of learning long-term dependencies. It was introduced byHochreiter & Schmidhuber (1997), and was refined and popularized in many followup work.

Table content: <table><tr><td/><td>2</td><td/></tr><tr><td>Log10 Scale</td><td>6 5 4 3</td><td>v_max v_min</td></tr><tr><td/><td>7</td><td/></tr><tr><td/><td>8</td><td/></tr><tr><td/><td colspan="2">0 10000 20000 30000 40000 50000 60000 70000 80000 #Iterations</td></tr><tr><td>Figure 5:</td><td/><td/></tr><tr><td>C.2 ARCHITECTURES</td><td/><td/></tr><tr><td colspan="3">VGGNet (Simonyan &amp; Zisserman, 2014): We use a modified VGG-16 architecture for this exper-</td></tr><tr><td colspan="3">iment. The VGG-16 network uses only 3 × 3 convolutional layers stacked on top of each other</td></tr><tr><td colspan="3">for increasing depth and adopts max pooling to reduce volume size. Finally, two fully-connected</td></tr><tr><td colspan="2">layers 7 are followed by a softmax classifier.</td><td/></tr><tr><td colspan="3">ResNet (He et al., 2016): Residual Neural Network (ResNet) introduces a novel architecture with</td></tr></table>

Final test accuracy of all algorithms on CIFAR-100 dataset. Bold number indicates the best result.

Table content: <table><tr><td>Models</td><td/><td/><td/><td colspan="2">Test Accuracy (%)</td><td/><td/></tr><tr><td/><td colspan="7">SGDM Adam Amsgrad AdamW Yogi AdaBound Padam</td></tr><tr><td>VGGNet</td><td>73.32</td><td>66.60</td><td>69.40</td><td>73.03</td><td>72.35</td><td>72.00</td><td>74.39</td></tr><tr><td>ResNet</td><td>77.77</td><td>71.72</td><td>72.62</td><td>76.69</td><td>74.55</td><td>75.29</td><td>77.85</td></tr><tr><td colspan="3">WideResNet 76.66 71.83</td><td>73.02</td><td>76.04</td><td>74.47</td><td>73.49</td><td>76.42</td></tr></table>

Final test accuracy on ImageNet dataset

Table content: <table><tr><td>Models</td><td/><td/><td/><td colspan="2">Test Accuracy (%)</td><td/><td/></tr><tr><td/><td colspan="7">SGDM Adam Amsgrad AdamW Yogi AdaBound Padam</td></tr><tr><td>ResNet Top-1</td><td colspan="2">70.23 63.79</td><td>67.69</td><td>67.93</td><td>68.23</td><td>68.13</td><td>70.07</td></tr><tr><td>ResNet Top-5</td><td>89.40</td><td>85.61</td><td>88.15</td><td>88.47</td><td>88.59</td><td>88.55</td><td>89.47</td></tr><tr><td>VGGNet Top-1</td><td>73.93</td><td>69.52</td><td>69.61</td><td>69.89</td><td>71.56</td><td>70.00</td><td>74.04</td></tr><tr><td>VGGNet Top-5</td><td>91.82</td><td>89.12</td><td>89.19</td><td>89.35</td><td>90.25</td><td>89.27</td><td>91.93</td></tr></table>

Final test perplexity (lower is better) on Penn Treebank dataset

Table content: <table><tr><td>Models</td><td/><td/><td/><td colspan="2">Test Perplexity</td><td/><td/></tr><tr><td/><td colspan="7">SGDM Adam Amsgrad AdamW Yogi AdaBound Padam</td></tr><tr><td>2-layer LSTM</td><td>63.37</td><td>61.58</td><td>62.56</td><td>63.93</td><td>64.13</td><td>63.14</td><td>61.53</td></tr><tr><td>3-layer LSTM</td><td>61.22</td><td>60.44</td><td>61.92</td><td>63.24</td><td>60.01</td><td>60.89</td><td>58.48</td></tr></table>

Peer review (iclr2020):
It has been empirically observed that adaptive optimizers such as Adam/Amsgrad lead to worse generalization than SGD + momentum when used to train neural networks. Motivated by this observation the authors suggest Padam, a modification of Adam/Amsgrad. Padam contains a parameter p that, when set to 0 reduces their method to SGD + momentum and when set to 1/2 reduces their method to Adam/Amsgrad. The Padam iteration is roughly where is an exponential moving average of stochastic gradients and is an exponential moving average of the squared gradient. In that way Padam is capable of interpolating between the two methods in order to find a good trade-off between the improved convergence of the one and the improved generalization capability of the other. The authors also suggest an explanation for why the generalization gap of Adam happens: They claim that it is due to the *small-learning rate dilemma* that happens as follows. A small second moment of the stochastic gradients as approximated by (think variance) in can lead to large effective steps in some components. To balance this effect out, Adam needs to choose smaller step sizes than SGD + momentum. If the same learning rate schedule is used based on a smaller base step size, Adam under-trains at the end of training. The authors suggest that Padam with p < 1/2 can use larger learning rates because it does not have as large effective steps if second moments are small. The authors also prove convergence rates for making gradients small with Padam. They use the setting of nonconvex optimization and not online convex regret analysis like the Adam/Amsgrad papers. Finally, the presented experiments using image data sets and standard neural network architectures suggest that Padam indeed shares the advantages of both SGD + momentum and Adam and thus obtains a best of both worlds. I suggest to accept the paper. It introduces an elegant generalization of Amsgrad that appears to be empirically useful in experiments. The experiments seem to be fairly performed (in terms of hyperparam search). The rigorous convergence analysis is laudable although perhaps not as relevant as the practical usefulness of the suggested approach. I do however suggest that some changes be made. 1. The *small learning rate dilemma* phenomenon needs to be more clearly defined and explained. The whole relationship of SGD step size schedules to generalization (e.g. simulated annealing analogy) is certainly nontrivial. I would rather the paper not make some nonrigorous claims if there is no proof. Or state clearly whether something is conjecture or proposition (with proof). 2. Explain why the same learning rate schedule should or is (in practice) used for Adam as for SGD + momentum, as this does not seem to make sense given the aforementioned dilemma. Specific notes / suggestions: - Page 2: *We proposed a novel* -> *We propose a novel* - Page 4: *bridging this generalization gap* *this* does not make sense to me in that context, rather use *the* - Page 5: Figure 1, p = 1/16 seems to still be the most attractive in terms of generalization in the long run. Why would I not want to wait till the model is fully trained? - Page 4: *It is very likely that Adam/Amsgrad is *over-adaptive* This seems to me a strong claim and the explanation that follows it to me is not rigorous enough.