400-123-4657
联系我们/CONTACT US
地址:广东省广州市天河区88号
电话:400-123-4657
传真:+86-123-4567
公司动态您当前的位置: 首页 > 腾越动态 > 公司动态

pytorch优化器详解:SGD

更新时间:2024-04-15

目录

说明

SGD参数

params

lr

momentum

dampening

weight_decay

nesterov

举例(nesterov为False)

第1轮迭代

第2轮迭代


模型每次反向传导都会给各个可学习参数p计算出一个偏导数g_{t},用于更新对应的参数p。通常偏导数g_{t}不会直接作用到对应的可学习参数p上,而是通过优化器做一下处理,得到一个新的\widehat{g}_t,处理过程用函数F表示(不同的优化器对应的F的内容不同),即\widehat{g}_t=F(g_{t}),然后和学习率lr一起用于更新可学习参数p,即p=p-\widehat{g}_t*lr

SGD是随机梯度下降(stochastic gradient descent)的首字母。

 
 

模型里需要被更新的可学习参数。

学习率。

动量值,通过上一次的v和当前的偏导数g,得到本次的v,即v_{t}=v_{t-1}*momentum+g_{t},这个就是上述的函数F。

动量是物理中的概念,它使v具有惯性,这样可以缓和v的抖动,有时候还可以帮助跳出局部盆地。比如上一次计算得到的v是10,参数更新后,本次的偏导数g是0,那么使用momentum=0.9后,最终用于更新可学习参数的v是10*0.9+0=9,而不是0,这样参数仍会得到较大的更新,就会增大离开当前局部盆地的可能性。

dampening是乘到偏导数g上的一个数,即:v_{t}=v_{t-1}*momentum+g_{t}*(1-dampening)。注意:dampening在优化器第一次更新时,不起作用。

weight_decay的作用是用当前可学习参数p的值修改偏导数,即:g_{t}=g_{t}+(p*weight\_decay),这里待更新的可学习参数p的偏导数就是g_{t}。然后再使用上述公式v_{t}=v_{t-1}*momentum+g_{t}*(1-dampening),计算得到v_{t}

对应的文献还没看,从pytorch源码来看,当nesterov为False时,使用上述公式g_{t}=g_{t}+(p*weight\_decay)v_{t}=v_{t-1}*momentum+g_{t}*(1-dampening)计算得到v{t}

当nesterov为True时,在上述得到的v_{t}的基础上,最终的v_{t}=g_{t}+v_{t}*momentum,即又使用了一次momentum和g_{t}

 
 

w^2的导数是2w,此时w=100,因此g_{1}=200。这里w就是可学习参数p

首先使用weight_decay:g_{1}=g_{1}+(p*weight\_decay)=200+(100*0.01)=201

然后使用momentum(第一次更新不使用dampening):v_{1}=v_{0}*momentum+g_{1}=g_{1}=201。这里v_{0}=0

最后更新w:w=w-lr*v_{1}=100-(0.1*201)=79.9

w^2的导数是2w,此时w=79.9,因此g_{2}=159.8。这里w就是可学习参数p

首先使用weight_decay:g_{2}=g_{2}+(p*weight\_decay)=159.8+(79.9*0.01)=160.599

然后使用momentum和dampening:v_{2}=v_{1}*momentum+g_{2}*(1-dampening),即:v_{2}=201*0.9+160.599*(1-0.5)=261.1995,这里用到的v_{1}是在SGD类中缓存的。

最后更新w:w=w-lr*v_{2}=79.9-(0.1*261.1995)=53.78005

?

【返回列表页】

关于腾越娱乐

腾越最安全、信誉、快捷的娱乐【skype:zhumeng1688】,是亚洲极具公信力的娱乐游戏公司,以互联网技术为基础,引进海外高科技人才,腾越公司自主研发了基于playcraft引擎的H5网页版游戏,致力于带给腾越平台玩家最新颖、最高效的服务与娱乐体验。...

联系我们

电话:400-123-4657

邮箱:admin@youweb.com

地址:广东省广州市天河区88号

传真:+86-123-4567

版权所有:Copyright © 2002-2017 腾越-腾越娱乐-全球注册中心 ICP备案编号:粤IP********** TOP

平台注册入口