深度学习优化算法大全系列3:NAG(Nesterov Acceleration Gradient)

1.NAG与SGD-M的区别

NAG全称为Nesterov Accelerated Gradient,是在SGD-Momentum基础进一步优化所得。
前面的文章我们提到过,SGD-M主要是利用历史累积动量来代替当前梯度从而达到减小震荡,加速收敛的目的。而NAG与SGD-M的区别在于,在计算当前梯度的时候,先跟随历史累积动量前进一步,然后再看下一步怎么走。

2.算法流程

假定初始参数为 θ \theta θ,初始动量 v v v,动量超参数为 α \alpha α,梯度超参数为 ϵ \epsilon ϵ,则整个NAG算法流程如下:

1.从训练集中采样m个样本 x ( 1 ) x^{(1)} x(1), x ( 2 ) x^{(2)} x(2)…, x ( m ) x^{(m)} x(m),对应的标签为 y ( i ) y^{(i)} y(i)
2.计算当前梯度之前先走一步: θ ^ = θ + α v \hat \theta = \theta + \alpha v θ^=θ+αv
3.计算当前梯度: g = ▽ θ ∑ i L ( f ( x ( i ) ; θ ^ ) , y ( i ) ) g = \bigtriangledown_\theta \sum_i L(f(x^{(i)}; \hat \theta), y^{(i)}) g=θiL(f(x(i);θ^),y(i))
4.计算动量更新: v = α v − ϵ g v = \alpha v - \epsilon g v=αvϵg
5.参数更新 θ = θ + v \theta = \theta + v θ=θ+v

从上面的流程不难看出,NAG与SGD-M的区别就在第二步,其他都一样。

  • 1
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值