从梯度下降到AdamW机器学习优化方法详解

1. 梯度下降

在深度学习里，目标函数通常是训练数据集中有关各个样本的损失函数的平均。设 $f_i(x)$ 是有关索引为 $i$ 的训练数据样本的损失函数， $n$ 是训练数据样本数， $x$ 是模型的参数向量，那么目标函数定义为

$$f(\mathbf{x})=\frac{1}{n}\sum _{i=1}^n{f}_i(\mathbf{x})$$

目标函数在 $x$ 处的梯度计算为
$$\nabla f(\mathbf{x})=\frac{1}{n}\sum _{i=1}^n\nabla f_i(\mathbf{x})$$

梯度下降公式为：
$$\mathbf{x}\leftarrow \mathbf{x}-\eta \nabla f(\mathbf{x})$$

如果使⽤用梯度下降，每次要计算所有样本的损失平均，每次自变量迭代的计算开销为 $O(n)$，它随着 $n$ 线性增长。因此，当训练数据样本数很大时，梯度下降每次迭代的计算开销很高

1.1 随机梯度下降

每次选择一个样本来计算梯度，并更新
$$\mathbf{x}\leftarrow \mathbf{x}-\eta \nabla f_i(\mathbf{x})$$

1.2 小批量随机梯度下降

小批量随机梯度下降通过重复采样或者不重复采样 $\mathcal{B}$ 个样本得到一个小批量样本集。
$${\mathbf{g}}_t\leftarrow \nabla f_{{\mathcal{B}}_t}\left({\mathbf{x}}_{t-1}\right)=\frac{1}{|\mathcal{B}|}\sum _{i\in {\mathcal{B}}_t}\nabla f_i\left({\mathbf{x}}_{t-1}\right)$$

$${\mathbf{x}}_t\leftarrow {\mathbf{x}}_{t-1}-{\eta }_t{\mathbf{g}}_t$$

小批量随机梯度在每个迭代周期的耗时介于梯度下降和随机梯度下降的耗时之间。

小结
梯度下降法相关的优化方法容易产生震荡，且容易被困在鞍点，迟迟不能到达全局最优值

2. 动量法

2.1 梯度下降的问题

考虑目标函数为 $f(x)=0.1x_1^2+2x_2^2$，这里 $x_1$的系数和 $x_2$的系数不在同一个数量级，使用学习率为 0.4 时自变量的迭代轨迹如下：

可以看到，同一位置上，目标函数在竖直方向（ $x_2$ 轴⽅方向）比在水平方向（ $x_1$ 轴方向）的斜率的绝对值更大。因此，给定学习率，梯度下降迭代自变量时会使自变量在竖直方向比在水平方向移动幅度更大。那么，我们需要一个较小的学习率从而避免自变量在竖直方向上越过目标函数最优解。然而，这会造成自变量在水平方向上朝最优解移动变慢

我们试着将学习率调得稍大一点，此时自变量在竖直方向不断越过最优解并逐渐发散

2.2 动量法

$$\begin{array}{l}{\mathbf{v}}_t\leftarrow \gamma {\mathbf{v}}_{t-1}+{\eta }_t{\mathbf{g}}_t\\ {\mathbf{x}}_t\leftarrow {\mathbf{x}}_{t-1}-{\mathbf{v}}_t\end{array}$$

其中，$g_t$ 表示一个batch的样本的梯度，在时间步0动量法创建速度 $v_0=0$，${\eta }_t$是学习率，动量超参数 $\gamma$ 满足 $0\le \gamma <1$

图2.1 动量法的迭代轨迹

动量法原理
我们知道 $y_t=\gamma y_{t-1}+(1-\gamma )x_t$ 代表 $y_t$ 是 $x_t$ 的指数加权平均，我们常常将 $y_t$ 看作是对最近 $1/(1-\gamma )$ 个时间步的 $x_t$ 值的加权平均。例如，当 $\gamma =0.95$ 时， $y_t$ 可以被看作对最近20个时间步的 $x_t$ 值的加权平均；

我们对动量法的速度变量做变形
$${\mathbf{v}}_t\leftarrow \gamma {\mathbf{v}}_{t-1}+(1-\gamma )\left(\frac{{\eta }_t}{1-\gamma }{\mathbf{g}}_t\right)$$

相比于小批量随机梯度下降， 动量法在每个时间步的自变量更新量近似于将最近 $1/(1-\gamma )$ 个时间步的普通更新量做了指数加权移动平均后再除以 $1-\gamma$。所以，在动量法中，自变量在各个方向上的移动幅度不仅取决当前梯度，还取决于过去的各个梯度在各个方向上是否一致。

在2.1节中提到的例子，使用动量法后，在x轴方向上(向右)每一步的梯度方向都大致相同（向右）所以指数加权平均后也向右，而在y轴方向上，每一步梯度方向时上时下，方向不同，指数加权平均后总体的跳动幅度就会减小，这样的话，在x轴上更新加快，在y轴上震荡减缓，总体上迭代更新更快

小结
使用动量法梯度方向不变的维度上速度变快，梯度方向有所改变的维度上的更新速度变慢，这样就可以加快收敛并减小震荡。

3 Adagrad

在之前介绍过的优化算法中，目标函数自变量的每一个元素在相同时间步都使⽤用同一个学习率来自我迭代，如下所示:
$$x_1\leftarrow x_1-\eta \frac{\partial f}{\partial x_1},x_2\leftarrow x_2-\eta \frac{\partial f}{\partial x_2}$$

上一节中我们看到当 $x_1$ 和 $x_2$ 的梯度值有较大差别时，需要选择足够小的学习率使得自变量
在梯度值较大的维度上不发散。但这样会导致自变量在梯度值较小的维度上迭代过慢。动量法依赖指数加权移动平均使得自变量的更新方向更加一致，从而降低发散的可能

本节AdaGrad算法，它根据自变量在每个维度的梯度值的大小来调整各个维度上的学习率，从而避免统一的学习率难以适应所有维度的问题

3.1 Adagrad算法

$${\mathbf{s}}_t\leftarrow {\mathbf{s}}_{t-1}+{\mathbf{g}}_t\odot {\mathbf{g}}_t$$

$${\mathbf{x}}_t\leftarrow {\mathbf{x}}_{t-1}-\frac{\eta }{\sqrt{{\mathbf{s}}_t+ϵ}}\odot \mathbf{g}$$

Adagrad算法如上式所示，首先他会使用一个小批量随机梯度 $g_t$ 将元素平方后累加到变量$s_t$ 。其中 $s_0=0$，$\odot$表示按元素相乘；接着，将目标函数自变量中每个元素的学习率通过按元素运算重新调整一下，其中 $\eta$ 是学习率， $ϵ$是非常小的常数，一般为 $10^{-6}$，这里开方、除法和乘法的运算都是按元素运算的。这些按元素运算使得目标函数自变量中每个元素都分别拥有自己的学习率。

小批量随机梯度按元素平方的累加变量 $s_t$ 出现在学习率的分母项中。因此，如果目标函数有关自变量中某个元素的偏导数一直都较大，那么该元素的学习率将下降较快；反之，如果目标函数有关自变量中某个元素的偏导数一直都较小，那么该元素的学习率将下降较慢

缺点
由于 $s_t$ 一直在累加按元素平方的梯度，自变量中每个元素的学习率在迭代过程中一直在降低，当学习率在迭代早期降得较快且当前解依然不佳时， AdaGrad算法在迭代后期由于学习率过小，可能较难找到一个有用的解

小结

• AdaGrad算法在迭代过程不断调整学习率，并让目标函数自变量中每个元素都分别拥有自己的学习率
• 使用AdaGrad算法时，自变量中每个元素的学习率在迭代过程中一直在降低（或不变）

4. RMSProp算法

Adagrad 调整学习率时分母上的变量一直在累加按元素平方的小批量随机梯度，所以目标函数自变量每个元素的学习率在迭代过程中一直在降低（或不变）。因此，当学习率在迭代早期降得较快且当前解依然不佳时， AdaGrad算法在迭代后期由于学习率过小，可能较难找到一个有用的解。RMSprop就是为了解决这个问题

4.1 RMSprop算法

不同于Adagrad算法直接将 $t$ 时刻前所有的梯度$g_t$平方后累加到 $s_t$，RMSProp算法将这些梯度按元素平方做指数加权移动平均。具体来说，给定超参数$0\le \gamma <1$， RMSProp算法在时间步 $t$ 计算
$${\mathbf{s}}_t\leftarrow \gamma {\mathbf{s}}_{t-1}+(1-\gamma ){\mathbf{g}}_t\odot {\mathbf{g}}_t$$

和AdaGrad算法一样， RMSProp算法将目标函数自变量中每个元素的学习率通过按元素运算重新调整，然后更新自变量

$${\mathbf{x}}_t\leftarrow {\mathbf{x}}_{t-1}-\frac{\eta }{\sqrt{s_t+ϵ}}\odot {\mathbf{g}}_t$$

其中 $\eta$ 是学习率，$ϵ$是很小的常数，一般取 $10^{-6}$，因为 $s_t$ 是前 $t$ 个时间步的指数加权平均，因此自变量每个元素的学习率在迭代过程中就不再一直降低

5 AdaDelta算法

AdaDelta算法也针对AdaGrad算法在迭代后期可能较难找到有用解的问题做了改进

AdaDelta算法也像RMSProp算法一样，使用了小批量随机梯度$g_t$按元素平方的指数加权移动平均变量$s_t$

$${\mathbf{s}}_t\leftarrow \rho {\mathbf{s}}_{t-1}+(1-\rho ){\mathbf{g}}_t\odot {\mathbf{g}}_t$$

AdaDelta算法还维护一个额外的状态变量 $\Delta x_t$，其元素同样在时间步0时被初始化为0。我们使用$\Delta x_{t-1}$来计算自变量的变化量：

$${\mathbf{g}}_t^{\prime }\leftarrow \sqrt{\frac{\Delta {\mathbf{x}}_{t-1}+ϵ}{s_t+ϵ}}\odot {\mathbf{g}}_t$$

$${\mathbf{x}}_t\leftarrow {\mathbf{x}}_{t-1}-{\mathbf{g}}_t^{\prime }$$

最后，我们使用$\Delta x_t$来记录自变量变化量$g_t^{\prime }$按元素平方的指数加权移动平均

$$\Delta {\mathbf{x}}_t\leftarrow \rho \Delta {\mathbf{x}}_{t-1}+(1-\rho ){\mathbf{g}}_t^{\prime }\odot {\mathbf{g}}_t^{\prime }$$

可以看到，如不考虑 $ϵ$ 的影响， AdaDelta算法跟RMSProp算法的不同之处在于使用 $\sqrt{\Delta x_{t-1}}$来替代学习率$\eta$

6. Adam算法

Adam算法将动量法和RMSprop结合，Adam 中对一阶动量也是用指数移动平均计算
$${\mathbf{v}}_t\leftarrow {\beta }_1{\mathbf{v}}_{t-1}+\left(1-{\beta }_1\right){\mathbf{g}}_t$$

$${\mathbf{s}}_t\leftarrow {\beta }_2{\mathbf{s}}_{t-1}+\left(1-{\beta }_2\right){\mathbf{g}}_t\odot {\mathbf{g}}_t$$

其中 ${\beta }_1\in [0,1)$，作者建议取0.9 ， ${\beta }_2\in [0,1)$，作者建议取 0.999

当 $t$ 较小时，过去各时间步小批量随机梯度权值之和会较小，例如，当 ${\beta }_1=0.9$ 时，$v_1=0.1g_1$ 。为了消除这样的影响，对于任意时间步$t$ ，我们可以将 $v_t$再除以$1-{\beta }_1^t$ ，从而使过去各时间步小批量随机梯度权值之和为1。这也叫作偏差修正。在Adam算法中，我们对变量 $v_t$ 和 $s_t$均作偏差修正
$${\hat{\mathbf{v}}}_t\leftarrow \frac{{\mathbf{v}}_t}{1-{\beta }_1^t}$$

$${\hat{\mathbf{s}}}_t\leftarrow \frac{{\mathbf{s}}_t}{1-{\beta }_2^t}$$

注意：（1）之所以偏差修正除以$1-{\beta }_1^t$ ， 是因为在时间步$t$，${\mathbf{v}}_t=\left(1-{\beta }_1\right){\sum }_{i=1}^t{\beta }_1^{t-i}{\mathbf{g}}_i$将过去各时间步小批量随机梯度的权值相加，得到$\left(1-{\beta }_1\right){\sum }_{i=1}^t{\beta }_1^{t-i}=1-{\beta }_1^t$

（2）为什么要使用偏差修正？
当 $t$ 较小时，过去各时间步小批量随机梯度权值之和会较小，例如，当 ${\beta }_1=0.9$ 时，$v_1=0.1g_1$ 。此时若不进行偏差修正，直接计算，即：$g_t^{\prime }=\frac{\eta v_1}{\sqrt{s_t}+ϵ}$ 则$g_t^{\prime }$会很小，因为本身$v_1$就变成了原来的1/10， 然后再乘上一个学习率，就会更小

接下来调整每个变量的学习率，并更新变量
$${\mathbf{g}}_t^{\prime }\leftarrow \frac{\eta {\hat{\mathbf{v}}}_t}{\sqrt{{\hat{\mathbf{s}}}_t}+ϵ}$$

$${\mathbf{x}}_t\leftarrow {\mathbf{x}}_{t-1}-{\mathbf{g}}_t^{\prime }$$

小结
(1) Adam 的工作流程

1. 初始化：设置初始学习率、一阶动量的指数衰减率 ${\beta }_1$ 、二阶矩估计的指数衰减率 ${\beta }_2$ 和小数值常数 $ϵ$ 。
2. 计算梯度：计算当前参数对应的梯度。
3. 计算一阶动量和二阶动量：通过指数移动平均方法计算梯度的一阶动量（均值）和二阶动量（方差）。
4. 修正偏差：对一阶矩（动量）和二阶动量的估计进行修正，以减轻因为一开始的估计偏差造成的影响。
5. 更新参数：根据修正后的一阶矩和二阶矩估计以及学习率来更新参数

(2) Adam 的优点

1. 自适应学习率：Adam能通过二介矩估计自适应地调节每个参数的学习率
2. 较快的收敛速度：通过引入一阶动量有助于在参数空间中获得更稳定和一致的更新方向，从而具备较快的收敛速度

7. AdamW

7.1 L2正则和权重衰减

L2正则如下：
$$loss=g(w)+\frac{1}{2}\eta ||w|{|}^2$$
权重衰减如下：
$$w=w-\alpha (g^{\prime }(w)-\eta w)=w-\alpha g^{\prime }(w)-\alpha \eta w$$

其中$\alpha$是学习率，$\eta$是正则化系数，(2)式中每一次更新都会减去一小部分权重， 因此叫做权重衰减

从上面两个式子可知，L2正则和权重衰减的说法对于原版 SGD 是等价的，而当我们使用如 Adam 那样复杂的最优化方法，L2 正则化和权重衰减就会存在很大的不同；因为 Adam 中的 L2 正则化需要添加 $\eta w$ 到梯度中，并分别计算梯度及其平方的移动均值，然后再能更新权重。然而权重衰减方法只是简单地求梯度更新权重，并每次从权重中减去一点

7.2 AdamW算法

使用Adam优化带L2正则的损失并不有效。如果引入L2正则项，在计算梯度的时候会加上对正则项求梯度的结果。那么如果本身比较大的一些权重对应的梯度也会比较大，由于Adam计算步骤中减去项会有除以梯度平方的累积，使得减去项偏小。按常理说，越大的权重应该惩罚越大，但是在Adam并不是这样。而权重衰减对所有的权重都是采用相同的系数进行更新，越大的权重显然惩罚越大。

AdamW算法如下：

$${\mathbf{v}}_t\leftarrow {\beta }_1{\mathbf{v}}_{t-1}+\left(1-{\beta }_1\right){\mathbf{g}}_t$$

$${\mathbf{s}}_t\leftarrow {\beta }_2{\mathbf{s}}_{t-1}+\left(1-{\beta }_2\right){\mathbf{g}}_t\odot {\mathbf{g}}_t$$

$${\hat{\mathbf{v}}}_t\leftarrow \frac{{\mathbf{v}}_t}{1-{\beta }_1^t}$$

$${\hat{\mathbf{s}}}_t\leftarrow \frac{{\mathbf{s}}_t}{1-{\beta }_2^t}$$

$${\mathbf{g}}_t^{\prime }\leftarrow \frac{\eta {\hat{\mathbf{v}}}_t}{\sqrt{{\hat{\mathbf{s}}}_t}+ϵ}$$

$${\mathbf{g}}_t^{\prime }={\mathbf{g}}_t^{\prime }+\delta x_t$$

$${\mathbf{x}}_t\leftarrow {\mathbf{x}}_{t-1}-{\mathbf{g}}_t^{\prime }$$

在Adam中加入L2正则，仅仅把权重的平方加入到损失函数中是不正确的，因为那样会与m和v相互作用，我们想要一种权重衰减的方式，它不会与m和v相互作用，这相当于把权重的平方加入到损失函数中向普通的SGD那样

现在有很多算法都是用AdamW，如BERT，GPT2，Transformer等

小结：
在adam中权重衰减的实现不完全符合正则化的定义，因为Adam计算步骤中有除以梯度平方的累积，使得减去项偏小（而权重衰减的目的是想让减去项较大，从而使得更新后的参数偏小），为了解决这个问题。AdamW通过将权重衰减从梯度计算中分离出来，并仅在参数更新时进行权重衰减，即权重衰减通过直接减去小参数值 ${\theta }_t$ 来实现，而不是通过修改梯度，这种方式更符合正则化的要求，并能够更有效地防止过拟合

8. 指数加权平均

常用的平均数法是：$v_{aver}=\frac{v_1+\dots +v_{100}}{100}$，假设现在有100天的温度值，通过上面的公式就可以直接求出100天的平均值。
指数加权平均本质上就是一种近似求平均的方法，其公式为：

$$v_t=\beta \ast v_{t-1}+(1-\beta ){\theta }_t$$

其中$v_t$代表到第$t$天的平均温度，${\theta }_t$代表第$t$天的温度值，$\beta$代表可调节的超参数

加入 $\beta =0.9$ ，可以得到指数平均公式的如下平均值求法：
$$\begin{array}{l}{v}_t=\beta v_{t-1}+(1-\beta ){\theta }_t\\ v_{100}=0.9v_{99}+0.1{\theta }_{100}\\ v_{99}=0.9v_{98}+0.1{\theta }_{99}\\ v_{98}=0.9v_{97}+0.1{\theta }_{98}\end{array}$$

化简得到如下表达式：
$$\begin{gathered} v_{100}=0.1{\theta }_{100}+0.9v_{99}=0.1{\theta }_{100}+0.9(0.1{\theta }_{99}+0.9v_{98}) \\ =0.1{\theta }_{100}+0.1\ast 0.9\ast {\theta }_{99}+0.1\ast 0.9^2{\theta }_{98}+..... \end{gathered}$$

通过上面表达式，我们可以看到，$V_{100}$ 等于每一个时刻天数的温度值再乘以一个权值。

本质就是以指数式递减加权的移动平均。各数值的权重随时间而指数式递减，越近的数据加权越重，但较旧的数据也给予一定的加权

而在我们上面提到的普通平均数求法，它的每一项的权值都是一样的，如果有n项，权值都为1/n。

指数加权平均的优势：
指数加权平均的求解过程实际上是一个递推的过程，那么这样就会有一个非常大的好处，每当我要求从0到某一时刻n的平均值的时候，并不需要像普通求解平均值那样保留所有的时刻值，类和然后除以n。
而是只需要保留0到n-1时刻的平均值和n时刻的温度值即可。也就是每次只需要保留常数值，然后进行运算即可，这对于深度学习中的海量数据来说，是一个很好的减少内存和空间的做法

---

🤗关注公众号 funNLPer快乐起飞🤪🤪

reference

• [1].动手学深度学习
• [2].跟着代码理解BERT中的优化器AdamW（AdamWeightDecayOptimizer）
• [3].通俗理解指数加权平均
• [4].当前训练神经网络最快的方式：AdamW优化算法+超级收敛