ML之梯度下降算法 机器学习初学者的梯度下降算法

[[197951]]

基本概念

本节尝试独立于机器学习算法, 单纯地来讲梯度下降算法 [Gradient Descent, GD], 以使梯度下降更具一般性。

开始之前, 先放 2 个基本概念, 带着这 2 个认识, 在几乎不具备机器学习知识的前提下, 应该也能很好地读懂本节内容:

• 机器学习的主要任务之一, 就是通过训练, 习得一组最优的参数. 常以成本函数 [Cost Function] 作为参数估计的函数, 因此,  机器学习的任务就转变为了最小化成本函数.

• 优化是机器学习算法非常重要的组成部分, 几乎每个机器学习算法都有一个优化算法.

梯度下降算法就是一个被广泛使用的优化算法, 它可以用于寻找最小化成本函数的参数值. 用中学数学的语言来描述梯度下降, 是这样的: 当函数 _  取得最小值时, 求所对应的自变量 的过程_ 此处, 就是机器要学习的参数, 就是用于参数估计的成本函数, 是关于 的函数. 因此, 基本上具备中学数学知识的,  都能理解梯度下降算法.

梯度下降的基本步骤是:

1. 对成本函数进行微分, 得到其在给定点的梯度. 梯度的正负指示了成本函数值的上升或下降:

2. 选择使成本函数值减小的方向, 即梯度负方向, 乘以以学习率 计算得参数的更新量, 并更新参数:

3. 重复以上步骤, 直到取得最小的成本

以上就是梯度下降算法最基础也是最核心的概念, 很简单吧.

下面讲讲梯度下降算法的几个变种, 包括: 批量梯度下降 [Batch Gradient Descent, BGD], 随机梯度下降 [Stochastic  Gradient Descent, SGD], 小批量梯度下降 [Mini-Batch Gradient Descent, MBGD]

算法简介

BGD

BGD 是梯度下降算法最原始的形式, 其特点是每次更新参数 时, 都使用整个训练集的数据.

BGD 的具体实现是这样的:

• 设假设函数为:

  • 所谓假设函数, 就是用于将输入映射为输出的工具, 其返回值也称为估计值

• 设成本函数为:

  • 注意, 才是函数自变量, 是模型输入, 是输入对应的真实值

  • 该成本函数中, 真正有效的部分是 , 前面的 是为后续计算方便添加的

• 对成本函数求导, 需要对每一个参数 分别求偏导, 得到它们各自的梯度:

  • 机器学习模型通常不止一个参数. 成本函数作为参数估计的工具, 要估计每个参数的最优值, 因此需要对每一个参数分别求偏导数

• 每个参数都按梯度负方向进行更新:

因此, BGD 的伪代码形式可以简单地写成:

复制

repeat {            (for every j = 0, 1, .. n)  }

• 1.

• 2.

• 3.

• 4.

• 5.

• 6.

• 7.

上式中的求和部分 就体现了每一次迭代, 都以整个训练集为对象进行梯度计算.

BGD 得到的是全局最优解, 因为它总是以整个训练集来计算梯度, 这是 BGD 的优点. 但也因此带来了巨大的计算量, 计算迭代速度很很慢.

SGD

SGD 每次以一个样本, 而不是整个数据集来计算梯度. 因此, SGD 从成本函数开始, 就不必再求和了, 针对单个样例的成本函数可以写成: (此处的  同样是为了后续计算方便设置的)

于是, SGD 的参数更新规则就可以写成:

SGD 的伪代码形式如下:

复制

repeat {      for i = 1, .., m {                    (for every j = 0, 1, .. n)      }  }

• 1.

• 2.

• 3.

• 4.

• 5.

• 6.

• 7.

• 8.

• 9.

• 10.

• 11.

SGD 的关键点在于以随机顺序选取样本. 因为 SGD 存在局部最优困境, 若每次都以相同的顺序选取样本, 其有很大的可能会在相同的地方陷入局部困境,  或者收敛减缓. 因此, 欲使 SGD 发挥更好的效果, 应充分利用随机化 [Randomise] 带来的优势: 可以在每次迭代之前 (伪代码中最外围循环),  对训练集进行随机排列.

因为每次只取一个样本来进行梯度下降, SGD 的训练速度很快, 但会引入噪声, 使准确度下降. 这意味着并不是每次迭代都向着全局最优而去,  即并不是每次迭代都能使成本函数值降低. 不过换个思路的话, 噪声在一定程度上以使算法避免了局部最优.

SGD 的另一个好处是, 可以使用在线学习 [online learning]. 也就是说, 在模型训练好之后, 只要有新的数据到来,  模型都可以利用新的数据进行再学习, 更新参数,以适应新的变化.

MBGD

MBGD 是为解决 BGD 与 SGD 各自缺点而发明的折中算法, 或者说它利用了 BGD 和 SGD 各自优点. 其基本思想是: 每次更新参数时, 使用  n 个样本, 既不是全部, 也不是 1. (SGD 可以看成是 n=1 的 MBGD 的一个特例)

此处就不再给出 MBGD 的成本函数或其求导公式或参数更新规则公式了, 基本同 BGD, 见上.

MBGD 的伪代码如下:

复制

say b=10, m=1000, repeat {     for i = 1, 11, 21, .., 991 {                  (for every j = 0, 1, .. n)     } }

• 1.

• 2.

• 3.

• 4.

• 5.

• 6.

• 7.

总结一下以上 3 个梯度下降算法的优缺点:

Tips

下面将介绍一些梯度下降算法的使用技巧:

• 挑选合适的学习率 . 最好的选择方法就是监测目标函数值 (本文中就是成本函数值) 随时间的学习曲线.

• 绘制成本-时间的曲线, 观察成本随迭代的变化情况, 即梯度的变化情况. 若成本没有随着迭代而下降, 说明学习率过大了 (发散了).

• 学习率通常是一个很小的实值, 比如 0.1, 0.001, 0.0001. 学习率如果过大, 成本函数可能无法收敛, 一直处于发散状态.

• 学习算法之前, 进行特征缩放能够改善梯度下降. (亲测! 原来使用较大学习率, 成本函数无法收敛; 使用特征缩放之后, 收敛了)

• 使用 MBGD 时, 在学习过程中, 可以逐渐增大小批量的大小, 更好地综合发挥 BGD 与 SGD 的优势.

进阶

以下内容有点深奥, 笔者对一些概念也不甚熟悉, 仅仅是做个记录

我注意到 Keras 实现的 SGD 提供了 3 个关键字参数: decay, momentum, nestrov:

1. decay – 衰减的意思, 表示每一次迭代, 学习率 的衰减量.

2. momentum – 动量的意思, 旨在加速学习, 稍后介绍.

3. nesterov – 算法名, 表示是否使用 Nesterov 动量算法.

使用 BGD 达到极小值时, 整个成本函数的真实梯度会变得很小, 最终减小为 0, 因此 BGD 可以使用固定的学习率; 然而, SGD  中梯度估计引入的噪声源不会在极小点处消失, 因此有必要随着时间的推移逐渐降低学习率, 以保证 SGD 收敛.

实践中, 一般让学习率线性衰减, 直到第 次迭代:

其中, 表示第 k 次迭代, 表示第 次迭代, .

上式中, 需要设置的量包括: 初始学习率 , 最终学习率 以及 终止迭代次数 . 通常取几百的大小, 则设为 的 . 因此,  剩下的主要问题就是选择一个合适的 : 取值太大, 学习曲线会剧烈抖动, 成本会明显增加; 取值太小, 学习过程会变得很缓慢.

上面提到, 在梯度下降中引入动量的概念, 是为了加速学习, 特别是对于处理高曲率 (曲率: 曲线偏离直线的程度), 小但一致的梯度, 或者带噪声的梯度,  有明显的加速效果. 因为动量算法积累了之前梯度指数级衰减的移动平均 (移动平均: 分析时间序列数据的有效工具), 能够继续沿该方向移动,  从而使成本持续减小.

从形式上看, 动量算法引入了 充当速度的角色, 代表参数在参数空间移动的方向和速率. 被设为负梯度的指数衰减平均, 其更新规则如下:

从上式可以得出的一个结论是: 相对于 , 越大, 之前梯度对现在方向的影响就越大.

在引入动量的概念之前, 的更新步长只是梯度范数乘以学习率, 引入动量之后则取决于梯度序列的大小和排列. 当许多连续的梯度指向相同时, 步长最大

Nesterov 动量算法是标准动量算法的变种, 其更新规则如下:

Nesterov 动量算法与标准动量算法的区别在于梯度的计算, 其梯度计算是在施加了当前速度之后, 可以解释为向标准动量方法中添加了一个校正因子.

题外话

研究优化算法的收敛率时, 一般会衡量额外误差: , 即当前成本超出最低可能成本的量. SGD 应用于凸问题 (研究定义于凸集中的凸函数最小化的问题)时,  k 步迭代的额外误差量级是 , 在强凸情况下是 . 除非假定额外条件, 否则不能进一步改进.

Cramer-Rao 界限指出, 泛化误差的下降速度不会快于 . Bottou and Bousquet 因此认为机器学习任务, 不值得探寻收敛快于  的优化算法, 因为:

更快的收敛可能对应过拟合

样例代码

以下是 BGD, SGD, MBGD 的 Python 代码实现, 暂时不包括进阶部分提到的高级内容.

复制

import numpy as np import pylab from sklearn.datasets.samples_generator import make_regression   def bgd(alpha, x, y, numIterations):     """Copied from Internet"""     m = x.shape[0]  # number of samples     theta = np.ones(2)     J_list = []      x_transpose = x.transpose()     for iter in range(0, numIterations):         hypothesis = np.dot(x, theta)         loss = y - hypothesis         J = np.sum(loss ** 2) / (2 * m)  # cost         J_list.append(J)         print("iter %s | J: %.3f" % (iter, J))          gradient = np.dot(x_transpose, loss) / m         theta += alpha * gradient  # update      pylab.plot(range(numIterations), J_list, "k-")     return theta   def sgd(alpha, x, y, num_iter):     """Writtern by kissg"""     m = x.shape[0]  # number of samples     theta = np.ones(2)     J_list = []      # 随机化序列     idx = np.random.permutation(y.shape[0])     x, y = x[idx], y[idx]      for j in range(num_iter):          for i in idx:             single_hypothesis = np.dot(x[i], theta)             single_loss = y[i] - single_hypothesis             gradient = np.dot(x[i].transpose(), single_loss)             theta += alpha * gradient  # update          hypothesis = np.dot(x, theta)         loss = y - hypothesis         J = np.sum(loss ** 2) / (2 * m)  # cost         J_list.append(J)         print("iter %s | J: %.3f" % (j, J))      pylab.plot(range(num_iter), J_list, "r-")     return theta   def mbgd(alpha, x, y, num_iter, minibatches):     """Writtern by kissg"""     m = x.shape[0]  # number of samples     theta = np.ones(2)     J_list = []      for j in range(num_iter):          idx = np.random.permutation(y.shape[0])         x, y = x[idx], y[idx]         mini = np.array_split(range(y.shape[0]), minibatches)          for i in mini:             mb_hypothesis = np.dot(x[i], theta)             mb_loss = y[i] - mb_hypothesis             gradient = np.dot(x[i].transpose(), mb_loss) / minibatches             theta += alpha * gradient  # update          hypothesis = np.dot(x, theta)         loss = y - hypothesis         J = np.sum(loss ** 2) / (2 * m)  # cost         J_list.append(J)         print("iter %s | J: %.3f" % (j, J))      pylab.plot(range(num_iter), J_list, "y-")     return theta   if __name__ == '__main__':      x, y = make_regression(n_samples=100, n_features=1, n_informative=1,                            random_state=0, noise=35)     m, n = np.shape(x)     x = np.c_[np.ones(m), x]  # insert column, bias     alpha = 0.01  # learning rate      pylab.plot(x[:, 1], y, 'o')      print("\n#***BGD***#\n")     theta_bgd = bgd(alpha, x, y, 800)     for i in range(x.shape[1]):         y_bgd_predict = theta_bgd * x     pylab.plot(x, y_bgd_predict, 'k--')      print("\n#***SGD***#\n")     theta_sgd = sgd(alpha, x, y, 10)     for i in range(x.shape[1]):         y_sgd_predict = theta_sgd * x     pylab.plot(x, y_sgd_predict, 'r--')      print("\n#***MBGD***#\n")     theta_mbgd = mbgd(alpha, x, y, 50, 10)     for i in range(x.shape[1]):         y_mbgd_predict = theta_mbgd * x     pylab.plot(x, y_mbgd_predict, 'y--')      pylab.show()     print("Done!")

• 1.

• 2.

• 3.

• 4.

• 5.

• 6.

• 7.

• 8.

• 9.

• 10.

• 11.

• 12.

• 13.

• 14.

• 15.

• 16.

• 17.

• 18.

• 19.

• 20.

• 21.

• 22.

• 23.

• 24.

• 25.

• 26.

• 27.

• 28.

• 29.

• 30.

• 31.

• 32.

• 33.

• 34.

• 35.

• 36.

• 37.

• 38.

• 39.

• 40.

• 41.

• 42.

• 43.

• 44.

• 45.

• 46.

• 47.

• 48.

• 49.

• 50.

• 51.

• 52.

• 53.

• 54.

• 55.

• 56.

• 57.

• 58.

• 59.

• 60.

• 61.

• 62.

• 63.

• 64.

• 65.

• 66.

• 67.

• 68.

• 69.

• 70.

• 71.

• 72.

• 73.

• 74.

• 75.

• 76.

• 77.

• 78.

• 79.

• 80.

• 81.

• 82.

• 83.

• 84.

• 85.

• 86.

• 87.

• 88.

• 89.

• 90.

• 91.

• 92.

• 93.

• 94.

• 95.

• 96.

• 97.

• 98.

• 99.

• 100.

• 101.

• 102.

• 103.

• 104.

• 105.

• 106.

• 107.

• 108.

• 109.

• 110.

• 111.

• 112.

执行以上代码, 得到 3 类梯度下降算法的函数图像如下图所示.

• 黑色是 BGD 的图像, 是一条光滑的曲线, 因为 BGD 每一次迭代求得的都是全局最优解;

• 红色是 SGD 的图像, 可见抖动很剧烈, 有不少局部最优解;

• 黄色是 MBGD 的图像, 相对 SGD 的成本-时间曲线平滑许多, 但仔细看, 仍然有抖动.