机器学习笔记:学习率预热 warmup

Posted UQI-LIUWJ

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了机器学习笔记:学习率预热 warmup相关的知识,希望对你有一定的参考价值。

1 学习率预热介绍

  • 在mini-batch 梯度下降方法中,如果batch比较大的话,通常需要比较大的学习率
  • 但在一开始的训练中,由于参数是随机初始化的,所以此时的梯度往往也很大
    • 如果此时学习率也很大的话,训练将变得很不稳定
  • ——>为了提高训练的稳定性,我们在最初几轮迭代时,采用较小的学习率,等梯度下降到一定程度之后,再恢复到初始的学习率
    • 这种方法称为学习率预热 learning rate warmup
    • 当预热过程结束的时候,再选择一种学习率衰减的方式来降低学习率

2 逐渐预热 gradual warmup

一种常见的方法是逐渐预热。假设预热的迭代次数为T‘,初始学习率为α0,那么在预热的过程中,每次更新的学习率为

以上是关于机器学习笔记:学习率预热 warmup的主要内容,如果未能解决你的问题,请参考以下文章

warmup与余弦退火学习率

warmup与余弦退火学习率

余弦衰减学习率与linear warmup结合版代码

ng机器学习视频笔记 ——机器学习系统调试(cv查准率与召回率等)

[转帖]阿里的JDK预热warmup过程

神经网络优化中的学习率调整(下)