深度学习中的优化算法串讲

Arzie100 · 发表于 2022-1-22 11:16

作者 | 谢文睿
来源 | ID：Datawhale
原文 | 深度学习中的优化算法串讲

本文对深度学习中的优化算法进行了梳理。首先，介绍整个优化算法的基本框架；然后将目前用的主流优化算法进行讲解，带领大家了解优化算法从SGD到Adam及Nadam的转变。
基本框架

为什么这些算法是能串讲的呢？因为这些算法都是相通的。为什么是相通的呢？因为所有的优化算法均基于下面优化算法的基本框架。

其中，一阶动量和二阶动量分别是历史梯度的一阶函数和二阶函数。
SGD随机梯度下降算法

首先，讲一下最基础的SGD算法:

缺点：容易陷入局部最优。由于SGD只考虑当前时刻的梯度，在局部最优点的当前梯度为0。由计算公式可知，此时参数不再进行更新，故陷入局部最优的状态。
改进策略及算法

引入历史梯度的一阶动量，代表算法有：Momentum、NAG
引入历史梯度的二阶动量，代表算法有：AdaGrad、RMSProp、AdaDelta
同时引入历史梯度的一阶动量及二阶动量，代表算法有：Adam、Nadam

改进策略1：引入一阶动量

Momentum
改进思想：为抑制SGD的震荡，Momentum认为梯度下降过程可以加入惯性，也就是在SGD基础上引入一阶动。

其中，关于历史梯度的一阶动量的计算采取指数加权移动平均值的思想进行计算，具体计算步骤详见PPT或视频讲解。
NAG
改进思想：除用惯性跳出局部沟壑以外，我们还可以尝试往前看一步。即：在Momentum考虑历史梯度的基础上，把当前梯度转换为未来梯度。

改进策略2：引入二阶动量

二阶动的出现，才意味着“自适应学习率”优化算法时代的到来。在SGD及其引入一阶动量的改进算法中，均已相同的学习率去更新参数。但是，以相同的学习率进行变化经常是不合理的。
在神经网络中，参数需要用不同的学习率进行更新。对于经常新的参数，我们已经积累大关于它的知识，希望被单个样本影响太大，希望学习速率慢一些；对于偶尔新的参数，我们解的信息太少，希望能从每个偶然出现的样本身上多学一些，即学习速率大一些。
以神经网络中的W及b为例，如下图为损失函数图像，W为x轴，b为y轴。发现每次b变动很大，而W每次仅更新一小步。但是，纵观整个损失函数我们发现，W其实可以迈开步子往前走，b是不是可以跳跃不那么大。

于是，出现了不同针对不同维度的参数采用不同学习率的二阶动量改进算法。
AdaGrad
改进思想：从数学的角度来看，新幅度很大的参数，通常历史累计梯度的平方和会很大；相反的，新幅度很小的参数，通常其累计历史梯度的平方和会很小。

缺点：随着时间步的拉长，历史累计梯度平方和会越来越大，这样会使得所有维度参数的学习率都断减小（单调递减），无论新幅度如何。
RMSProp/AdaDelta
改进思想：累积全部历史梯度，而只关注过去一段时间窗口的下降梯度，采用Momentum中的指数加权移动平均值的思。

改进策略3：同时引入一阶动量及二阶动量

Adam
改进思想：加入Momentum的一阶动计算方法及AdaGrad的二阶动计算方法。

Nadam
改进思想：Nadam = Nesterov + Adam。核心在于计算当前时刻的梯度时使用未来梯度。

		自动登录	找回密码
密码			立即注册

深度学习中的优化算法串讲

本帖子中包含更多资源