机器学习梯度下降法，最通俗易懂的解释

来源：简书，作者：六尺帐篷

链接：https://www.jianshu.com/p/c7e642877b0e

本文从一个下山场景开始，提出梯度下降算法的基本思想，接着从数学上解释梯度下降算法原理，最后实现一个简单的梯度下降算法实例！

梯度下降的场景假设

梯度下降法的基本思想可以类比为一个下山的过程。假设这样一个场景：一个人被困在山上，需要从山上下来(i.e. 找到山的最低点，也就是山谷)。但此时山上的浓雾很大，导致可视度很低。因此，下山的路径就无法确定，他必须利用自己周围的信息去找到下山的路径。这个时候，他就可以利用梯度下降算法来帮助自己下山。具体来说就是，以他当前的所处的位置为基准，寻找这个位置最陡峭的地方，然后朝着山的高度下降的地方走，同理，如果我们的目标是上山，也就是爬到山顶，那么此时应该是朝着最陡峭的方向往上走。然后每走一段距离，都反复采用同一个方法，最后就能成功的抵达山谷。

我们同时可以假设这座山最陡峭的地方是无法通过肉眼立马观察出来的，而是需要一个复杂的工具来测量，同时，这个人此时正好拥有测量出最陡峭方向的能力。所以，此人每走一段距离，都需要一段时间来测量所在位置最陡峭的方向，这是比较耗时的。那么为了在太阳下山之前到达山底，就要尽可能的减少测量方向的次数。这是一个两难的选择，如果测量的频繁，可以保证下山的方向是绝对正确的，但又非常耗时，如果测量的过少，又有偏离轨道的风险。所以需要找到一个合适的测量方向的频率，来确保下山的方向不错误，同时又不至于耗时太多！

梯度下降

梯度下降的基本过程就和下山的场景很类似。

首先，我们有一个可微分的函数。这个函数就代表着一座山。我们的目标就是找到这个函数的最小值，也就是山底。根据之前的场景假设，最快的下山的方式就是找到当前位置最陡峭的方向，然后沿着此方向向下走，对应到函数中，就是找到给定点的梯度 ，然后朝着梯度相反的方向，就能让函数值下降的最快！因为梯度的方向就是函数之变化最快的方向(在后面会详细解释)
 所以，我们重复利用这个方法，反复求取梯度，最后就能到达局部的最小值，这就类似于我们下山的过程。而求取梯度就确定了最陡峭的方向，也就是场景中测量方向的手段。那么为什么梯度的方向就是最陡峭的方向呢？接下来，我们从微分开始讲起

微分

看待微分的意义，可以有不同的角度，最常用的两种是：

• 函数图像中，某点的切线的斜率

• 函数的变化率，几个微分的例子：

  

上面的例子都是单变量的微分，当一个函数有多个变量的时候，就有了多变量的微分，即分别对每个变量进行求微分：

梯度

梯度实际上就是多变量微分的一般化。
 下面这个例子：

我们可以看到，梯度就是分别对每个变量进行微分，然后用逗号分割开，梯度是用<>包括起来，说明梯度其实一个向量。

梯度是微积分中一个很重要的概念，之前提到过梯度的意义

• 在单变量的函数中，梯度其实就是函数的微分，代表着函数在某个给定点的切线的斜率

• 在多变量函数中，梯度是一个向量，向量有方向，梯度的方向就指出了函数在给定点的上升最快的方向

这也就说明了为什么我们需要千方百计的求取梯度！我们需要到达山底，就需要在每一步观测到此时最陡峭的地方，梯度就恰巧告诉了我们这个方向。梯度的方向是函数在给定点上升最快的方向，那么梯度的反方向就是函数在给定点下降最快的方向，这正是我们所需要的。所以我们只要沿着梯度的方向一直走，就能走到局部的最低点！

梯度下降算法的数学解释

上面我们花了大量的篇幅介绍梯度下降算法的基本思想和场景假设，以及梯度的概念和思想。下面我们就开始从数学上解释梯度下降算法的计算过程和思想！

此公式的意义是：J 是关于Θ的一个函数，我们当前所处的位置为Θ0 点，要从这个点走到 J 的最小值点，也就是山底。首先我们先确定前进的方向，也就是梯度的反向，然后走一段距离的步长，也就是α，走完这个段步长，就到达了Θ1 这个点！

下面就这个公式的几个常见的疑问：

• α是什么含义？
  α在梯度下降算法中被称作为学习率或者步长，意味着我们可以通过α来控制每一步走的距离，以保证不要步子跨的太大扯着蛋，哈哈，其实就是不要走太快，错过了最低点。同时也要保证不要走的太慢，导致太阳下山了，还没有走到山下。所以α的选择在梯度下降法中往往是很重要的！α不能太大也不能太小，太小的话，可能导致迟迟走不到最低点，太大的话，会导致错过最低点！

• 为什么要梯度要乘以一个负号？
  梯度前加一个负号，就意味着朝着梯度相反的方向前进！我们在前文提到，梯度的方向实际就是函数在此点上升最快的方向！而我们需要朝着下降最快的方向走，自然就是负的梯度的方向，所以此处需要加上负号。

梯度下降算法的实例

我们已经基本了解了梯度下降算法的计算过程，那么我们就来看几个梯度下降算法的小实例，首先从单变量的函数开始

单变量函数的梯度下降

我们假设有一个单变量的函数

函数的微分

初始化，起点为

学习率为

根据梯度下降的计算公式

我们开始进行梯度下降的迭代计算过程：

如图，经过四次的运算，也就是走了四步，基本就抵达了函数的最低点，也就是山底

多变量函数的梯度下降

我们假设有一个目标函数：

现在要通过梯度下降法计算这个函数的最小值。我们通过观察就能发现最小值其实就是 (0，0)点。但是接下来，我们会从梯度下降算法开始一步步计算到这个最小值！
 我们假设初始的起点为：

初始的学习率为：

函数的梯度为：

进行多次迭代：

我们发现，已经基本靠近函数的最小值点

梯度下降算法的实现

下面我们将用 python 实现一个简单的梯度下降算法。场景是一个简单的线性回归的例子：假设现在我们有一系列的点，如下图所示

我们将用梯度下降法来拟合出这条直线！

首先，我们需要定义一个代价函数，在此我们选用均方误差代价函数：

此公示中

• m 是数据集中点的个数

• ½是一个常量，这样是为了在求梯度的时候，二次方乘下来就和这里的½抵消了，自然就没有多余的常数系数，方便后续的计算，同时对结果不会有影响

• y 是数据集中每个点的真实 y 坐标的值

• h 是我们的预测函数，根据每一个输入 x，根据Θ 计算得到预测的 y 值，即：

  

我们可以根据代价函数看到，代价函数中的变量有两个，所以是一个多变量的梯度下降问题，求解出代价函数的梯度，也就是分别对两个变量进行微分

明确了代价函数和梯度，以及预测的函数形式。我们就可以开始编写代码了。但在这之前，需要说明一点，就是为了方便代码的编写，我们会将所有的公式都转换为矩阵的形式，python 中计算矩阵是非常方便的，同时代码也会变得非常的简洁。

为了转换为矩阵的计算，我们观察到预测函数的形式：

我们有两个变量，为了对这个公式进行矩阵化，我们可以给每一个点 x 增加一维，这一维的值固定为 1，这一维将会乘到Θ0 上。这样就方便我们统一矩阵化的计算：

然后我们将代价函数和梯度转化为矩阵向量相乘的形式：

coding time

首先，我们需要定义数据集和学习率

 1import numpy as np
 2
 3# Size of the points dataset.
 4m = 20
 5
 6# Points x-coordinate and dummy value (x0, x1).
 7X0 = np.ones((m, 1))
 8X1 = np.arange(1


    
, m+1).reshape(m, 1)
 9X = np.hstack((X0, X1))
10
11# Points y-coordinate
12y = np.array([
13    3, 4, 5, 5, 2, 4, 7, 8, 11, 8, 12,
14    11, 13, 13, 16, 17, 18, 17, 19, 21
15]).reshape(m, 1)
16
17# The Learning Rate alpha.
18alpha = 0.01

接下来我们以矩阵向量的形式定义代价函数和代价函数的梯度

1def error_function(theta, X, y):
2    '''Error function J definition.'''
3    diff = np.dot(X, theta) - y
4    return (1./2*m) * np.dot(np.transpose(diff), diff)
5
6def gradient_function(theta, X, y):
7    '''Gradient of the function J definition.'''
8    diff = np.dot(X, theta) - y
9    return (1./m) * np.dot(np.transpose(X), diff)

最后就是算法的核心部分，梯度下降迭代计算

    
1def gradient_descent(X, y, alpha):
2    '''Perform gradient descent.'''
3    theta = np.array([1, 1]).reshape(2, 1)
4    gradient = gradient_function(theta, X, y)
5    while not np.all(np.absolute(gradient) <= 1e-5):
6        theta = theta - alpha * gradient
7        gradient = gradient_function(theta, X, y)
8    return theta

当梯度小于 1e-5 时，说明已经进入了比较平滑的状态，类似于山谷的状态，这时候再继续迭代效果也不大了，所以这个时候可以退出循环！

完整代码如下：

 1import numpy as np
 2
 3# Size of the points dataset.
 4m = 20
 5
 6# Points x-coordinate and dummy value (x0, x1).
 7X0 = np.ones((m, 1))
 8X1 = np.arange(1, m+1).reshape(m, 1)
 9X = np.hstack((X0, X1))
10
11# Points y-coordinate
12y = np.array([
13    3, 4, 5, 5, 2, 4, 7, 8, 11, 8, 12,
14    11, 13, 13, 16, 17, 18


    
, 17, 19, 21
15]).reshape(m, 1)
16
17# The Learning Rate alpha.
18alpha = 0.01
19
20def error_function(theta, X, y):
21    '''Error function J definition.'''
22    diff = np.dot(X, theta) - y
23    return (1./2*m) * np.dot(np.transpose(diff), diff)
24
25def gradient_function(theta, X, y):
26    '''Gradient of the function J definition.'''
27    diff = np.dot(X, theta) - y
28    return (1./m) * np.dot(np.transpose(X), diff)
29
30def gradient_descent(X, y, alpha):
31    '''Perform gradient descent.'''
32    theta = np.array([1, 1]).reshape(2, 1)
33    gradient = gradient_function(theta, X, y)
34    while not np.all(np.absolute(gradient) <= 1e-5):
35        theta = theta - alpha * gradient
36        gradient = gradient_function(theta, X, y)
37    return theta
38
39optimal = gradient_descent(X, y, alpha)
40print('optimal:', optimal)
41print('error function:', error_function(optimal, X, y)[0,0])

运行代码，计算得到的结果如下：

所拟合出的直线如下：

小结

至此，我们就基本介绍完了梯度下降法的基本思想和算法流程，并且用 Python 实现了一个简单的梯度下降算法拟合直线的案例！
最后，我们回到文章开头所提出的场景假设:

这个下山的人实际上就代表了反向传播算法，下山的路径其实就代表着算法中一直在寻找的参数Θ，山上当前点的最陡峭的方向实际上就是代价函数在这一点的梯度方向，场景中观测最陡峭方向所用的工具就是微分 。在下一次观测之前的时间就是有我们算法中的学习率α所定义的。

可以看到场景假设和梯度下降算法完美对应。

●编号800，输入编号直达本文

●输入m获取文章目录

更多推荐《25个技术类公众微信》

涵盖：程序人生、算法与数据结构、黑客技术与网络安全、大数据技术、前端开发、Java、Python、Web开发、安卓开发、iOS开发、C/C++、.NET、Linux、数据库、运维等。