今天记录一下我花了一些时间才理解的一个概念 —— 指数衰减法（exponential decay）
为什么要使用指数衰减法
在神经网络的参数更新过程中，学习率不能太大也不能太小，太大可能会导致参数在最优值两侧来回移动，太小会大大降低优化速度，为了解决学习率的问题，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 的增大，导数在减小，即新学习率的变化率在减小，也就是衰减的程度在降低。
如此就实现了开头所说的
通过这个函数，可以先使用较大的学习率来快速得到一个比较优的解，然后随着迭代的继续逐步减小学习率，使得模型在训练后期更加稳定。
网友评论