Adam(Adaptive Moment Estimation)
Adam是常见的优化深度神经网络的最优化方法,普通的最优化方法(比如BGD,SGD)面对更为复杂的代价函数往往会陷入局部震荡或者陷入鞍点,难以收敛。于是便引入了 动量(Momentum) 这种物理学概念,让优化方向不只沿着梯度方向进行,还需要参考参数之前一步的运动方向。
SGD with Momentum
对于一个代价函数 L ( w i ) L(w_i) L ( w i ) 来说,普通的 SGD 算法的某一步只需要沿着某一个方向前进一步:
w i : = w i − η ∂ L ∂ w i w_i := w_i – \eta\frac{\partial L}{\partial w_i} w i := w i − η ∂ w i ∂ L
带有动量的 SGD 算法则引入了速度的变化。令 w t w_t w t 为 第 t t t 次后迭代的参数的值,Δ w t \Delta w_t Δ w t 是算上学习率 η \eta η ,w w w 在本次迭代中沿着负梯度方向应该前进的方向与大小,即:
Δ w t = − η ∇ L ( w ) \Delta w_t =- \eta\nabla L(w) Δ w t = − η ∇ L ( w )
则带有动量的 SGD 算法的迭代公式为:
v t + 1 = γ v t + Δ w t w t + 1 = w t + v t + 1 v_{t+1} = \gamma v_t +\Delta w_t\\
w_{t+1} = w_t + v_{t+1} v t + 1 = γ v t + Δ w t w t + 1 = w t + v t + 1
其中 γ \gamma γ 为速度的衰减系数,一般为 0.9。
可以将这种方法比作一个斜坡上的小球,斜坡越陡,他的加速度越大,他的下降也就越快。所以带有动量的 SGD 算法会加速收敛,同时也可以有效的减少参数陷入鞍点的问题,但在优化过程中还有一些震荡的现象。
NAG(Nesterov Accelerated Gradient)
带有动量的 SGD 算法似乎是先计算梯度,再累积速度,这样很有可能会导致下降过头而导致震荡,NAG 算法则与 SGD 算法相反,先预见性的看见了自己之后的位置 w t ′ w_t’ w t ′ ,再进行梯度计算,这样相当于走了两步:
w t ′ = w t + γ v t Δ w t = − η ∇ L ( w t ′ ) v t + 1 = γ v t + Δ w t w t + 1 = w t + v t + 1 w_t’ = w_t + \gamma v_t\\
\Delta w_t =- \eta\nabla L(w_t’)\\
v_{t+1} = \gamma v_t +\Delta w_t\\
w_{t+1} = w_t + v_{t+1} w t ′ = w t + γ v t Δ w t = − η ∇ L ( w t ′ ) v t + 1 = γ v t + Δ w t w t + 1 = w t + v t + 1
根据这篇 博文 的结论,这种走了两步这种方法更像牛顿法的前瞻性,因此比带有动量的 SGD 算法的震荡更小,但 η ∇ L ( w t ′ ) \eta\nabla L(w_t’) η ∇ L ( w t ′ ) 计算会出现问题,需要更多的数学技巧来解决。
Adagrad
通常在深度网络训练到一定程度时,当前的学习率相对与代价函数来说太大以至于不能继续收敛,于是需要手动减小学习率 η \eta η ,而 Adagrad 则是一种可以自动调节学习率大小的优化算法。
假设有一组待优化的参数 w = { w 1 , w 2 , . . . , w n } w=\{w_1, w_2,…,w_n\} w = { w 1 , w 2 , … , w n } ,则参数 w i w_i w i 的梯度方向为:
Δ w i = ∇ w i L ( w ) \Delta w_i =\nabla_{w_i} L(w) Δ w i = ∇ w i L ( w )
那么迭代方程如下:
w i : = w i − η g i + ϵ Δ w i w_i:=w_i-\frac{\eta}{\sqrt{g_{i}+\epsilon} }\Delta w_i w i := w i − g i + ϵ η Δ w i
其中 g i g_i g i 储存着历史梯度 ∇ w i L ( w ) \nabla_{w_i}L(w) ∇ w i L ( w ) 平方和,即:
g i = ∑ t ∇ w i L ( w ( t ) ) 2 g_i = \sum^t \nabla_{ w_i} L(w^{(t)})^2 g i = ∑ t ∇ w i L ( w ( t ) ) 2
可以看出 g i g_i g i 随着训练逐渐递增,则其学习率会逐渐变小,实现了自适应调节学习率。同时,学习率的变化也会与参数直接挂钩,在同一问题中出现次数少的特征优化速率会比出现次数多的特征快得多。
但是这种方法会导致在学习一定时间后学习率会变得无限小,在数据过多或者 batch 较小时,累积的梯度平方和会将优化将陷入停滞。
RMSProp
比起累计历史梯度平方和 g i g_i g i ,RMSProp 算法将其替换为历史梯度平方和的期望。与动量的累积相似,在 t t t 时刻的梯度期望(或者说指数衰减的平均)的更新公式为:
E ( ∇ w i L ( w ( t ) ) 2 ) = γ E ( ∇ w i L ( w ( t − 1 ) ) 2 ) + ( 1 − γ ) ∇ w i L ( w ( t ) ) 2 \mathbb E(\nabla_{ w_i} L(w^{(t)})^2) = \gamma \mathbb E(\nabla_{ w_i} L(w^{(t-1)})^2) + (1-\gamma)\nabla_{ w_i} L(w^{(t)})^2 E ( ∇ w i L ( w ( t ) ) 2 ) = γ E ( ∇ w i L ( w ( t − 1 ) ) 2 ) + ( 1 − γ ) ∇ w i L ( w ( t ) ) 2
可以看到经过一段时间的衰减,离现在时间距离远的梯度信息会慢慢减小,越靠近当前时间,梯度信息占比越大。
上式的期望与动量有什么区别呢?我觉得除了换了一种赋权方式和从梯度转变为了梯度和之外没什么区别。
那么与 Adagrad 算法相似,最终的迭代方程为:
w i : = w i − η E ( ∇ w i L ( w ) 2 ) + ϵ Δ w i w_i:=w_i-\frac{ \eta}{ \sqrt{ \mathbb E(\nabla_{ w_i } L(w)^2)+\epsilon } }\Delta w_i w i := w i − E ( ∇ w i L ( w ) 2 ) + ϵ η Δ w i
通常 γ = 0.9 \gamma = 0.9 γ = 0.9
AdaDelta
AdaDelta 算法与上面的 RMSProp 算法同时被提出。AdaDelta 算法不再拥有学习率,而是将他替换成为
w i : = w i − E ( x ) + ϵ E ( ∇ w i L ( w ) 2 ) + ϵ Δ w i w_i:=w_i-\sqrt\frac{ {\mathbb E(x)+\epsilon} }{\mathbb E(\nabla_{ w_i} L(w)^2)+\epsilon}\Delta w_i w i := w i − E ( ∇ w i L ( w ) 2 ) + ϵ E ( x ) + ϵ Δ w i
其中
E ( x ( t + 1 ) ) = γ E ( x ( t ) ) + ( 1 − γ ) ( E ( x ( t ) ) + ϵ E ( ∇ w i L ( w ) 2 ) + ϵ Δ w i 2 ) \mathbb E(x^{(t+1)}) = \gamma \mathbb E(x^{(t)})+(1-\gamma)\left(\frac{ {\mathbb E(x^{(t)})+\epsilon} }{ {\mathbb E(\nabla_{ w_i} L(w)^2)+\epsilon} }\Delta w_i^2\right) E ( x ( t + 1 ) ) = γ E ( x ( t ) ) + ( 1 − γ ) ( E ( ∇ w i L ( w ) 2 ) + ϵ E ( x ( t ) ) + ϵ Δ w i 2 )
这样 AdaDelta 算法就不需要依赖全局学习率而做到真正的自适应啦!
Adam
Adam 采用了一阶梯度矩估计与二阶梯度矩估计的指数衰减均值 m i ( t ) , v i ( t ) m_i^{(t)},v_i^{(t)} m i ( t ) , v i ( t ) 作为计算梯度均值于方差的手段,其中
m i ( t ) = β 1 m i ( t − 1 ) + ( 1 − β 1 ) ∇ L ( w i ) v i ( t ) = β 2 v i ( t − 1 ) + ( 1 − β 2 ) ∇ 2 L ( w i ) m_i^{(t)} = \beta_1m_i^{(t-1)} + (1-\beta_1)\nabla L(w_i)\\
v_i^{(t)} = \beta_2v_i^{(t-1)} + (1-\beta_2)\nabla^2 L(w_i) m i ( t ) = β 1 m i ( t − 1 ) + ( 1 − β 1 ) ∇ L ( w i ) v i ( t ) = β 2 v i ( t − 1 ) + ( 1 − β 2 ) ∇ 2 L ( w i )
m i ( t ) m_i^{(t)} m i ( t ) 可以理解为带有动量的梯度方向,而 v i ( t ) v_i^{(t)} v i ( t ) 来自于 RMSProp 算法,用来调节自适应优化的大小。
在算法的启动阶段,会另 m i ( 0 ) , v i ( 0 ) = 0 m_i^{(0)},v_i^{(0)}=0 m i ( 0 ) , v i ( 0 ) = 0 ,这会导致在开始优化时 m i ( 0 ) , v i ( 0 ) m_i^{(0)},v_i^{(0)} m i ( 0 ) , v i ( 0 ) 值过小,难以启动。则引入偏差修正:
m ^ i ( t ) = m i ( t ) 1 − β 1 t v ^ i ( t ) = v i ( t ) 1 − β 2 t \hat m^{(t)}_i= \frac{m_i^{(t)} }{1-\beta^t_1}\\
\hat v^{(t)}_i= \frac{v_i^{(t)} }{1-\beta^t_2} m ^ i ( t ) = 1 − β 1 t m i ( t ) v ^ i ( t ) = 1 − β 2 t v i ( t )
最终的迭代过程为:
w i ( t + 1 ) = w i ( t ) − η v ^ i ( t ) + ϵ m i ( t ) w_i^{(t+1)} = w_i^{(t)}-\frac{\eta}{\sqrt{\hat v_i^{(t)} }+\epsilon}m_i^{(t)} w i ( t + 1 ) = w i ( t ) − v ^ i ( t ) + ϵ η m i ( t )
为什么需要偏差修正呢?当优化刚刚开始时,假设我们得到初始下降梯度 ∇ L ( w i ) \nabla L(w_i) ∇ L ( w i ) ,由于 m i ( 0 ) = 0 m_i^{(0)} = 0 m i ( 0 ) = 0 ,m i ( 1 ) = 0.1 × ∇ L ( w i ) m_i^{(1)} = 0.1 \times \nabla L(w_i) m i ( 1 ) = 0.1 × ∇ L ( w i ) ,则会使初始优化步长缩小很多,可能会导致优化效果出现偏差,于是需要乘以一个系数来修正这种算法来修正在开始优化时出现的偏差。
如图,黄色的曲线就是没有偏差修正的结果,在前期会与蓝色曲线(E ( ∇ L ( w ) ) \mathbb E(\nabla L(w)) E ( ∇ L ( w )) )产生较大偏差。
我们分解 m ( t ) m^{(t)} m ( t ) ,可以得到:
m ( t ) = ( 1 − β 1 ) ∑ i = 1 t β 1 t − i ∇ L ( w ( i ) ) m^{(t)} = (1-\beta_1)\sum_{i=1}^t\beta_1^{t-i}\nabla L(w^{(i)}) m ( t ) = ( 1 − β 1 ) i = 1 ∑ t β 1 t − i ∇ L ( w ( i ) )
求 m ( t ) m^{(t)} m ( t ) 期望 E ( m ( t ) ) \mathbb E(m^{(t)}) E ( m ( t ) ) :
E ( m ( t ) ) = E ( ( 1 − β 1 ) ∑ i = 1 t β 1 t − i ∇ L ( w ( i ) ) ) = ( 1 − β 1 ) ( β 1 t − 1 E ( ∇ L ( w ( 1 ) ) + β 1 t − 2 E ( ∇ L ( w ( 2 ) ) + ⋯ + E ( ∇ L ( w ( t ) ) ) = ( 1 − β 1 ) ( β 1 t − 1 E ( ∇ L ( w ) + β 1 t − 2 E ( ∇ L ( w ) ) + ⋯ + E ( ∇ L ( w ) ) = ( 1 − β 1 ) ( 1 + β 1 + β 1 2 + ⋯ + β 1 t − 1 ) E ( ∇ L ( w ) ) = ( 1 − β 1 t ) E ( ∇ L ( w ) ) \begin{align}
\mathbb E(m^{(t)}) &= \mathbb E\left((1-\beta_1)\sum_{i=1}^t\beta_1^{t-i}\nabla L(w^{(i)})\right)\\
&=(1-\beta_1)\left(\beta_1^{t-1} \mathbb E\left(\nabla L(w^{(1)}\right)+ \beta_1^{t-2} \mathbb E\left(\nabla L(w^{(2)}\right) + \dots+\mathbb E\left(\nabla L(w^{(t)}\right)\right)\\
&=(1-\beta_1)\left(\beta_1^{t-1} \mathbb E\left(\nabla L(w\right)+ \beta_1^{t-2} \mathbb E\left(\nabla L(w)\right) + \dots+\mathbb E\left(\nabla L(w\right)\right)\\
&= (1-\beta_1)(1+\beta_1+ \beta_1^2 +\dots+\beta_1^{t-1})\mathbb E(\nabla L(w))\\
&= (1-\beta_1^t)\mathbb E(\nabla L(w))
\end{align} E ( m ( t ) ) = E ( ( 1 − β 1 ) i = 1 ∑ t β 1 t − i ∇ L ( w ( i ) ) ) = ( 1 − β 1 ) ( β 1 t − 1 E ( ∇ L ( w ( 1 ) ) + β 1 t − 2 E ( ∇ L ( w ( 2 ) ) + ⋯ + E ( ∇ L ( w ( t ) ) ) = ( 1 − β 1 ) ( β 1 t − 1 E ( ∇ L ( w ) + β 1 t − 2 E ( ∇ L ( w ) ) + ⋯ + E ( ∇ L ( w ) ) = ( 1 − β 1 ) ( 1 + β 1 + β 1 2 + ⋯ + β 1 t − 1 ) E ( ∇ L ( w )) = ( 1 − β 1 t ) E ( ∇ L ( w ))
所以,m m m 是一阶梯度的有偏估计,同理,v v v 是二阶梯度的有偏估计,而随着迭代步数的增加,两者的偏差会越来越小,采用偏差修正会减小因为偏差在优化开始时对迭代方向的干扰。
reference:
1 、2 、3