Deep Learning 優化方法總結

查志強發表於2015-08-07

【原文:https://yufeigan.github.io/2014/11/29/Deep-Learning-%E4%BC%98%E5%8C%96%E6%96%B9%E6%B3%95%E6%80%BB%E7%BB%93/

Stochastic Gradient Descent (SGD)

SGD的引數

在使用隨機梯度下降(SGD)的學習方法時,一般來說有以下幾個可供調節的引數:

  • Learning Rate 學習率
  • Weight Decay 權值衰減
  • Momentum 動量
  • Learning Rate Decay 學習率衰減

再此之中只有第一的引數(Learning Rate)是必須的,其餘部分都是為了提高自適應性的引數,也就是說後3個引數不需要時可以設為0。

Learning Rate

學習率決定了權值更新的速度,設定得太大會使結果越過最優值,太小會使下降速度過慢。僅靠人為干預調整引數需要不斷修改學習率,因此後面3種引數都是基於自適應的思路提出的解決方案。
$$w_i←w_i-\eta\frac{\partial{E}}{\partial{w_i}}$$

Weight decay

在實際運用中,為了避免模型的over-fitting,需要對cost function加入規範項,在SGD中我們加入$−ηλw_i$這一項來對cost function進行規範化。
$$w_i←w_i-\eta\frac{\partial{E}}{\partial{w_i}}-\eta\lambda w_i$$
這個公式的基本思路是減小不重要的引數對結果的影響,而有用的權重則不會受到Weight decay的影響,這種思路與Dropout的思路原理上十分相似。

Link 1
Link 2

Learning Rate Decay

一種提高SGD尋優能力的方法,具體做法是每次迭代減小學習率的大小。

  • initial learning rate $\eta=\eta_0$
  • learning rate decay $\eta_d$
  • At each iteration $s$:

$$\eta(s)=\frac{\eta_0}{1+s\cdot{\eta_d}}$$
在許多論文中,另一種比較常見的方法是迭代30-50次左右直接對學習率進行操作($\eta←0.5\cdot\eta$)

Momentum

靈感來自於牛頓第一定律,基本思路是為尋優加入了“慣性”的影響,這樣一來,當誤差曲面中存在平坦區SGD可以一更快的速度學習。
$$w_i←m\cdot w_i-\eta\frac{\partial{E}}{\partial{w_i}}$$
注意:這裡的表示方法並沒有統一的規定,這裡只是其中一種

Link 1
Link 2
Link 3
Link 4

SGD優缺點

  • 實現簡單,當訓練樣本足夠多時優化速度非常快
  • 需要人為調整很多引數,比如學習率,收斂準則等

Averaged Stochastic Gradient Descent (ASGD)

在SGD的基礎上計算了權值的平均值。
$$\bar{w}t=\frac{1}{t-t_0}\sum^t{i=t_0+1} w_t$$

ASGD的引數

在SGD的基礎上增加引數$t_0$

  • 學習率 $\eta$
  • 引數 $t_0$

ASGD優缺點

  • 運算花費和second order stochastic gradient descent (2SGD)一樣小。
  • 比SGD的訓練速度更為緩慢。
  • $t_0$的設定十分困難

Link 1

3. Conjugate Gradient(共軛梯度法)

介於最速下降法與牛頓法之間的一個方法,它僅僅需要利用一階導數的資訊,克服了GD方法收斂慢的特點。

Link 1

Limited-memory Broyden-Fletcher-Goldfarb-Shanno (LBFGS) (一種擬牛頓演算法)

L-BFGS演算法比較適合在大規模的數值計算中,具備牛頓法收斂速度快的特點,但不需要牛頓法那樣儲存Hesse矩陣,因此節省了大量的空間以及計算資源。

Link 1
Link 2
Link 3

應用分析

不同的優化演算法有不同的優缺點,適合不同的場合:

  • LBFGS演算法在引數的維度比較低(一般指小於10000維)時的效果要比SGD(隨機梯度下降)和CG(共軛梯度下降)效果好,特別是帶有convolution的模型。
  • 針對高維的引數問題,CG的效果要比另2種好。也就是說一般情況下,SGD的效果要差一些,這種情況在使用GPU加速時情況一樣,即在GPU上使用LBFGS和CG時,優化速度明顯加快,而SGD演算法優化速度提高很小。
  • 在單核處理器上,LBFGS的優勢主要是利用引數之間的2階近視特性來加速優化,而CG則得得益於引數之間的共軛資訊,需要計算器Hessian矩陣。

Link 1


相關文章