为什么局部下降最快的方向就是梯度的负方向?

什么是梯度?

对于梯度下降算法(Gradient Descent Algorithm),我们都已经很熟悉了。无论是在线性回归(Linear Regression)、逻辑回归(Logistic Regression)还是神经网络(Neural Network)等等,都会用到梯度下降算法。我们先来看一下梯度下降算法的直观解释:

假设我们位于黄山的某个山腰处,山势连绵不绝,不知道怎么下山。于是决定走一步算一步,也就是每次沿着当前位置最陡峭最易下山的方向前进一小步,然后继续沿下一个位置最陡方向前进一小步。这样一步一步走下去,一直走到觉得我们已经到了山脚。这里的下山最陡的方向就是梯度的负方向。

首先理解什么是梯度?通俗来说,梯度就是表示某一函数在该点处的方向导数沿着该方向取得最大值,即函数在当前位置的导数。

=df(θ)dθ\nabla = \frac{df(\theta)}{d\theta}

上式中,θ\theta 是自变量,f(θ)f(\theta) 是关于 θ\theta 的函数。

梯度下降算法

如果函数 f(θ)f(\theta) 是凸函数,那么就可以使用梯度下降算法进行优化。梯度下降算法的公式我们已经很熟悉了:

θ=θ0ηf(θ)\theta = \theta_0 - \eta \cdot \nabla f(\theta)

其中,θ0\theta_0 是自变量参数,即下山位置坐标,η\eta 是学习因子,即下山每次前进的一小步(步进长度),θ\theta 是更新后的 θ0\theta_0,即下山移动一小步之后的位置。

梯度下降算法的公式非常简单!但是“沿着梯度的反方向(坡度最陡)”是我们日常经验得到的,其本质的原因到底是什么呢?为什么局部下降最快的方向就是梯度的负方向呢?也许很多朋友还不太清楚。没关系,接下来我将以通俗的语言来详细解释梯度下降算法公式的数学推导过程。

一阶泰勒展开式

这里需要一点数学基础,对泰勒展开式有些了解。简单地来说,泰勒展开式利用的就是函数的局部线性近似这个概念。我们以一阶泰勒展开式为例:

f(θ)f(θ0)+(θθ0)f(θ0)f(\theta) \approx f(\theta_0) + (\theta - \theta_0) \cdot \nabla f(\theta_0)

不懂上面的公式?没有关系。我用下面这张图来解释。

凸函数 f(θ)f(\theta) 的某一小段 [θ0,θ][\theta_0,\theta] 由上图黑色曲线表示,可以利用线性近似的思想求出 f(θ)f(\theta) 的值,如上图红色直线。该直线的斜率等于 f(θ)f(\theta)θ0\theta_0 处的导数。则根据直线方程,很容易得到 f(θ)f(\theta) 的近似表达式为:

f(θ)f(θ0)+(θθ0)f(θ0)f(\theta) \approx f(\theta_0) + (\theta - \theta_0) \cdot \nabla f(\theta_0)

这就是一阶泰勒展开式的推导过程,主要利用的数学思想就是曲线函数的线性拟合近似。

梯度下降数学原理

知道了一阶泰勒展开式之后,接下来就是重点了!我们来看一下梯度下降算法是如何推导的。

先写出一阶泰勒展开式的表达式:

f(θ)f(θ0)+(θθ0)f(θ0)f(\theta) \approx f(\theta_0) + (\theta - \theta_0) \cdot \nabla f(\theta_0)

其中,θθ0\theta - \theta_0 是微小矢量,它的大小就是我们之前讲的步进长度 η\eta,类比于下山过程中每次前进的一小步,η\eta 为标量,而 θθ0\theta - \theta_0 的单位向量用 vv 表示。则 θθ0\theta - \theta_0 可表示为:

θθ0=ηv\theta - \theta_0 = \eta v

特别需要注意的是,θθ0\theta - \theta_0 不能太大,因为太大的话,线性近似就不够准确,一阶泰勒近似也不成立了。替换之后,f(θ)f(\theta) 的表达式为:

f(θ)f(θ0)+ηvf(θ0)f(\theta) \approx f(\theta_0) + \eta v \cdot \nabla f(\theta_0)

重点来了,局部下降的目的是希望每次 θ\theta 更新,都能让函数值 nf(θ)f(\theta) 变小。也就是说,上式中,我们希望 f(θ)<f(θ0)f(\theta) < f(\theta_0)。则有:

f(θ)f(θ0)ηvf(θ0)<0f(\theta) - f(\theta_0) \approx \eta v \cdot \nabla f(\theta_0) < 0

因为 η\eta 为标量,且一般设定为正值,所以可以忽略,不等式变成了:

vf(θ0)<0v \cdot \nabla f(\theta_0) < 0

上面这个不等式非常重要!vvf(θ0)\nabla f(\theta_0) 都是向量,f(θ0)\nabla f(\theta_0) 是当前位置的梯度方向,vv 表示下一步前进的单位向量,是需要我们求解的,有了它,就能根据 θθ0=ηv\theta - \theta_0 = \eta v 确定 θ\theta 值了。

想要两个向量的乘积小于零,我们先来看一下两个向量乘积包含哪几种情况:

AABB 均为向量,α\alpha 为两个向量之间的夹角。AABB 的乘积为:

AB=ABcos(α)A \cdot B = \Vert A \Vert \cdot \Vert B \Vert \cdot \cos(\alpha)

A\Vert A \VertB\Vert B \Vert 均为标量,在 A\Vert A \VertB\Vert B \Vert 确定的情况下,只要 cos(α)=1\cos(\alpha) = -1,即 AABB 完全反向,就能让 AABB 的向量乘积最小(负最大值)。

顾名思义,当 vvf(θ0)\nabla f(\theta_0) 互为反向,即 vv 为当前梯度方向的负方向的时候,能让 vf(θ0)v \cdot \nabla f(\theta_0) 最大程度地小,也就保证了 vv 的方向是局部下降最快的方向。

知道 vvf(θ0)\nabla f(\theta_0) 的反方向后,可直接得到:

v=f(θ0)f(θ0)v = -\frac{\nabla f(\theta_0)}{\Vert \nabla f(\theta_0) \Vert}

之所以要除以 f(θ0)\nabla f(\theta_0) 的模 f(θ0)\Vert \nabla f(\theta_0) \Vert,是因为 vv 是单位向量。

求出最优解 vv 之后,带入到 θθ0=ηv\theta - \theta_0 = \eta v 中,得:

θ=θ0ηf(θ0)f(θ0)\theta = \theta_0 - \eta \frac{\nabla f(\theta_0)}{\Vert \nabla f(\theta_0) \Vert}

一般地,因为 f(θ0)\Vert \nabla f(\theta_0) \Vert 是标量,可以并入到步进因子 η\eta 中,即简化为:

θ=θ0ηf(θ0)\theta = \theta_0 - \eta \nabla f(\theta_0)

这样,我们就推导得到了梯度下降算法中 θ\theta 的更新表达式。