400-123-4657

行业资讯 分类
pytorch学习十 ---- 优化器发布日期:2024-03-12 浏览次数:

1、什么是优化器?

首先我们回忆一下机器学习的五大模块:数据、模型、损失函数、优化器、迭代训练

在损失函数中我们会得到一个loss值,即真实标签与预测标签的差异值,对于loss我们通常会采用pytorch中的autograd自动求导机制进行求导,优化器拿到每个参数的导数会根据优化策略去更新我们的模型的参数,并使得模型的loss值呈下降趋势。因此优化器的主要作用是采用梯度去更新我们模型中的可学习参数,使得模型的输出与真实差异更小,即让loss值下降。

pytorch的优化器:管理 更新 模型中可学习参数的值,使得模型输出更接近真实标签。

管理:指优化器去管理哪部分参数,可修改哪部分参数;

更新:优化器中有一些优化的策略,我们采用这些策略去更新模型中可学习参数的值;在神经网络中通常采用梯度下降的策略方法去更新我们的参数。

什么是梯度下降,为什么梯度下降会使得loss值降低?想了解梯度下降就得先了解什么是梯度。梯度是一个向量,梯度的方向是使得方向导数最大的那个方向。那什么是方向导数?什么是导数呢?

导数:函数在指定坐标轴上的变化率;

方向导数:指定方向上的变化率

梯度:一个向量,方向为方向导数取得最大值的方向。

梯度下降:梯度的负方向

pytorch中的优化器

基本属性:

  • defaults:优化器超参数
  • state:参数的缓存,如momentum的缓存
  • param_groups:管理的参数组,为一个list,每个元素为一个字典
  • _step_count:记录更新次数,学习率调整中使用

基本方法:

  • zero_grad():清空所管理参数的梯度
 

pytorch特性:张量梯度不自动清零

  • step():执行一步更新
 
  • add_param_group():添加一组参数到优化器中
 
  • state_dict():获取优化器当前状态信息字典
  • load_state_dict():加载状态信息字典
 

这两个方法主要用在断点时能通常保存的信息直接获取模型最近的状态,而不需要从头开始。load_state_dict是在间隔一定epoch时,保存模型信息,state_dict():获取保存中最新的状态信息。

在人民币分类模型训练中:

 

debug进入optimizer中:

 

最后依据debug进入:

 

这是看下面的defaults:

那下面学习五个方法的具体使用:

首先构建可学习参数:

 

然后构建优化器,将可学习参数放进去,同时设置一个学习率:

 

然后我们采用梯度下降的方法,梯度下降就是让可学习参数朝着参数梯度的负方向更新,即让参数值加上负梯度值。

 

即用梯度-lr*weigt.grad

下面观察zero_grad的使用,zero_grad是清空我们的梯度:

 

这里保存参数是保存参数的地址

下面是add_param_group:

 

 

 

?

优化器Optimizer(二)

学习率 learning rate

梯度下降:w_{i+1} = w_i - g(w_i),其中g(w_i)为其梯度;

首先定义函数,并绘制函数:

 

下面用代码演示梯度下降的过程:

 

从上面可以看出,左边是loss迭代图,从运行结果可知,在0-4次迭代过程中,loss值越来越大;

上面我们采用梯度下降法去更新参数值,然而y并没有减小,反而逐渐增大。

这是因为这里减去的梯度尺度太大了,一般我们会在其g(x_i)前乘以一个系数,来缩小其尺度:

w_{i+1} = LR*g(w_i) , 其中LR为学习率

学习率:learning rate 控制更新的步伐

当我们把学习率设置为0.5时,得到如下结果:

我们发现这个loss值仍然是增大趋势,然而其增加量明显有缩小。

接着我们将学习率设为0.2,得到如下结果:

现在可以发现loss是一个下降的趋势,若我们增加迭代次数,观测loss是否继续下降:

设置迭代次数为20,得到如下结果:

从上可以看出,当迭代次数到达7次左右时,loss几乎趋于直线形式,呈收敛状况,同时loss值也是逐渐下降的。这就是设置一个良好的学习率,可以使得loss值逐渐减小。那应该选什么学习率比较合适呢?下面我们设置多个学习率观测:在0.1-0.5之间,线性设置10个学习率:

 

上面得到了10个不同的学习率情况下10种不同的loss曲线,横轴是迭代次数,纵轴是loss值,我们发现第一条蓝色的学习率为0.5,在迭代到接近40次时loss值出现了激增情况,同时我们也发现loss值出现激增的都是学习率较大的曲线。下面我们参数修改学习率的最大值,修改为0.3:

可以看出学习率为0.3仍出现loss激增,但loss值整体所有下降,因此我们再减小学习率最大值为0.2:

我们可以看出这十条曲线都是呈下降趋势,是我们所想要的情况。

?

通过代码的演示,我们知道通过设置学习率可以使loss值逐渐减小,直到收敛。学习率主要用来控制更新的步伐,不能太大,也不能太小。在优化器中除了学习率还有一个很重要的概念:Momentum(动力,冲量)

Momentum:结合当前梯度与上一次更新信息,用于当前更新

指数加权平均:v_t = \beta *v_{t-1} +(1-\beta)*	heta_t,是在时间序列中常用的用于求解平均值的方法,其思想为:我们求取当前时刻的平均值,距离当前时刻越近的参数值的参考性越大,所占的权重越大,这个权重随着时间间隔的的增大呈指数下降趋势。

v_{100} = \beta*v_{99} +(1-\beta)*	heta_{100}

? ? = (1-\beta)*	heta_{100} +\beta*(\beta*v_{98}+(1-\beta)*	heta_{99})

? ? ? ?= (1-\beta)*	heta_{100} +(1-\beta)*\beta*	heta_{99}+(\beta^2*v_{98})

? ? = (1-\beta)*	heta_{100} +(1-\beta)*\beta*	heta_{99}+(1-\beta)*\beta^2*	heta_{98}+(\beta^3*v_{97})

? ? ?= \sum_i^N(1-\beta)*\beta^i*	heta_{N-i}

 

我们可以发现权重呈一个下降趋势,在近50次左右,权重趋于稳定。距离当前时刻越近,影响越大,权重越大,距离当前时刻越远,影响越小。下面设置不同的beta,观测权重变换曲线:

 

我们发现最上面红色beta为0.8,这也是最小的beta,往下beta逐渐增大,权重逐渐趋向平缓,因此我们可以将beta值理解为记忆周期的概念,beta值越小,记忆周期越短。红色曲线在time为20时,记忆就逐渐消失了;而对于最大的蓝色曲线,它的记忆周期就比较长,八十天的记忆值基本一样。由此可以看出贝塔是用来控制记忆周期,值越大,记忆越久,值越小,记忆越短。通常beta值会设置为0.9,因为该值会更关注当前10天左右的数据。1/(1-beta)

?

pytorch 中加上momentum后的更新公式:

v_i = m*v_{i-1} +g(w_i)

w_{i+1} = w_i -lr*v_i

其中?w_{i+1}为第 i+1次更新的参数,lr为学习率,v_i为更新量,m为momentum系数,g(w_i)w_i的梯度

v_{100} = m*v_{99} +g(v_{100})

? ? ? ?=g(w_{100}) + m*(m*v_{98}+g(w_{99}))

? ? =g(w_{100}) +m*g(w_{99})+m^2*v_{98}

? ? =g(w_{100})+m*g(w_{99})+m^2*g(w_{98})+m^3*v_{97}

下面我们采用两个学习率,一个小的学习率加上momentum,一个较大的学习率不加momentum两种情况下的loss曲线:

首先我们来观测两个学习率都不加momentum的情况:

 

从上面结果可知蓝色曲线是学习率为0.01,Momentum为0,橙色的是学习率为0.03,Momentum为0,我们看到此时0.03收敛更快,由于其学习率比较大,收敛比较快。下面我们将小的学习率加一个momentum为0.9,观测是收敛速度是否会超过大的学习率,得到如下结果:

我们发现此时的收敛曲线会波浪型震荡曲线,逐渐收敛。同时我们可观测到蓝色曲线比橙色曲线在更快地达到了最小的点,只不过蓝色曲线达到最小值后又反弹回去了。这是由于momentum太大了,导致已经接近极小值时,但其更新的梯度很大,因此虽然已达到极小值,但受到前一些时候梯度的影响,一下就反弹回去了。下面我们将momentum设置为0.63,然后观测曲线:

我们发现设置为0.63的时候是一个比较合适的momentum,它会比0.03更快的收敛。

下面我们来学习pytorch提供的最常用,也是最实用的优化器SGD

1. Optim.SGD(params,lr =<object object>,momentum=0,dampening=0,weight_decay=0,nesterov=False)

主要参数:

  • param:管理的参数组
  • lr:初始学习率
  • momentum:动量系数,贝塔
  • weight_decay:L2正则化系数
  • nesterov:是否采用NAG

NAG参考文献:《On the importance of initialization and momentum in deep learning》

pytorch提供的其他优化器

1. optim.SGD:随机梯度下降法

2. optim.Adagrad:自适应学习率梯度下降法

3. optim.RMSprop:Adagrad的改进

4. optim.Adadelta:Adagrad的改进

5. optim.Adam:RMSprop结合Momentum

6. optim.Adamax:Adam增加学习率上限

7. optim.SparseAdam:稀疏版的Adam

8. optim.ASGD:随机平均梯度下降

9. optim.Rprop:弹性反向传播

10. optim.LBFGS:BFGS的改进

Copyright © 2012-2018 天宏-天宏娱乐国际科技集团 版权所有 非商用版本

琼ICP备xxxxxxxx号

平台注册入口