使用动量的梯度下降法

本文介绍如何使用动量法改进梯度下降算法,通过计算指数加权平均梯度,减少权重更新过程中的振荡,加速收敛。文章详细解释了动量法的工作原理,并给出了具体的计算公式。

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

整理自吴恩达深度学习系列视频:
https://mooc.study.163.com/learn/2001281003?tid=2001391036#/learn/content?type=detail&id=2001702123
In one sentence, the basic idea is to compute an exponentially weighted average of your gradients, and then use that gradient to update your weights instead

指数加权平均参考前一篇博客:https://blog.youkuaiyun.com/Solo95/article/details/84837217

使用动量的梯度下降法

在这里插入图片描述

如图所示,普通的梯度下降法如图中蓝色画线所示,它在接近最优值红点时,会上下摆动,导致不能很快的收敛到红点,而且如果摆动的幅度过大还会导致发散(紫色画线所示),这也是为什么不能采用很大的learning_rate来加快学习速度。

所以我们引入了指数加权平均来计算梯度的平均值,这会抵消大部分梯度的垂直方向上的摆动,同时保留水平方向上的前进速度,使其更快收敛。使用动量的梯度下降法,“动量”,来自对它的物理上的解释,相当于在一个碗里丢一个小球,通过赋予小球动量,使其减少在碗壁上的左右摆动,让它更快到达碗底,。

使用动量的梯度下降法计算方法

在这里插入图片描述
在每次迭代中,我们计算:
vdw=βvdw+(1−β)dWv_{dw}=\beta v_{dw}+(1-\beta)dWvdw=βvdw+(1β)dW 即指数加权平均,下同。
vdb=βvdb+(1−β)dbv_{db}=\beta v_{db}+(1-\beta)dbvdb=βvdb+(1β)db

注意β=0\beta=0β=0时,就退化成了普通的梯度下降。

起始bias修正:
因为我们取vdwv_{dw}vdwvdbv_{db}vdb为零,所以一开始计算出的vdwv_{dw}vdwvdbv_{db}vdb将会小于实际值,为了修正起始阶段这个偏差,使用以下计算方法:
vdw=vdw1−βtv_{dw}=\frac{v_{dw}}{1-\beta^t}vdw=1βtvdw
vdb=vdb1−βtv_{db}=\frac{v_{db}}{1-\beta^t}vdb=1βtvdb

注意随着t增大1−βt1-\beta^t1βt越来越接近1,也就是说修正起的作用越来越小,它只在warm up阶段有效。

更新parameters的过程变为:
W=W−αvdwW = W-\alpha v_{dw}W=Wαvdwb=b−αvdbb = b-\alpha v_{db}b=bαvdb

现在,除了超参数α\alphaα,我们又多出了一个β\betaβ,但β\betaβ一般取0.9,所以你不用担心它的取值问题,你也可以尝试取其他值,但0.9已经被证明很健壮。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值