原文题目:An overview of gradient descent optimization algorithms
原文链接:http://sebastianruder.com/optimizing-gradient-descent
博文地址:http://blog.csdn.net/wangxinginnlp/article/details/50974594
梯度下降是最流行的优化算法之中的一个而且眼下为止是优化神经网络最常见的算法。
与此同一时候,每个先进的深度学习库都包括各种算法实现的梯度下降(比方lasagne's, caffe's, 和 keras'的文档)。然而,这些算法常常作为黑盒优化程序使用。所以难以感受到各种算法的好处和不足。
此博文旨在为你提供对不同梯度算法的直观感受,以期会帮助你更好地使用不同的梯度下降算法。
首先,我们会罗列各种梯度下降算法的变种并简单地总结算法训练阶段的挑战。
然后。我们会通过展示解决这个问题的动机和根据这些动机来推导更新法则。以介绍最常见的优化算法。我们也顺带罗列下并行分布式环境下的算法和体系结构。最后,我们会讨论其它有利于梯度下降优化算法的策略。
梯度下降是一种以通过在目标函数梯度 的反向上更新模型參数,来最小化模型參数的目标函数
的方法。学习速率
决定了我们前往(局部)极小值的步长。
换言之,我们沿着目标函数所构造曲面的斜面按向下的方向走动。直到我们到达山谷。假设你对梯度下降不熟悉。你能够点击此处点击打开链接去了解一篇关于优化神经网络的介绍。
梯度下降算法变种
存在三种梯度下降的变种,他们不同之处在于我们在计算目标函数梯度时所用数据量的多少。
根据数据的规模,我们在更新參数的准确性和运行一次更新所用时间之间进行一种折中。
批量梯度下降
普通的梯度下降,也称批量梯度下降,利用全部的训练数据计算目标函数的梯度。
因为我们每进行一次參数更新须要计算总体训练数据的梯度,批量梯度下降会变得非常慢而且一遇到内存吃不下数据就挂了。同一时候批量梯度下降也无法支持模型的在线更新,比如,新的样本不停的到来。
代码中。批量梯度下降大概长这样:
for i in range(nb_epochs):
params_grad = evaluate_gradient(loss_function, data, params)
params = params - learning_rate * params_grad
对于一个预先定义迭代轮数。我们首先以总体数据计算损失函数的梯度向量weights_grad关于參数向量params。值得注意的是先进的深度学习库提供对一些參数进行自己主动求导能够有效地计算梯度。假设你是自己来推梯度。梯度检查是一个不错的注意。(点击点击打开链接查看关于怎样正确地检查梯度的建议)
我们随后以梯度的反方向更新我们的參数。此时学习速率决定着我们每次运行多大的更新。批量梯度下降能够保证在convex error surfaces 条件下取得全局最小值,在non-convex surfaces条件下取得局部极小值。
随机梯度下降
随机梯度下降(SGD)以一个训练例子和标签
进行一次參数更新。
因为在每次參数更新前对相似的例子进行梯度反复计算, 批量梯度下降会在大数据集上进行冗余计算。SGD通过每次计算一个例子的方式避开这样的冗余。
因此。SGD速度会更快并支持在线学习。
SGD频繁的运行更新所伴随的高方差(a high variance)会导致目标函数如图1所看到的的剧烈波动。
图1. SGD波动
批量梯度下降收敛到盆面的极小值,SGD的波动一方面可以使(损失函数)跳到一个全新而且可鞥呢更优的局部极小值,还有一方面这样的波动因为一直overshooting终究会非常难收敛到确切的极小值。
然而,(实验)表明当我们慢慢地减小学习速率时SGD表现出和批量梯度下降相同的收敛行为。差点儿确定地在non-convex and convex optimization中各自收敛到一个局部或者全局极小值在。
SGD的代码片段简单在训练实例上套一个循环并评估每个训练例子的梯度。值得注意的是我们在每轮迭代时候会打乱训练数据。相关解释见点击打开链接
for i in range(nb_epochs):
np.random.shuffle(data)
for example in data:
params_grad = evaluate_gradient(loss_function, example, params)
params = params - learning_rate * params_grad
Mini-batch gradient descent
Mini-batch gradient descent 吸收了上述两个算法的好处。利用小批量训练例子运行一次更新。
以这样的方式。它能够 a) 减小參数更新的方差,导致更平稳的收敛。b) 利用先进深度学习库中常见的高度优化矩阵操作来高效地计算小批量的梯度。普通小批量的规模在50到256之间。但在不同的应用中会变化。Mini-batch gradient desent 是训练神经网络的经典选择。採用mini-bathes时经常也会称为SGD。注意在后文中SGD改进中,为简单起见。我们省略參数。
代码中,我们每轮迭代的mini-bathes规模设置为50。
for i in range(nb_epochs):
np.random.shuffle(data)
for batch in get_batches(data, batch_size=50):
params_grad = evaluate_gradient(loss_function, batch, params)
params = params - learning_rate * params_grad
挑战
然而,普通的mini-batch gradient descent不能保证较好的收敛性,这一点引出了下述挑战:
- 选择一个合适的学习速率是非常难的。学习速率太小会导致收敛慢,太大会阻碍收敛并导致损失函数在极小值周围波动甚至背离。
- 尝试通过设置调度在训练时候调整训练速率,比方,模拟退火依照预先定义好的调度算法或者当相邻的迭代中目标变化小于一个阈值时候减小学习速率。可是这些调度和阈值须要预先设置,无法对数据集特征进行自适应。
- 除此之外。对全部的參数更新都依照同一个学习速率也是一个问题。假设我们的数据非常稀疏而且我们的特征出现的次数不同。我们可能不会希望全部的參数以某种同样的幅度进行更新。而是针对非常少出现的特征进行一次大幅度更新。
- 在神经网络中常见的极小化highly non-convex error functions的一个关键挑战是避免步入大量的suboptimal local minima。
Dauphin等人觉得实践中的困难来自saddle points而非local minima。
所谓saddle points是指那些维度梯度不一致的点。
这些saddle points常常被一个相等误差的平原包围。导致SGD非常难摆脱,由于梯度在全部方向都近似于0。
梯度下降优化算法
我们不会讨论那些实践中对高维数据集合计算上不可行的算法,比方二阶方法(比方牛顿法)。
Momentum
Nesterov accelerated gradient
我们清楚我们会使用momentum term 来更新參数
。计算
会让我们看到更新后參数的近似值(完整的更新还须要考虑梯度),让我们大致知道參数朝那地方更新。我们如今可以通过计算下一个位置參数的梯度(而不是当前位置的參数) 进行提前准备:
我们也希望我们的更新适应于单独的參数。依照參数自身的重要性来进行大幅度或者小幅度的更新。
Adagrad
有趣的是,假设不用平方根操作。算法性能会变差非常多。
Adadelta
时刻 的running average
只依赖于之前average和当前的梯度:
简洁起见,我们从參数更新向量 角度重写普通SGD的參数更新:
为了实现这个目的,他们首先定义了另外一个exponentially decaying average。这一次对更新參数的平方进行操作,而不仅仅是对梯度的平方进行操作:
将曾经的更新规则中的学习速率替换为參数更新的RMS。则得到Adadelta更新规则:
因为Adadelta更新规则中没有了学习速率这一项,我们甚至都不用对学习速率进行设置。
RMSprop
实际上RMSprop和我们在Adadelta中推到的第一个更新向量是同样的:
Adam
他们建议默认设置
算法的可视化
然而,因为NAG拥有通过远眺所提高的警惕,它可以修正他的轨迹并转向极小值。
採用哪种
这么做另外一个优点是你不必去调学习速率。仅用默认值就能够取得最好的结果。
Adam是将bias-correction and momentum增加到RMSprop中。RMSprop、Adadelta和Adam是非常相似的算法而且在相似的环境中性能都不错。
Kingma等人发如今优化后期因为梯度越来越稀疏,bias-correction帮助Adam微弱地击败RMSprop。
综合看来,Adam可能是最佳选择。SGD经常会达到极小值,可是花掉的时间比其它的算法多得多。SGD在更加依赖于鲁棒的初始化和模拟退火(annealing schedule)并可能被saddle points而不是局部极小值困住。
所以,假设你在意高速收敛或者在训练一个非常深非常复杂的神经网络,你应该採用一种自适应学习速率方法。
并行式与分布式SGD
SGD跑起来收敛性好可是在速度非常慢,尤其是在大数据集上。相反,异步方式的SGD速度非常快。可是workers之间次优的通信会导致收敛较差。另外,我们也能够在一台机器上将SGD并行,而不用大的计算集群。以下是一些关于优化并行式和分布式SGD的算法和体系结构。