2018-07-12

在caffe源码的caffe-master/sec/caffe/proto/caffe.proto下记录了不同的学习策略的计算方法:

// The learning rate decay policy. The currently implemented learning rate
  // policies are as follows:
  //    - fixed: always return base_lr.
  //    - step: return base_lr * gamma ^ (floor(iter / step))
  //    - exp: return base_lr * gamma ^ iter
  //    - inv: return base_lr * (1 + gamma * iter) ^ (- power)
  //    - multistep: similar to step but it allows non uniform steps defined by
  //      stepvalue
  //    - poly: the effective learning rate follows a polynomial decay, to be
  //      zero by the max_iter. return base_lr (1 - iter/max_iter) ^ (power)
  //    - sigmoid: the effective learning rate follows a sigmod decay
  //      return base_lr ( 1/(1 + exp(-gamma * (iter - stepsize))))
  //
  // where base_lr, max_iter, gamma, step, stepvalue and power are defined
  // in the solver parameter protocol buffer, and iter is the current iteration.

fixed

参数:

base_lr: 0.01
lr_policy: "fixed"
max_iter: 400000

学习率曲线:


fixed.jpeg

step

参数:

base_lr: 0.01
lr_policy: "step"
gamma: 0.1
stepsize: 30
max_iter: 100

学习率曲线:


step.jpeg

exp

参数:

base_lr: 0.01
lr_policy: "exp"
gamma: 0.1
max_iter: 100

学习率曲线:


image.png

参数 gamma 的值要小于1。当等于1的时候,学习策略变为了 fixed。由exp的学习率计算方式可以看出,在 gamma = 0.1 的情况下,学习率每迭代一次变为上一次迭代的0.1倍。

inv

参数:

base_lr: 0.01
lr_policy: "inv"
gamma: 0.1
power: 0.75
max_iter: 10000

学习率曲线:


inv.jpeg

由上图可以看出,参数 gamma 控制曲线下降的速率,而参数 power 控制曲线在饱和状态下学习率达到的最低值。

multistep

参数:

base_lr: 0.01
lr_policy: "multistep"
gamma: 0.5
stepvalue: 1000
stepvalue: 3000
stepvalue: 4000
stepvalue: 4500
stepvalue: 5000
max_iter: 6000

学习率曲线:


multistep.jpeg

每一次学习率下降到之前的 gamma 倍。

poly

参数:

base_lr: 0.01
lr_policy: "poly"
power: 0.5
max_iter: 10000

学习率曲线:


poly.jpeg

学习率曲线的形状主要由参数 power 的值来控制。当 power = 1 的时候,学习率曲线为一条直线。当 power < 1 的时候,学习率曲线是凸的,且下降速率由慢到快。当 power > 1 的时候,学习率曲线是凹的,且下降速率由快到慢。

sigmoid

参数:

base_lr: 0.01
lr_policy: "sigmoid"
gamma: -0.001
stepsize: 5000
max_iter: 10000

学习率曲线:


sigmoid.jpeg

参数 gamma 控制曲线的变化速率。当 gamma < 0 时,才能控制学习率曲线呈下降趋势,而且 gamma 的值越小,学习率在两头变化越慢,在中间区域变化越快。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 为什么长大后的我们,越来越难交到知心朋友? 「 1 」 我们以前聊过很多次 社会交换理论(Social Excha...
    鞋说歪道阅读 306评论 0 0
  • 我的妻子是一位内向,勤劳,朴实,稍有点远见的农村妇女。 她家的亲戚曾和我聊起她的事,结婚前,有一次,一位...
    香自苦寒阅读 569评论 1 1
  • 错误的记忆方式让你的效率止步不前 在中国的应试教育下,记忆量成为了大部分中国学生想要逃避的问题,想要解决这个问题,...
    不行鸡阅读 527评论 0 2
  • 下午第一节课间,昨天那位同学再一次坐到了读书吧的固定位置。下午的阳光直射在他所在的书桌,我建议他坐到我的对面来,我...
    青眸m阅读 418评论 0 6
  • 一天如果没有坐下来做冥想,似乎缺少了什么,实际上,一天也没什么事情要做,但时间似乎就在不经意间流逝,被谁偷走了呢?...
    春雨_373c阅读 125评论 0 0