Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OpenDocCN
d2l-zh
提交
b475d79c
D
d2l-zh
项目概览
OpenDocCN
/
d2l-zh
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
D
d2l-zh
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
b475d79c
编写于
4月 04, 2018
作者:
A
Aston Zhang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
opt boldsym
上级
987bdb2b
变更
6
展开全部
隐藏空白更改
内联
并排
Showing
6 changed file
with
186 addition
and
66 deletion
+186
-66
chapter_optimization/adadelta-scratch.md
chapter_optimization/adadelta-scratch.md
+7
-7
chapter_optimization/adagrad-scratch.md
chapter_optimization/adagrad-scratch.md
+10
-10
chapter_optimization/adam-scratch.md
chapter_optimization/adam-scratch.md
+10
-10
chapter_optimization/gd-sgd-scratch.md
chapter_optimization/gd-sgd-scratch.md
+141
-21
chapter_optimization/momentum-scratch.md
chapter_optimization/momentum-scratch.md
+12
-12
chapter_optimization/rmsprop-scratch.md
chapter_optimization/rmsprop-scratch.md
+6
-6
未找到文件。
chapter_optimization/adadelta-scratch.md
浏览文件 @
b475d79c
# Adadelta --- 从0开始
我们在
[
Adagrad
](
adagrad-scratch.md
)
里提到,由于学习率分母上的变量$
\
m
athbf
{s}$一直在累加按元素平方的梯度,每个元素的学习率在迭代过程中一直在降低或不变。所以在有些问题下,当学习率在迭代早期降得较快时且当前解依然不理想时,Adagrad在迭代后期可能较难找到一个有用的解。我们在
[
RMSProp
](
rmsprop-scratch.md
)
介绍了应对这一问题的一种方法:对梯度按元素平方使用指数加权移动平均而不是累加。
我们在
[
Adagrad
](
adagrad-scratch.md
)
里提到,由于学习率分母上的变量$
\
b
oldsymbol
{s}$一直在累加按元素平方的梯度,每个元素的学习率在迭代过程中一直在降低或不变。所以在有些问题下,当学习率在迭代早期降得较快时且当前解依然不理想时,Adagrad在迭代后期可能较难找到一个有用的解。我们在
[
RMSProp
](
rmsprop-scratch.md
)
介绍了应对这一问题的一种方法:对梯度按元素平方使用指数加权移动平均而不是累加。
事实上,Adadelta也是一种应对这个问题的方法。有意思的是,它没有学习率参数。
## Adadelta算法
Adadelta算法也像RMSProp一样,使用了一个梯度按元素平方的指数加权移动平均变量$
\
m
athbf{s}$,并将其中每个元素初始化为0。在每次迭代中,首先计算
[
小批量梯度
](
gd-sgd-scratch.md
)
$
\m
athbf{g}$,然后对该梯度按元素平方后做指数加权移动平均并计算$
\m
athbf
{s}$:
Adadelta算法也像RMSProp一样,使用了一个梯度按元素平方的指数加权移动平均变量$
\
b
oldsymbol{s}$,并将其中每个元素初始化为0。在每次迭代中,首先计算
[
小批量梯度
](
gd-sgd-scratch.md
)
$
\b
oldsymbol{g}$,然后对该梯度按元素平方后做指数加权移动平均并计算$
\b
oldsymbol
{s}$:
$$
\
m
athbf{s} :=
\r
ho
\m
athbf{s} + (1 -
\r
ho)
\m
athbf{g}
\o
dot
\m
athbf
{g} $$
$$
\
b
oldsymbol{s} :=
\r
ho
\b
oldsymbol{s} + (1 -
\r
ho)
\b
oldsymbol{g}
\o
dot
\b
oldsymbol
{g} $$
然后我们计算当前需要更新的参数的变化量:
$$
\
m
athbf{g}^
\p
rime =
\f
rac{
\s
qrt{
\D
elta
\m
athbf{x} +
\e
psilon}}{
\s
qrt{
\m
athbf{s} +
\e
psilon}}
\o
dot
\m
athbf
{g} $$
$$
\
b
oldsymbol{g}^
\p
rime =
\f
rac{
\s
qrt{
\D
elta
\b
oldsymbol{x} +
\e
psilon}}{
\s
qrt{
\b
oldsymbol{s} +
\e
psilon}}
\o
dot
\b
oldsymbol
{g} $$
其中$
\e
psilon$是为了维持数值稳定性而添加的常数,例如$10^{-5}$。和Adagrad一样,模型参数中每个元素都分别拥有自己的学习率。其中$
\D
elta
\
m
athbf{x}$初始化为零张量,并做如下$
\m
athbf
{g}^
\p
rime$按元素平方的指数加权移动平均:
其中$
\e
psilon$是为了维持数值稳定性而添加的常数,例如$10^{-5}$。和Adagrad一样,模型参数中每个元素都分别拥有自己的学习率。其中$
\D
elta
\
b
oldsymbol{x}$初始化为零张量,并做如下$
\b
oldsymbol
{g}^
\p
rime$按元素平方的指数加权移动平均:
$$
\D
elta
\
m
athbf{x} :=
\r
ho
\D
elta
\m
athbf{x} + (1 -
\r
ho)
\m
athbf{g}^
\p
rime
\o
dot
\m
athbf
{g}^
\p
rime $$
$$
\D
elta
\
b
oldsymbol{x} :=
\r
ho
\D
elta
\b
oldsymbol{x} + (1 -
\r
ho)
\b
oldsymbol{g}^
\p
rime
\o
dot
\b
oldsymbol
{g}^
\p
rime $$
同样地,最后的参数迭代步骤与小批量随机梯度下降类似。只是这里梯度前的学习率已经被调整过了:
$$
\
m
athbf{x} :=
\m
athbf{x} -
\m
athbf
{g}^
\p
rime $$
$$
\
b
oldsymbol{x} :=
\b
oldsymbol{x} -
\b
oldsymbol
{g}^
\p
rime $$
## Adadelta的实现
...
...
chapter_optimization/adagrad-scratch.md
浏览文件 @
b475d79c
...
...
@@ -18,28 +18,28 @@ Adagrad就是一个在迭代过程中不断自我调整学习率,并让模型
## Adagrad算法
由于小批量随机梯度下降包含了梯度下降和随机梯度下降这两种特殊形式,我们在之后的优化章节里提到的梯度都指的是小批量随机梯度。由于我们会经常用到按元素操作,这里稍作介绍。假设$
\
m
athbf
{x} = [4, 9]^
\t
op$,以下是一些按元素操作的例子:
由于小批量随机梯度下降包含了梯度下降和随机梯度下降这两种特殊形式,我们在之后的优化章节里提到的梯度都指的是小批量随机梯度。由于我们会经常用到按元素操作,这里稍作介绍。假设$
\
b
oldsymbol
{x} = [4, 9]^
\t
op$,以下是一些按元素操作的例子:
*
按元素相加: $
\
m
athbf
{x} + 1 = [5, 10]^
\t
op$
*
按元素相乘: $
\
m
athbf{x}
\o
dot
\m
athbf
{x} = [16, 81]^
\t
op$
*
按元素相除: $72 /
\
m
athbf
{x} = [18, 8]^
\t
op$
*
按元素开方: $
\s
qrt{
\
m
athbf
{x}} = [2, 3]^
\t
op$
*
按元素相加: $
\
b
oldsymbol
{x} + 1 = [5, 10]^
\t
op$
*
按元素相乘: $
\
b
oldsymbol{x}
\o
dot
\b
oldsymbol
{x} = [16, 81]^
\t
op$
*
按元素相除: $72 /
\
b
oldsymbol
{x} = [18, 8]^
\t
op$
*
按元素开方: $
\s
qrt{
\
b
oldsymbol
{x}} = [2, 3]^
\t
op$
Adagrad的算法会使用一个梯度按元素平方的累加变量$
\
m
athbf{s}$,并将其中每个元素初始化为0。在每次迭代中,首先计算
[
小批量梯度
](
gd-sgd-scratch.md
)
$
\m
athbf{g}$,然后将该梯度按元素平方后累加到变量$
\m
athbf
{s}$:
Adagrad的算法会使用一个梯度按元素平方的累加变量$
\
b
oldsymbol{s}$,并将其中每个元素初始化为0。在每次迭代中,首先计算
[
小批量梯度
](
gd-sgd-scratch.md
)
$
\b
oldsymbol{g}$,然后将该梯度按元素平方后累加到变量$
\b
oldsymbol
{s}$:
$$
\
m
athbf{s} :=
\m
athbf{s} +
\m
athbf{g}
\o
dot
\m
athbf
{g} $$
$$
\
b
oldsymbol{s} :=
\b
oldsymbol{s} +
\b
oldsymbol{g}
\o
dot
\b
oldsymbol
{g} $$
然后我们将模型参数中每个元素的学习率通过按元素操作重新调整一下:
$$
\
m
athbf{g}^
\p
rime :=
\f
rac{
\e
ta}{
\s
qrt{
\m
athbf{s} +
\e
psilon}}
\o
dot
\m
athbf
{g} $$
$$
\
b
oldsymbol{g}^
\p
rime :=
\f
rac{
\e
ta}{
\s
qrt{
\b
oldsymbol{s} +
\e
psilon}}
\o
dot
\b
oldsymbol
{g} $$
其中$
\e
ta$是初始学习率,$
\e
psilon$是为了维持数值稳定性而添加的常数,例如$10^{-7}$。请注意其中按元素开方、除法和乘法的操作。这些按元素操作使得模型参数中每个元素都分别拥有自己的学习率。
需要强调的是,由于梯度按元素平方的累加变量$
\
m
athbf{s}$出现在分母,Adagrad的核心思想是:如果模型损失函数有关一个参数元素的偏导数一直都较大,那么就让它的学习率下降快一点;反之,如果模型损失函数有关一个参数元素的偏导数一直都较小,那么就让它的学习率下降慢一点。然而,由于$
\m
athbf
{s}$一直在累加按元素平方的梯度,每个元素的学习率在迭代过程中一直在降低或不变。所以在有些问题下,当学习率在迭代早期降得较快时且当前解依然不理想时,Adagrad在迭代后期可能较难找到一个有用的解。
需要强调的是,由于梯度按元素平方的累加变量$
\
b
oldsymbol{s}$出现在分母,Adagrad的核心思想是:如果模型损失函数有关一个参数元素的偏导数一直都较大,那么就让它的学习率下降快一点;反之,如果模型损失函数有关一个参数元素的偏导数一直都较小,那么就让它的学习率下降慢一点。然而,由于$
\b
oldsymbol
{s}$一直在累加按元素平方的梯度,每个元素的学习率在迭代过程中一直在降低或不变。所以在有些问题下,当学习率在迭代早期降得较快时且当前解依然不理想时,Adagrad在迭代后期可能较难找到一个有用的解。
最后的参数迭代步骤与小批量随机梯度下降类似。只是这里梯度前的学习率已经被调整过了:
$$
\
m
athbf{x} :=
\m
athbf{x} -
\m
athbf
{g}^
\p
rime $$
$$
\
b
oldsymbol{x} :=
\b
oldsymbol{x} -
\b
oldsymbol
{g}^
\p
rime $$
...
...
chapter_optimization/adam-scratch.md
浏览文件 @
b475d79c
...
...
@@ -6,39 +6,39 @@ Adam是一个组合了[动量法](momentum-scratch.md)和[RMSProp](rmsprop-scrat
## Adam算法
Adam算法会使用一个动量变量$
\
m
athbf{v}$和一个RMSProp中梯度按元素平方的指数加权移动平均变量$
\m
athbf{s}$,并将它们中每个元素初始化为0。在每次迭代中,首先计算
[
小批量梯度
](
gd-sgd-scratch.md
)
$
\m
athbf
{g}$,并递增迭代次数
Adam算法会使用一个动量变量$
\
b
oldsymbol{v}$和一个RMSProp中梯度按元素平方的指数加权移动平均变量$
\b
oldsymbol{s}$,并将它们中每个元素初始化为0。在每次迭代中,首先计算
[
小批量梯度
](
gd-sgd-scratch.md
)
$
\b
oldsymbol
{g}$,并递增迭代次数
$$t := t + 1$$
然后对梯度做指数加权移动平均并计算动量变量$
\
m
athbf
{v}$:
然后对梯度做指数加权移动平均并计算动量变量$
\
b
oldsymbol
{v}$:
$$
\
m
athbf{v} :=
\b
eta_1
\m
athbf{v} + (1 -
\b
eta_1)
\m
athbf
{g} $$
$$
\
b
oldsymbol{v} :=
\b
eta_1
\b
oldsymbol{v} + (1 -
\b
eta_1)
\b
oldsymbol
{g} $$
该梯度按元素平方后做指数加权移动平均并计算$
\
m
athbf
{s}$:
该梯度按元素平方后做指数加权移动平均并计算$
\
b
oldsymbol
{s}$:
$$
\
m
athbf{s} :=
\b
eta_2
\m
athbf{s} + (1 -
\b
eta_2)
\m
athbf{g}
\o
dot
\m
athbf
{g} $$
$$
\
b
oldsymbol{s} :=
\b
eta_2
\b
oldsymbol{s} + (1 -
\b
eta_2)
\b
oldsymbol{g}
\o
dot
\b
oldsymbol
{g} $$
在Adam算法里,为了减轻$
\
m
athbf{v}$和$
\m
athbf
{s}$被初始化为0在迭代初期对计算指数加权移动平均的影响,我们做下面的偏差修正:
在Adam算法里,为了减轻$
\
b
oldsymbol{v}$和$
\b
oldsymbol
{s}$被初始化为0在迭代初期对计算指数加权移动平均的影响,我们做下面的偏差修正:
$$
\h
at{
\
m
athbf{v}} :=
\f
rac{
\m
athbf
{v}}{1 -
\b
eta_1^t} $$
$$
\h
at{
\
b
oldsymbol{v}} :=
\f
rac{
\b
oldsymbol
{v}}{1 -
\b
eta_1^t} $$
和
$$
\h
at{
\
m
athbf{s}} :=
\f
rac{
\m
athbf
{s}}{1 -
\b
eta_2^t} $$
$$
\h
at{
\
b
oldsymbol{s}} :=
\f
rac{
\b
oldsymbol
{s}}{1 -
\b
eta_2^t} $$
可以看到,当$0
\l
eq
\b
eta_1,
\b
eta_2 < 1$时(算法作者建议分别设为0.9和0.999),当迭代后期$t$较大时,偏差修正几乎就不再有影响。我们使用以上偏差修正后的动量变量和RMSProp中梯度按元素平方的指数加权移动平均变量,将模型参数中每个元素的学习率通过按元素操作重新调整一下:
$$
\
m
athbf{g}^
\p
rime :=
\f
rac{
\e
ta
\h
at{
\m
athbf{v}}}{
\s
qrt{
\h
at{
\m
athbf
{s}} +
\e
psilon}} $$
$$
\
b
oldsymbol{g}^
\p
rime :=
\f
rac{
\e
ta
\h
at{
\b
oldsymbol{v}}}{
\s
qrt{
\h
at{
\b
oldsymbol
{s}} +
\e
psilon}} $$
其中$
\e
ta$是初始学习率,$
\e
psilon$是为了维持数值稳定性而添加的常数,例如$10^{-8}$。和Adagrad一样,模型参数中每个元素都分别拥有自己的学习率。
同样地,最后的参数迭代步骤与小批量随机梯度下降类似。只是这里梯度前的学习率已经被调整过了:
$$
\
m
athbf{x} :=
\m
athbf{x} -
\m
athbf
{g}^
\p
rime $$
$$
\
b
oldsymbol{x} :=
\b
oldsymbol{x} -
\b
oldsymbol
{g}^
\p
rime $$
## Adam的实现
...
...
chapter_optimization/gd-sgd-scratch.md
浏览文件 @
b475d79c
此差异已折叠。
点击以展开。
chapter_optimization/momentum-scratch.md
浏览文件 @
b475d79c
...
...
@@ -7,11 +7,11 @@
## 梯度下降的问题
考虑一个输入为二维向量$
\
m
athbf
{x} = [x_1, x_2]^
\t
op$,输出为标量的目标函数$f:
\m
athbb{R}^2
\r
ightarrow
\m
athbb{R}$。下面为该函数的等高线示意图(每条等高线表示相同函数值的点:越靠近中间函数值越小)。
考虑一个输入为二维向量$
\
b
oldsymbol
{x} = [x_1, x_2]^
\t
op$,输出为标量的目标函数$f:
\m
athbb{R}^2
\r
ightarrow
\m
athbb{R}$。下面为该函数的等高线示意图(每条等高线表示相同函数值的点:越靠近中间函数值越小)。
![](
../img/gd-move.png
)
上图中,红色三角形代表参数$
\
m
athbf
{x}$的初始值。带箭头的线段表示每次迭代时参数的更新。由于目标函数在竖直方向($x_2$轴方向)上比在水平方向($x_1$轴方向)弯曲得更厉害,梯度下降迭代参数时会使参数在竖直方向比在水平方向移动更猛烈。因此,我们需要一个较小的学习率从而避免参数在竖直方向上overshoot。这就造成了上图中参数向最优解移动速度的缓慢。
上图中,红色三角形代表参数$
\
b
oldsymbol
{x}$的初始值。带箭头的线段表示每次迭代时参数的更新。由于目标函数在竖直方向($x_2$轴方向)上比在水平方向($x_1$轴方向)弯曲得更厉害,梯度下降迭代参数时会使参数在竖直方向比在水平方向移动更猛烈。因此,我们需要一个较小的学习率从而避免参数在竖直方向上overshoot。这就造成了上图中参数向最优解移动速度的缓慢。
## 动量法
...
...
@@ -19,30 +19,30 @@
动量法的提出是为了应对梯度下降的上述问题。广义上,以小批量随机梯度下降为例(当批量大小等于训练集大小时,该算法即为梯度下降;批量大小为1即为随机梯度下降),我们对
[
梯度下降和随机梯度下降
](
./gd-sgd-scratch.md
)
的章节中的小批量随机梯度算法做如下修改:
\b
egin{align
*
}
\
m
athbf{v} &:=
\g
amma
\m
athbf{v} +
\e
ta
\n
abla f_
\m
athcal{B}(
\m
athbf
{x}),
\\
\
m
athbf{x} &:=
\m
athbf{x} -
\m
athbf
{v},
\
b
oldsymbol{v} &:=
\g
amma
\b
oldsymbol{v} +
\e
ta
\n
abla f_
\m
athcal{B}(
\b
oldsymbol
{x}),
\\
\
b
oldsymbol{x} &:=
\b
oldsymbol{x} -
\b
oldsymbol
{v},
\e
nd{align
*
}
其中$
\
m
athbf{v}$是当前速度,$
\g
amma$是动量参数。其余符号如学习率$
\e
ta$、有关小批量$
\m
athcal{B}$的随机梯度$
\n
abla f_
\m
athcal{B}(
\m
athbf
{x})$都已在
[
梯度下降和随机梯度下降
](
./gd-sgd-scratch.md
)
的章节中定义,此处不再赘述。
其中$
\
b
oldsymbol{v}$是当前速度,$
\g
amma$是动量参数。其余符号如学习率$
\e
ta$、有关小批量$
\m
athcal{B}$的随机梯度$
\n
abla f_
\m
athcal{B}(
\b
oldsymbol
{x})$都已在
[
梯度下降和随机梯度下降
](
./gd-sgd-scratch.md
)
的章节中定义,此处不再赘述。
当前速度$
\
m
athbf{v}$的更新可以理解为对$[
\e
ta / (1 -
\g
amma)]
\n
abla f_
\m
athcal{B}(
\m
athbf
{x})$做指数加权移动平均。因此,动量法的每次迭代中,参数在各个方向上移动幅度不仅取决当前梯度,还取决过去各个梯度在各个方向上是否一致。当过去的所有梯度都在同一方向,例如都是水平向右,那么参数在水平向右的移动幅度最大。如果过去的梯度中在竖直方向上时上时下,那么参数在竖直方向的移动幅度将变小。这样,我们就可以使用较大的学习率,从而如下图收敛更快。
当前速度$
\
b
oldsymbol{v}$的更新可以理解为对$[
\e
ta / (1 -
\g
amma)]
\n
abla f_
\m
athcal{B}(
\b
oldsymbol
{x})$做指数加权移动平均。因此,动量法的每次迭代中,参数在各个方向上移动幅度不仅取决当前梯度,还取决过去各个梯度在各个方向上是否一致。当过去的所有梯度都在同一方向,例如都是水平向右,那么参数在水平向右的移动幅度最大。如果过去的梯度中在竖直方向上时上时下,那么参数在竖直方向的移动幅度将变小。这样,我们就可以使用较大的学习率,从而如下图收敛更快。
![](
../img/momentum-move.png
)
## 动量参数
为了有助于理解动量参数$
\g
amma$,让我们考虑一个简单的问题:每次迭代的小批量随机梯度$
\n
abla f_
\m
athcal{B}(
\
m
athbf{x})$都等于$
\m
athbf
{g}$。由于所有小批量随机梯度都在同一方向,动量法在该方向使参数移动加速:
为了有助于理解动量参数$
\g
amma$,让我们考虑一个简单的问题:每次迭代的小批量随机梯度$
\n
abla f_
\m
athcal{B}(
\
b
oldsymbol{x})$都等于$
\b
oldsymbol
{g}$。由于所有小批量随机梯度都在同一方向,动量法在该方向使参数移动加速:
\b
egin{align
*
}
\
m
athbf{v}_1 &:=
\e
ta
\m
athbf
{g},
\\
\
m
athbf{v}_2 &:=
\g
amma
\m
athbf{v}_1 +
\e
ta
\m
athbf{g} =
\e
ta
\m
athbf
{g} (
\g
amma + 1),
\\
\
m
athbf{v}_3 &:=
\g
amma
\m
athbf{v}_2 +
\e
ta
\m
athbf{g} =
\e
ta
\m
athbf
{g} (
\g
amma^2 +
\g
amma + 1),
\\
\
b
oldsymbol{v}_1 &:=
\e
ta
\b
oldsymbol
{g},
\\
\
b
oldsymbol{v}_2 &:=
\g
amma
\b
oldsymbol{v}_1 +
\e
ta
\b
oldsymbol{g} =
\e
ta
\b
oldsymbol
{g} (
\g
amma + 1),
\\
\
b
oldsymbol{v}_3 &:=
\g
amma
\b
oldsymbol{v}_2 +
\e
ta
\b
oldsymbol{g} =
\e
ta
\b
oldsymbol
{g} (
\g
amma^2 +
\g
amma + 1),
\\
&
\l
dots
\\
\
m
athbf{v}_{
\i
nf} &:=
\f
rac{
\e
ta
\m
athbf
{g}}{1 -
\g
amma}.
\
b
oldsymbol{v}_{
\i
nf} &:=
\f
rac{
\e
ta
\b
oldsymbol
{g}}{1 -
\g
amma}.
\e
nd{align
*
}
例如,当$
\g
amma = 0.99$, 最终的速度将是学习率乘以相应小批量随机梯度$
\e
ta
\
m
athbf
{g}$的100倍大。
例如,当$
\g
amma = 0.99$, 最终的速度将是学习率乘以相应小批量随机梯度$
\e
ta
\
b
oldsymbol
{g}$的100倍大。
## 算法实现和实验
...
...
chapter_optimization/rmsprop-scratch.md
浏览文件 @
b475d79c
# RMSProp --- 从0开始
我们在
[
Adagrad
](
adagrad-scratch.md
)
里提到,由于学习率分母上的变量$
\
m
athbf
{s}$一直在累加按元素平方的梯度,每个元素的学习率在迭代过程中一直在降低或不变。所以在有些问题下,当学习率在迭代早期降得较快时且当前解依然不理想时,Adagrad在迭代后期可能较难找到一个有用的解。
我们在
[
Adagrad
](
adagrad-scratch.md
)
里提到,由于学习率分母上的变量$
\
b
oldsymbol
{s}$一直在累加按元素平方的梯度,每个元素的学习率在迭代过程中一直在降低或不变。所以在有些问题下,当学习率在迭代早期降得较快时且当前解依然不理想时,Adagrad在迭代后期可能较难找到一个有用的解。
为了应对这一问题,RMSProp算法对Adagrad做了一点小小的修改。我们先给出RMSProp算法。
## RMSProp算法
RMSProp算法会使用一个梯度按元素平方的指数加权移动平均变量$
\
m
athbf{s}$,并将其中每个元素初始化为0。在每次迭代中,首先计算
[
小批量梯度
](
gd-sgd-scratch.md
)
$
\m
athbf{g}$,然后对该梯度按元素平方后做指数加权移动平均并计算$
\m
athbf
{s}$:
RMSProp算法会使用一个梯度按元素平方的指数加权移动平均变量$
\
b
oldsymbol{s}$,并将其中每个元素初始化为0。在每次迭代中,首先计算
[
小批量梯度
](
gd-sgd-scratch.md
)
$
\b
oldsymbol{g}$,然后对该梯度按元素平方后做指数加权移动平均并计算$
\b
oldsymbol
{s}$:
$$
\
m
athbf{s} :=
\g
amma
\m
athbf{s} + (1 -
\g
amma)
\m
athbf{g}
\o
dot
\m
athbf
{g} $$
$$
\
b
oldsymbol{s} :=
\g
amma
\b
oldsymbol{s} + (1 -
\g
amma)
\b
oldsymbol{g}
\o
dot
\b
oldsymbol
{g} $$
然后我们将模型参数中每个元素的学习率通过按元素操作重新调整一下:
$$
\
m
athbf{g}^
\p
rime :=
\f
rac{
\e
ta}{
\s
qrt{
\m
athbf{s} +
\e
psilon}}
\o
dot
\m
athbf
{g} $$
$$
\
b
oldsymbol{g}^
\p
rime :=
\f
rac{
\e
ta}{
\s
qrt{
\b
oldsymbol{s} +
\e
psilon}}
\o
dot
\b
oldsymbol
{g} $$
其中$
\e
ta$是初始学习率,$
\e
psilon$是为了维持数值稳定性而添加的常数,例如$10^{-8}$。和Adagrad一样,模型参数中每个元素都分别拥有自己的学习率。
同样地,最后的参数迭代步骤与小批量随机梯度下降类似。只是这里梯度前的学习率已经被调整过了:
$$
\
m
athbf{x} :=
\m
athbf{x} -
\m
athbf
{g}^
\p
rime $$
$$
\
b
oldsymbol{x} :=
\b
oldsymbol{x} -
\b
oldsymbol
{g}^
\p
rime $$
需要强调的是,RMSProp只在Adagrad的基础上修改了变量$
\
m
athbf
{s}$的更新方法:把累加改成了指数加权移动平均。因此,每个元素的学习率在迭代过程中既可能降低又可能升高。
需要强调的是,RMSProp只在Adagrad的基础上修改了变量$
\
b
oldsymbol
{s}$的更新方法:把累加改成了指数加权移动平均。因此,每个元素的学习率在迭代过程中既可能降低又可能升高。
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录