【cs231n】深度学习优化方法

最新推荐文章于 2025-01-05 21:39:18 发布

原创最新推荐文章于 2025-01-05 21:39:18 发布 · 372 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#sgd #adam

cs231n 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了深度学习中常用的优化方法，包括随机梯度下降（SGD）、minibatch-SGD、学习率衰减、动量更新和Nesterov动量更新。此外，还探讨了逐参数适应学习率的Adagrad、RMSprop和Adam优化算法。这些方法通过动态调整学习率和考虑过去梯度信息，提升了模型训练的效率和准确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

随机梯度下降

minibatch-SGD

最简单的更新形式，沿着梯度负方向改变参数，其中dx由小批量数据求得

x += - learning_rate * dx

学习率衰减

随步数衰减：经过多少步，衰减为之前得0.9
指数衰减： $\alpha = \alpha _ { 0 } e ^ { - k t }$
1/t衰减： $\alpha = \alpha _ { 0 } / ( 1 + k t )$

动量（Momentum）更新

基于动量的随机梯度下降的每次更新，不仅取决于当前的梯度，还取决于过去的梯度大小。当当前的梯度于之前不同时，那么真实的参数更新梯度会变小；相反，当当前梯度于之前相同时，真实参数的更新梯度会变大。在迭代后期，动量法会起到减少震荡，增加稳定性的作用。
$\begin{aligned} v _ { t } & = \gamma v _ { t - 1 } + g _ { t } \\ \theta _ { t } & = \theta _ { t - 1 } - \eta v _ { t } \end{aligned}$

$v_t$ 当前时刻梯度的指数衰减滑动平均
$v_{t-1}$ 上一时刻梯度的指数衰减滑动平均
$g_t$ 当前时刻的梯度
$\gamma$ 动量因子，控制着历史梯度信息对当前时刻梯度指数衰减滑动平均影响的大小

在训练初期， $g_t$ 每次的方向都相同，因此下降速度越快；若 $g_t$ 的方向发生改变，则下降速度会减缓。可以计算出 $v_t$ 将无限趋近于 $\frac { g _ { t } } { 1 - \gamma }$ ，即梯度下降的过程是先加速再匀速。因此在调整参数时，若 $\gamma$ 减小，则学习率应该增大。

在cs231n中对这部分描述为：与上述公式相同

# 动量更新
v = mu * v - learning_rate * dx # 与速度融合
x += v # 与位置融合

Nesterov动量更新

如下图所示，普通的动量更新是将动量和当时梯度进行求和而得到最后参数更新梯度。Nesterov动量更新则是先根据动量进行位移，再在该位置计算梯度，如下图所示。
在这里插入图片描述
代码如下：

x_ahead = x + mu * v
# 计算dx_ahead(在x_ahead处的梯度，而不是在x处的梯度)
v = mu * v - learning_rate * dx_ahead
x += v

## 也可以改写为
v_prev = v # 存储备份
v = mu * v - learning_rate * dx # 速度更新保持不变
x += -mu * v_prev + (1 + mu) * v # 位置更新变了形式

逐参数适应学习率方法

在之前的随机梯度下降中，不管用不用动量，所有的参数更新都使用相同的学习率。而这部分的方法会对学习率进行逐参数适应，也就是说每个参数的学习率都会不同。

Adagrad

$\odot { g }$ $\eta } = \frac { \eta } { \sqrt {{ G } + \epsilon } }$ $\eta } { g }$

变量 ${ G }$ 的大小和 ${ g }$ 是一样的，是对每个参数的梯度的平方和的累加。

优点：在迭代的过程中不断自我调整学习率。如果模型中某个参数的梯度一直都很大，那么该参数的学习率就会下降的快一些；反之则相反。
缺点：学习率的下降是单调的，随着迭代次数的增加，G越来越大，学习率不断减小。若学习率降得过快，则很难找到最优解。

#首先在计算梯度后，将梯度平方累加
cache += dx**2
#然后根据累加结果对学习率进行微调
learning_rate = learning_rate/ (np.sqrt(cache) + eps)
# 进行参数更新
x += - learning_rate * dx

RMSprop

Adagrad的缺点就是学习率的更新是单调下降的，RMSprop使用了滑动平均解决了这个问题，使得学习率既可以增大也可以减小。
$\gamma G _ { t - 1 } + ( 1 - \gamma ) g _ { t } \odot g _ { t }$ $\theta _ { t } = \theta _ { t - 1 } - \frac { \eta } { \sqrt { G _ { t } + \epsilon } } \odot g _ { t }$

$G _ { t }$ 指梯度平方的滑动平均
$g_ { t }$ 指当前的梯度
$\gamma$ 是衰减率，控制着历史梯度信息滑动平均的长度范围，一般取[0.9,0.99,0.999]。

cache =  decay_rate * cache + (1 - decay_rate) * dx**2
learning_rate = learning_rate / (np.sqrt(cache) + eps)
x += - learning_rate * dx

Adam

Adam可以看作是动量与RMSprop的结合，使模型不仅可以对学习率进行逐参数优化，而且使用动量来调整参数梯度的方向。
$\beta _ { 1 } M _ { t - 1 } + \left( 1 - \beta _ { 1 } \right) g _ { t }$ $\beta _ { 2 } G _ { t - 1 } + \left( 1 - \beta _ { 2 } \right) g _ { t } \odot g _ { t }$ $\hat { M } _ { t } : = \frac { M _ { t } } { 1 - \beta _ { 1 } ^ { t } }$ $\hat { G } _ { t } : = \frac { G _ { t } } { 1 - \beta _ { 2 } ^ { t } }$ $\theta _ { t } = \theta _ { t - 1 } - \frac { \eta \hat { M } _ { t } } { \sqrt { \hat { G } _ { t } + \epsilon } }$

$M _ { t }$ 是指经过动量调整后的梯度值，可以看作是梯度的滑动平均，由公式可以看出 $M _ { t }$ 也会受之前梯度的影像
$G _ { t }$ 当前梯度平方的滑动平均，与RMSprop一样
$\beta 1$ $\beta 2$ 指两个滑动平均的衰减率
$\hat { M } _ { t }$ $\hat {G } _ { t }$ 是对之前的偏差修正，减小 $M _ { t }$ $G _ { t }$ 初始化为0时在迭代初期对它们的衰减滑动平均的影响

t+=1 
m = beta1 * m + (1 - beta1) * dw
v = beta2 * v + (1 - beta2) * (dw ** 2)
mb = m/(1-beta1**t)
vb = v/(1-beta2**t)
next_w = w - learning_rate* mb / (np.sqrt(vb) + epsilon)