摘要：二階動量的出現，才意味著自適應學習率優化算法時代的到來。自適應學習率類優化算法為每個參數設定了不同的學習率，在不同維度上設定不同步長，因此其下降方向是縮放過的一階動量方向。

說到優化算法，入門級必從SGD學起，老司機則會告訴你更好的還有AdaGrad / AdaDelta，或者直接無腦用Adam?？墒强纯磳W術界的paper，卻發現一眾大神還在用著入門級的SGD，最多加個Moment或者Nesterov ，還經常會黑一下Adam。比如 UC Berkeley的一篇論文就在Conclusion中寫道：

Despite the fact that our experimental evidence demonstrates that adaptive ?methods are not advantageous for machine learning, the Adam algorithm remains incredibly popular. We are not sure exactly as to why ……

無奈與酸楚之情溢于言表。

這是為什么呢？難道平平淡淡才是真？

1、一個框架回顧優化算法

首先我們來回顧一下各類優化算法。

深度學習優化算法經歷了 SGD -> SGDM -> NAG ->AdaGrad -> AdaDelta -> Adam -> Nadam 這樣的發展歷程。Google一下就可以看到很多的教程文章，詳細告訴你這些算法是如何一步一步演變而來的。在這里，我們換一個思路，用一個框架來梳理所有的優化算法，做一個更加高屋建瓴的對比。

1.1優化算法通用框架

首先定義：待優化參數：w ，目標函數： f(w)，初始學習率 α。

而后，開始進行迭代優化。在每個epoch t：

掌握了這個框架，你可以輕輕松松設計自己的優化算法。

我們拿著這個框架，來照一照各種玄乎其玄的優化算法的真身。步驟3、4對于各個算法都是一致的，主要的差別就體現在1和2上。

2、固定學習率的優化算法

一階動量是各個時刻梯度方向的指數移動平均值，約等于最近 1/(1-β1) 個時刻的梯度向量和的平均值。

也就是說，t 時刻的下降方向，不僅由當前點的梯度方向決定，而且由此前累積的下降方向決定。β1的經驗值為0.9，這就意味著下降方向主要是此前累積的下降方向，并略微偏向當前時刻的下降方向。想象高速公路上汽車轉彎，在高速向前的同時略微偏向，急轉彎可是要出事的。

2.3SGD with Nesterov Acceleration

SGD 還有一個問題是困在局部最優的溝壑里面震蕩。想象一下你走到一個盆地，四周都是略高的小山，你覺得沒有下坡的方向，那就只能待在這里了?？墒侨绻闩郎细叩?，就會發現外面的世界還很廣闊。因此，我們不能停留在當前位置去觀察未來的方向，而要向前一步、多看一步、看遠一些。

NAG全稱Nesterov ?Accelerated Gradient，是在SGD、SGD-M的基礎上的進一步改進，改進點在于步驟1。我們知道在時刻t的主要下降方向是由累積動量決定的，自己的梯度方向說了也不算，那與其看當前梯度方向，不如先看看如果跟著累積動量走了一步，那個時候再怎么走。因此，NAG在步驟1，不計算當前位置的梯度方向，而是計算如果按照累積動量走了一步，那個時候的下降方向：

然后用下一個點的梯度方向，與歷史累積動量相結合，計算步驟2中當前時刻的累積動量。

3、自適應學習率的優化算法

此前我們都沒有用到二階動量。二階動量的出現，才意味著“自適應學習率”優化算法時代的到來。SGD及其變種以同樣的學習率更新每個參數，但深度神經網絡往往包含大量的參數，這些參數并不是總會用得到（想想大規模的embedding）。對于經常更新的參數，我們已經積累了大量關于它的知識，不希望被單個樣本影響太大，希望學習速率慢一些；對于偶爾更新的參數，我們了解的信息太少，希望能從每個偶然出現的樣本身上多學一些，即學習速率大一些。

3.1AdaGrad

怎么樣去度量歷史更新頻率呢？那就是二階動量——該維度上，迄今為止所有梯度值的平方和：

3.2AdaDelta/RMSProp

由于AdaGrad單調遞減的學習率變化過于激進，我們考慮一個改變二階動量計算方法的策略：不累積全部歷史梯度，而只關注過去一段時間窗口的下降梯度。這也就是AdaDelta名稱中Delta的來歷。

修改的思路很簡單。前面我們講到，指數移動平均值大約就是過去一段時間的平均值，因此我們用這一方法來計算二階累積動量：

3.4Nadam

最后是Nadam。我們說Adam是集大成者，但它居然遺漏了Nesterov，這還能忍？必須給它加上，按照NAG的步驟1：

就是Nesterov + Adam = Nadam了。

說到這里，大概可以理解為什么j經常有人說 Adam / Nadam 目前最主流、較好用的優化算法了。新手上路，先拿來一試，收斂速度嗖嗖滴，效果也是杠杠滴。

那為什么Adam還老招人黑，被學術界一頓鄙夷？難道只是為了發paper灌水嗎？簡單看看paper里都在說什么。

4、Adam:可能不收斂

這篇是正在深度學習領域較高級會議之一 ICLR 2018 匿名審稿中的一篇論文《On the Convergence of Adam and Beyond》，探討了Adam算法的收斂性，通過反例證明了Adam在某些情況下可能會不收斂。

從而使得學習率單調遞減。

5、Adam:可能錯過全局最優解

深度神經網絡往往包含大量的參數，在這樣一個維度極高的空間內，非凸的目標函數往往起起伏伏，擁有無數個高地和洼地。有的是高峰，通過引入動量可能很容易越過；但有些是高原，可能探索很多次都出不來，于是停止了訓練。

近期Arxiv上的兩篇文章談到這個問題。

第一篇就是前文提到的吐槽Adam最狠的UC Berkeley的文章《The Marginal Value of Adaptive Gradient Methods in Machine Learning》。文中說到，同樣的一個優化問題，不同的優化算法可能會找到不同的答案，但自適應學習率的算法往往找到非常差的答案（very poor solution）。他們設計了一個特定的數據例子，自適應學習率算法可能會對前期出現的特征過擬合，后期才出現的特征很難糾正前期的擬合效果。但這個文章給的例子很極端，在實際情況中未必會出現。

另外一篇是《Improving Generalization Performance by Switching from Adam to SGD》，進行了實驗驗證。他們CIFAR-10數據集上進行測試，Adam的收斂速度比SGD要快，但最終收斂的結果并沒有SGD好。他們進一步實驗發現，主要是后期Adam的學習率太低，影響了有效的收斂。他們試著對Adam的學習率的下界進行控制，發現效果好了很多。

于是他們提出了一個用來改進Adam的方法：前期用Adam，享受Adam快速收斂的優勢；后期切換到SGD，慢慢尋找最優解。這一方法以前也被研究者們用到，不過主要是根據經驗來選擇切換的時機和切換后的學習率。這篇文章把這一切換過程傻瓜化，給出了切換SGD的時機選擇方法，以及學習率的計算方法，效果看起來也不錯。

這個算法挺有趣，下一篇我們可以來談談，這里先貼個算法框架圖：

6、到底用Adam還是SGD

所以，談到現在，到底Adam好還是SGD好？這可能是很難一句話說清楚的事情。去看學術會議中的各種paper，用SGD的很多，Adam的也不少，還有很多偏愛AdaGrad或者AdaDelta。可能研究員把每個算法都試了一遍，哪個出來的效果好就用哪個了。畢竟paper的重點是突出自己某方面的貢獻，其他方面當然是無所不用其極，怎么能輸在細節上呢？

而從這幾篇怒懟Adam的paper來看，多數都構造了一些比較極端的例子來演示了Adam失效的可能性。這些例子一般過于極端，實際情況中可能未必會這樣，但這提醒了我們，理解數據對于設計算法的必要性。優化算法的演變歷史，都是基于對數據的某種假設而進行的優化，那么某種算法是否有效，就要看你的數據是否符合該算法的胃口了。

算法固然美好，數據才是根本。

另一方面，Adam之流雖然說已經簡化了調參，但是并沒有一勞永逸地解決問題，默認的參數雖然好，但也不是放之四海而皆準。因此，在充分理解數據的基礎上，依然需要根據數據特性、算法特性進行充分的調參實驗。

7、不同算法的核心差異

從第一篇的框架中我們看到，不同優化算法最核心的區別，就是第三步所執行的下降方向：

這個式子中，前半部分是實際的學習率（也即下降步長），后半部分是實際的下降方向。SGD算法的下降方向就是該位置的梯度方向的反方向，帶一階動量的SGD的下降方向則是該位置的一階動量方向。自適應學習率類優化算法為每個參數設定了不同的學習率，在不同維度上設定不同步長，因此其下降方向是縮放過（scaled）的一階動量方向。

由于下降方向的不同，可能導致不同算法到達完全不同的局部最優點?！禔n empirical analysis of the optimization of deep network loss surfaces》 這篇論文中做了一個有趣的實驗，他們把目標函數值和相應的參數形成的超平面映射到一個三維空間，這樣我們可以直觀地看到各個算法是如何尋找超平面上的較低點的。

上圖是論文的實驗結果，橫縱坐標表示降維后的特征空間，區域顏色則表示目標函數值的變化，紅色是高原，藍色是洼地。他們做的是配對兒實驗，讓兩個算法從同一個初始化位置開始出發，然后對比優化的結果?？梢钥吹?，幾乎任何兩個算法都走到了不同的洼地，他們中間往往隔了一個很高的高原。這就說明，不同算法在高原的時候，選擇了不同的下降方向。

8、Adam+SGD組合策略

正是在每一個十字路口的選擇，決定了你的歸宿。如果上天能夠給我一個再來一次的機會，我會對那個女孩子說：SGD！

不同優化算法的優劣依然是未有定論的爭議話題。據我在paper和各類社區看到的反饋，主流的觀點認為：Adam等自適應學習率算法對于稀疏數據具有優勢，且收斂速度很快；但精調參數的SGD（+Momentum）往往能夠取得更好的最終結果。

那么我們就會想到，可不可以把這兩者結合起來，先用Adam快速下降，再用SGD調優，一舉兩得？思路簡單，但里面有兩個技術問題：

什么時候切換優化算法？——如果切換太晚，Adam可能已經跑到自己的盆地里去了，SGD再怎么好也跑不出來了。

切換算法以后用什么樣的學習率？——Adam用的是自適應學習率，依賴的是二階動量的累積，SGD接著訓練的話，用什么樣的學習率？

上一篇中提到的論文 Improving Generalization Performance by Switching from Adam to SGD 提出了解決這兩個問題的思路。

首先來看第二個問題，切換之后的學習率。

SGD下降方向必定可以分解為Adam下降方向及其正交方向上的兩個方向之和，那么其在Adam下降方向上的投影就意味著SGD在Adam算法決定的下降方向上前進的距離，而在Adam下降方向的正交方向上的投影是 SGD 在自己選擇的修正方向上前進的距離。

如果SGD要走完Adam未走完的路，那就首先要接過Adam的大旗——沿著 方向走一步，而后在沿著其正交方向走相應的一步。

這樣我們就知道該如何確定SGD的步長（學習率）了——SGD在Adam下降方向上的正交投影，應該正好等于Adam的下降方向（含步長）。也即：

這里直接復用了Adam的 beta 參數。

然后來看第一個問題，何時進行算法的切換。

9、優化算法的常用tricks

最后，分享一些在優化算法的選擇和使用方面的一些tricks。

首先，各大算法孰優孰劣并無定論。如果是剛入門，優先考慮 SGD+Nesterov Momentum或者Adam.（Standford 231n : The two recommended updates to use are either SGD+Nesterov Momentum or Adam）

選擇你熟悉的算法——這樣你可以更加熟練地利用你的經驗進行調參。

充分了解你的數據——如果模型是非常稀疏的，那么優先考慮自適應學習率的算法。

根據你的需求來選擇——在模型設計實驗過程中，要快速驗證新模型的效果，可以先用Adam進行快速實驗優化；在模型上線或者結果發布前，可以用精調的SGD進行模型的極致優化。

先用小數據集進行實驗。有論文研究指出，隨機梯度下降算法的收斂速度和數據集的大小的關系不大。（The mathematics of stochastic gradient descent are amazingly independent of the training set size. In particular, the asymptotic SGD convergence rates are independent from the sample size. [2]）因此可以先用一個具有代表性的小數據集進行實驗，測試一下較好的優化算法，并通過參數搜索來尋找最優的訓練參數。

考慮不同算法的組合。先用Adam進行快速下降，而后再換到SGD進行充分的調優。切換策略可以參考本文介紹的方法。

數據集一定要充分的打散（shuffle）。這樣在使用自適應學習率算法的時候，可以避免某些特征集中出現，而導致的有時學習過度、有時學習不足，使得下降方向出現偏差的問題。

訓練過程中持續監控訓練數據和驗證數據上的目標函數值以及精度或者AUC等指標的變化情況。對訓練數據的監控是要保證模型進行了充分的訓練——下降方向正確，且學習率足夠高；對驗證數據的監控是為了避免出現過擬合。

制定一個合適的學習率衰減策略?？梢允褂枚ㄆ谒p策略，比如每過多少個epoch就衰減一次；或者利用精度或者AUC等性能指標來監控，當測試集上的指標不變或者下跌時，就降低學習率。

這里只列舉出一些在優化算法方面的trick，如有遺漏，歡迎各位在評論中補充。提前致謝！

神經網絡模型的設計和訓練要復雜得多，initialization, activation, normalization 等等無不是四兩撥千斤，這些方面的技巧我再慢慢寫。

補充：指數移動平均值的偏差修正

參考文獻：

[1] Stanford CS231n Convolutional Neural Networks for Visual Recognition

[2] Stochastic Gradient Descent Tricks.

[3] Efficient BackProp

商業智能與數據分析群

興趣范圍包括各種讓數據產生價值的辦法，實際應用案例分享與討論，分析工具，ETL工具，數據倉庫，數據挖掘工具，報表系統等全方位知識

QQ群：81035754