深度学习中学习率的指数衰减

为什么要使用指数衰减法

在神经网络的参数更新过程中，学习率不能太大也不能太小，太大可能会导致参数在最优值两侧来回移动，太小会大大降低优化速度，为了解决学习率的问题，TensorFlow 提供了一种灵活的学习率设置方法，即指数衰减法。

函数及参数

对应的命令为 tf.train.exponential_decay，通过这个函数，可以先使用较大的学习率来快速得到一个比较优的解，然后随着迭代的继续逐步减小学习率，使得模型在训练后期更加稳定。

这个函数的参数为

• learning_rate —— 事先设定的初始学习率，

• global_step —— 当前迭代轮数

• decay_steps —— 衰减速度，即多少轮可以迭代完一次所有样本数据

• decay_rate —— 衰减系数

• staircase —— boolean 型，默认为 False，暂时不讲它的作用

• name 名字，默认为 None

原理

每迭代一轮，新的学习率都会根据以上参数更新一次，式子如下

decayed_learning_rate = learning_rate * decay_rate ^ (global_step / decay_steps)

式子比较长，不过用到的都是上面介绍的参数，这个式子实质上就是一个指数乘上一个常数，最后得到新一轮的学习率，由于我们的目的是减小学习率，所以根据指数的知识，decay_rate 必须在零到一之间，只有这样，随着指数项即（global_step / decay_steps）的增大，新一轮的学习率才是减小的。

那么为什么这个式子得到的新学习率减小的程度是在降低的呢，我们可以对上式求导，由于上式的右边除了 global_step，其他项都是常数，所以可以把上式简写为

y = a * (b^x)

对其求导结果为

y` = a * ln(b) * (b^x)

由于 b 在零到一之间，所以随着 x 的增大，导数在减小，即新学习率的变化率在减小，也就是衰减的程度在降低。

如此就实现了开头所说的

通过这个函数，可以先使用较大的学习率来快速得到一个比较优的解，然后随着迭代的继续逐步减小学习率，使得模型在训练后期更加稳定。

作者：LiuHDme
链接：https://www.jianshu.com/p/564e9f2c5982
来源：简书
简书著作权归作者所有，任何形式的转载都请联系作者获得授权并注明出处。