使用Hessian矩阵的梯度下降牛顿法

时间:2016-04-01 10:39:55

标签: python optimization machine-learning gradient-descent newtons-method

我正在使用牛顿法实现回归梯度下降,如机器学习概率透视(墨菲)书的8.3部分所述。我正在使用此实现中的二维数据。我使用以下符号。
    x =输入数据点m * 2
    y =对应于输入数据的标记输出(m)
    H = Hessian矩阵定义为
    

梯度下降更新
    

其中损失函数定义为     

就我而言     数组,H是

这是我的python实现。然而,由于成本在每次迭代中都在增加,因此无法正常工作。

def loss(x,y,theta):
   m,n = np.shape(x)
   cost_list = []
   for i in xrange(0,n): 
     x_0 = x[:,i].reshape((m,1)) 
     predicted = np.dot(x_0, theta[i])    
     error = predicted - y
     cost = np.sum(error ** 2) /  m
     cost_list.append(cost)

   cost_list = np.array(cost_list).reshape((2,1))
   return cost_list


def NewtonMethod(x,y,theta,maxIterations):
   m,n = np.shape(x)
   xTrans  = x.transpose()
   H       = 2 * np.dot(xTrans,x) / m
   Hinv    = np.linalg.inv(H)
   thetaPrev = np.zeros_like(theta)
   best_iter = maxIterations
   for i in range(0,maxIterations):
     cost = loss(x,y,theta)
     theta  = theta - np.dot(Hinv,cost))
     if(np.allclose(theta,thetaPrev,rtol=0.001,atol=0.001)):
        break;
     else:
       thetaPrev = theta
       best_iter = i

   return theta

以下是我使用的示例值

import numpy as np

x = np.array([[-1.7, -1.5],[-1.0 , -0.3],[ 1.7 ,  1.5],[-1.2, -0.7 ][  0.6,  0.1]])  
y = np.array([ 0.3 ,  0.07, -0.2,  0.07,  0.03 ])
theta = np.zeros(2)
NewtonMethod(x,y,theta,100)

需要帮助/建议来解决这个问题 谢谢

1 个答案:

答案 0 :(得分:0)

您实际上使用的步长为1.尝试缩小步长并查看是否有帮助。也就是说,而不是

enter image description here

这样做:

enter image description here

的值小于1。