随机梯度下降
随机梯度下降法每次采样单个样本来估计当前的梯度,即:
g t = ∇ θ t − 1 f ( θ t − 1 ) Δ θ t = − η g t \begin{aligned} g_{t} &=\nabla_{\theta_{t-1}}f(\theta_{t-1}) \\ \Delta \theta_{t} &= -\eta g_t \end{aligned} gtΔθt=∇θt−1f(θt−1)=−ηgt
其中 η \eta η是学习率, g t g_t gt是SGD完全依赖于当前样本计算的梯度,所以 η \eta η可理解为当前样本多大程度上能影响参数更新。
动量(Momentum)法
momentum模拟物理中动量的概念,累计之前的动量替代之前的梯度。公式如下:
v d W [ l ] = β v d W [ l ] + ( 1 − β ) d W [ l ] W [ l ] = W [ l ] − α v d W [ l ] \begin{aligned} v_{dW^{[l]}} &= \beta v_{dW^{[l]}} + (1-\beta) dW^{[l]} \\ W^{[l]} &= W^{[l]} - \alpha v_{dW^{[l]}} \end{aligned} vdW[l]W[l]=βvdW[l]+(1−β)dW[l]=W[l]−αvdW[l]
其中 α \alpha α时学习率, β \beta β是动量因子。 v d W [ l ] v_{dW^{[l]}} vdW[l]
Nesterov动量法
nesterov项在梯度更新时做了一个校正,避免前进的太快,同时提高灵敏度。将momentum中的公式展开可得:
Δ θ t = − η ∗ μ ∗ m t − 1 − η ∗ g t \Delta \theta_t = -\eta * \mu * m_{t-1} - \eta * g_t Δθt=−η∗μ∗mt−1−η∗gt
可以看出, m t − 1 m_{t-1} mt−1并没有直接改变当前梯度 g t g_t gt,所以Nesterov的改进就是让之前的动量直接影响当前的动量。即:
g t = ∇ θ t − 1 f ( θ t − 1 − η ∗ μ ∗ m t − 1 ) m t = μ ∗ m t − 1 + g t Δ θ t = − η ∗ m t \begin{aligned} g_t & = \nabla_{\theta_{t-1}}f(\theta_{t-1} - \eta * \mu * m_{t-1}) \\ m_t & = \mu * m_{t-1} + g_t \\ \Delta \theta_t & = - \eta * m_t \end{aligned} gtmtΔθt=∇θt−1f(θt−1−η∗μ∗mt−1)=μ∗mt−1+gt=−η∗mt
所以,加了Nesterov项之后,梯度在大的跳跃后,进行计算对当前梯度进行校正。如下图:
momentum首先计算一个梯度(短的蓝色向量),然后在加速更新梯度的方向进行一个大的跳跃(长的蓝色向量),nesterov项首先在之前加速的梯度方向进行一个大的跳跃(棕色向量),计算梯度然后进行校正(绿色梯向量)
AdaGrad
AdaGrad其实是对学习率进行了一个约束。即:
n t = n t − 1 + g t 2 Δ θ t = − η n t + ϵ ∗ g t \begin{aligned} n_t & = n_{t-1} + g_t^2 \\ \Delta \theta_t & = - \frac{\eta}{\sqrt{n_t + \epsilon}} * g_t \end{aligned} ntΔθt=nt−1+gt2=−nt+ϵη∗gt
此处,对 g t g_t gt从1到 t t t进行一个递推形成一个约束项regularizer, − 1 ∑ r = 1 t ( g τ ) 2 + ϵ - \frac{1}{\sqrt{\sum_{r=1}^t(g_\tau)^2 + \epsilon}} −∑r=1t(gτ)2+ϵ1, ϵ \epsilon ϵ用来保证分母非0。
AdaDelta
Adadelta是对Adagrad的扩展,最初方案依然是对学习率进行自适应约束,但是进行了计算上的简化。 Adagrad会累加之前所有的梯度平方,而Adadelta只累加固定大小的项,并且也不直接存储这些项,仅仅是近似计算对应的平均值。即:
n t = ν ∗ n t − 1 + ( 1 − ν ) ∗ g t 2 Δ θ t = − η n t + ϵ ∗ g t \begin{aligned} n_t & = \nu * n_{t-1} + (1 - \nu) * g_t^2 \\ \Delta \theta_t & = - \frac{\eta}{\sqrt{n_t + \epsilon}} * g_t \end{aligned} ntΔθt=ν∗nt−1+(1−ν)∗gt2=−nt+ϵη∗gt
在此处Adadelta其实还是依赖于全局学习率的,但是作者做了一定处理,经过近似牛顿迭代法之后:
E ∣ g 2 ∣ t = ρ ∗ E ∣ g 2 ∣ t − 1 + ( 1 − ρ ) ∗ g t 2 Δ x t = − ∑ r = 1 t − 1 Δ x r E ∣ g 2 ∣ t + ϵ \begin{aligned} E \vert g^2 \vert_t & = \rho * E \vert g^2 \vert_{t-1} + (1 - \rho) * g_t^2 \\ \Delta x_t &= - \frac{\sqrt{\sum_{r=1}^{t-1} \Delta x_r}}{\sqrt{E \vert g^2 \vert_t + \epsilon}} \end{aligned} E∣g2∣tΔxt=ρ∗E∣g2∣t−1+(1−ρ)∗gt2=−E∣g2∣t+ϵ∑r=1t−1Δxr
其中,
E
E
E代表求期望
此时,可以看出Adadelta已经不用依赖全局学习率了。
RMSprop
RMSprop可以算作Adadelta的一个特例:
当
ρ
=
0.5
\rho=0.5
ρ=0.5时,
E
∣
g
2
∣
t
E\vert g^2 \vert_t
E∣g2∣t就变成了梯度平方和的平均数。
如果再求平方根就变成了RMS(均方根):
R M S ∣ g ∣ t = E ∣ g 2 ∣ t + ϵ RMS \vert g \vert_t = \sqrt{E \vert g^2 \vert_t + \epsilon} RMS∣g∣t=E∣g2∣t+ϵ
此时,这个RMS就可以作为学习率 η \eta η的一个约束:
Δ x t = − η R M S ∣ g ∣ t ∗ g t \Delta x_t = - \frac{\eta}{RMS \vert g \vert_t} * g_t Δxt=−RMS∣g∣tη∗gt
Adam
Adam(Adaptive Momentum Estimation)本质上是带有动量项的RMSprop,它利用梯度的一阶矩估计和二阶矩估计动态调整每个参数的学习率。Adam的优点主要在于经过偏置校正后,每一次迭代学习率都有个确定范围,使得参数比较平稳。公式如下:
m t = μ ∗ m t − 1 + ( 1 − μ ) ∗ g t n t = ν ∗ n t − 1 + ( 1 − ν ) ∗ g t 2 m t ^ = m t 1 − μ t n t ^ = n t 1 − ν t Δ θ t = − m t ^ n t ^ + ϵ ∗ η \begin{aligned} m_t &=\mu * m_{t-1} + (1 - \mu) * g_t \\ n_t &=\nu * n_{t-1} + (1 - \nu) * g_t^2 \\ \hat{m_t} &= \frac{m_t}{1-\mu^t} \\ \hat{n_t} &= \frac{n_t}{1-\nu^t} \\ \Delta \theta_t &= -\frac{\hat{m_t}}{\sqrt{\hat{n_t}} + \epsilon} * \eta \end{aligned} mtntmt^nt^Δθt=μ∗mt−1+(1−μ)∗gt=ν∗nt−1+(1−ν)∗gt2=1−μtmt=1−νtnt=−nt^+ϵmt^∗η
其中, m t m_t mt, n t n_t nt分别是对梯度的一阶矩估计和二阶矩估计,可以看作对期望 E ∣ g t ∣ E\vert g_t \vert E∣gt∣, E ∣ g t 2 ∣ E \vert g_t^2 \vert E∣gt2∣的估计; m t ^ \hat{m_t} mt^, n t ^ \hat{n_t} nt^是对 m t m_t mt, n t n_t nt的校正,这样可以近似为对期望的无偏估计。 可以看出,直接对梯度的矩估计对内存没有额外的要求,而且可以根据梯度进行动态调整,而 − m t ^ n t ^ + ϵ -\frac{\hat{m_t}}{\sqrt{\hat{n_t}} + \epsilon} −nt^+ϵmt^对学习率形成一个动态约束,而且有明确的范围。