Processing math: 100%

牛顿法基于泰勒公式Hessian矩阵的应用。

牛顿法的推导

将损失函数$f(x)$在$x_0$处用泰勒公式展开,并保留到二阶项,得:
f(x)=f(x0)+(xx0)g+12(xx0)H(xx0)+...

牛顿法的思想是“直接找到令g=0”的位置。
方法是对f(x)在$x_0$处的偏导并令所有偏导为0。

f(x)=xf(x0)+x(xx0)g+x12(xx0)H(xx0) =0+g+H(xx0)

令f'(x)=0得:
x=H1g+x0

牛顿法 VS 梯度下降法

牛顿法:$x = -H^{-1}g + x_0$
梯度下降法:$x = -\eta g + x_0$

牛顿法相对于梯度下降法的改进,是将学习率变成了Hessian矩阵的逆。
$H^{-1}$的作用:

  1. 改变梯度的方向
  2. 决定了step的size

举个例子

假设loss function为图中的黑线。
取x0的位置,按泰勒公式展开,保留前三项,得到红色曲线。
红色曲线是二次曲线,可直接计算出来它的最小值处为x1。
令x1为新的x0,开始下一轮迭代。
如果f(x)本身就是二次曲线,牛顿法可以一步到位。

牛顿法近似

在H是正定的情况下,就能正常迭代。
当H不是正定时,牛顿法会出错。
解决方法:正则化,即H=H+aI
当H的负特征非常大时,a必须也很大,此时H被aI主导。

牛顿法的缺点

  1. $H^-1$的计算量大
  2. 这种方法只能保证找到f'(x)=0的点。但这种点不一定是minima。也有可能是maxima或者saddle point。

因此,牛顿法不适用于深度学习。