Python实现梯度下降

那么我们如何使用 Python 实现梯度下降呢？在本节我们就一起来看一下。

已知函数 y=x² 以及它的导数 y′=2^x

1. 实现步骤分析

接下来我们想一下如何用编程的思想实现这个算法。

• 第 1 步，随机初始化一个坐标（x₁,y₁）。
• 第 2 步，将 x1 移动至 x₁-α×2x₁ 位置，记为 x₂，并求出 y₂。这里，α 是学习速率，2x₁ 是该点的导数。
• 第 3 步，重复第 2 步，获得点（x₃,y₃）。
• ……
• 第 1000 步，获得点（x₁₀₀₀,y₁₀₀₀）的位置。

2. 代码实现实现

Python实现代码如下。

def grad(x,alpha,iter):
    x_list=[]
    y_list=[]
    x__list=[]
    g_list=[]
    for i in range(iter):
        y=x*x
        x_=2*x
        gradient=alpha*x_
        x_list.append(x)
        y_list.append(y)
        x__list.append(x_)
        g_list.append(gradient)
        x=x-gradient
    return x_list,y_list,x__list,g_list

3. 学习速率α详解

初始值的选择依托于所假设的数学模型，具体的数学模型会在后面的章节讨论。不同的数学模型反映在二次函数上则会是随机的初始点。本节主要学习速率 α 详细介绍。

上述代码中，速率 α 的选择是梯度下降思想的核心，α 直接影响了最后结果的好坏以及整个算法的效率。

1) α过小，将导致无法找到最小值

选择点（100,100）作为初始点，如图 1 所示。当 α=0.003 时，移动了 100 次，才到达 x=55 附近。虽然在这种情况下我们仍可以通过增加移动次数最终到达最小值的点，但是会耗费大量的时间和算力。特别是在工程应用中，时间和算力决定了一个算法是否有实际价值。

图1：α=0.003过小，导致迭代速度极慢

在极端情况下，当α足够小时，将无法到达最小值点。

2) α适中，将很快到达最小值

选择点（100,100）作为初始点，如图 2 所示。当 α=0.5 时，只移动了 2 次就到达了最小值。

图2：α=0.5时很快到达最小值

3) α超过一定阈值，可能从两端收敛到最低点

选择点（40,1600）作为初始点，如图 3 所示。当 α=0.9 时，将从两端收敛到最小值，详见表 1。

图3：　α=0.9时从两端收敛到最小值

α=0.9 时从两端收敛到最小值（保留2位小数）

4) α为一定阈值，可能导致来回振荡

选择点（40,1600）作为初始点，如图 4 所示。当α=1时，移动点将在点（40,1600）和点（-40,1600）之间来回振动。

图4：α=1导致来回振荡

5) α过大，将导致无法找到最小值，甚至发散

α 过小，虽然可能到达不了最小值的点，但是移动的方向仍是正确的，即向着最小值方向移动。但α过大，则可能造成偏离最小值的情况，即向着最小值的反方向移动。

选择点（20,400）作为初始点，如图 5 所示。当 α=1.1 时，每次移动都跨越了最小值，而且越来越偏离最小值。

图5：α=1.1过大，导致发散，无法到达最小值

详见下表：

综上所述，学习速率 α 的选择规律为：随着学习速率 α 的递增，学习过程从速度极慢，到两边振荡收敛，再到两边振荡发散。那么在实际应用中我们应该如何选择 α 呢？这要根据不同的应用场景来具体设置。

到目前为止，我们已经详细了解了机器学习中的一个重要思想，即“梯度下降”。它是线性回归、逻辑回归以及神经网络的核心思想，接下来就让我们一起学习梯度下降是如何应用到这 3 类算法中的。