在前面的章节中,我们学习了牛顿法,牛顿法的核心是先通过将函数泰勒展开,近似为一个二阶项目,对这个二阶项求导,可以得到极值点,则直接找到了在函数展开点附近的最优点。注意,我们这里说的是函数展开点附近的最优点。因为泰勒展开存在截断误差,我们是不能认为该点就是精确解的。
下面是公式层面的一个推导。
min x f ( x ) \min_x f(x) xminf(x)
对 x x x,我们于 x t x_t xt处(第 t t t次迭代的 x x x)位置进行二阶泰勒展开,有
f ( x ) ≈ f ( x t ) + f ( 1 ) ( x t ) ( x − x t ) + 1 2 f ( 2 ) ( x t ) ( x − x t ) 2 f(x) \approx f(x_t)+f^{(1)}(x_t)(x-x_t)+\frac{1}{2}f^{(2)}(x_t)(x-x_t)^2 f(x)≈f(xt)+f(1)(xt)(x−xt)+21f(2)(xt)(x−xt)2
令 x − x t = Δ x x-x_t=\Delta x x−xt=Δx
f ( x ) ≈ f ( x t ) + f ( 1 ) ( x t ) Δ x + 1 2 f ( 2 ) ( x t ) Δ x 2 f(x) \approx f(x_t)+f^{(1)}(x_t)\Delta x+\frac{1}{2}f^{(2)}(x_t)\Delta x^2 f(x)≈f(xt)+f(1)(xt)Δx+21f(2)(xt)Δx2
f ^ ( x ) = f ( x t ) + f ( 1 ) ( x t ) Δ x + 1 2 f ( 2 ) ( x t ) Δ x 2 \hat f(x)= f(x_t)+f^{(1)}(x_t)\Delta x+\frac{1}{2}f^{(2)}(x_t)\Delta x^2 f^(x)=f(xt)+f(1)(xt)Δx+21f(2)(xt)Δx2
很容易的,我们对上式求一阶导数可以得到极值点
f ^ ′ ( x ) = f ( 1 ) ( x t ) + f ( 2 ) ( x t ) Δ x = 0 \hat f'(x)= f^{(1)}(x_t)+f^{(2)}(x_t)\Delta x=0 f^′(x)=f(1)(xt)+f(2)(xt)Δx=0
即当 Δ x = − f ( 1 ) ( x t ) f ( 2 ) ( x t ) \Delta x=-\frac{ f^{(1)}(x_t)}{f^{(2)}(x_t)} Δx=−f(2)(xt)f(1)(xt)时,有极值点。
此处 x − x t = Δ x x-x_t=\Delta x x−xt=Δx可以认为是优下降的方向。令 g t = f ( 1 ) ( x t ) g_t=f^{(1)}(x_t) gt=f(1)(xt), h t = f ( 2 ) ( x t ) h_t=f^{(2)}(x_t) ht=f(2)(xt),分别代表近似函数的梯度和hessian,那么牛顿法的迭代过程就可以表示为
Δ x = − h − 1 g \Delta x = -h^{-1} g Δx=−h−1g
不难看出,牛顿法存在两个问题
因此,我们引入了拟牛顿法来解决相关问题。
在概念上,我们使用一个矩阵 U U U来近似hessian,在二次条件下,hessian满足如下条件
x t + 1 − x t = h t − 1 ( f ′ ( x t + 1 ) − f ′ ( x t ) ) x_{t+1}-x_t = h_t^{-1}(f'(x_{t+1})-f'(x_t)) xt+1−xt=ht−1(f′(xt+1)−f′(xt))
要求近似矩阵U也应当满足相应条件
x t + 1 − x t = U t ( f ′ ( x t + 1 ) − f ′ ( x t ) ) x_{t+1}-x_t = U_t(f'(x_{t+1})-f'(x_t)) xt+1−xt=Ut(f′(xt+1)−f′(xt))
拟牛顿法的基础形式一般有DFP和BFGS法两种。DFP法和BFGS法都是求解无约束优化问题的二次型拟牛顿法,其核心思想是通过构建二次模型来近似原始函数,利用该模型求解最优解的方向和步长,从而迭代地逼近全局最优解。
具体来说,DFP法和BFGS法都通过逐步构建Hessian矩阵的逆矩阵来求解最优解,但它们的不同之处在于构建逆矩阵的方式不同。
对上面的式中,我们知道存在
Δ x = − h t − 1 Δ g t \Delta x = -h_t^{-1} \Delta g_t Δx=−ht−1Δgt
这里我们构造一个矩阵 D t D_t Dt来逼近这个函数,认为存在
Δ x = D t Δ g t \Delta x = D_t \Delta g_t Δx=DtΔgt
这里注意, D t D_t Dt是我们构造的一个矩阵,本身是不准确的,我们想要逐步迭代去逼近真实的 D t D_t Dt,有点自举式算法的味道。因此我们令
D t + 1 = D t + Δ D t D_{t+1} = D_t + \Delta D_t Dt+1=Dt+ΔDt
代入上面的式子,有
Δ x = D t Δ g t + Δ D t Δ g t \Delta x = D_t \Delta g_t + \Delta D_t \Delta g_t Δx=DtΔgt+ΔDtΔgt
我们重点关注的是“自举”的过程,因此将上面的式子变式为
Δ D t Δ g t = Δ x − D t Δ g t \Delta D_t \Delta g_t = \Delta x - D_t \Delta g_t ΔDtΔgt=Δx−DtΔgt
在这里我们假设存在一个向量 q t q_t qt, w t w_t wt,使得下面的式子成立:
Δ D t = Δ x q t T + D t Δ g t w t T \Delta D_t = \Delta x q_t^T + D_t \Delta g_t w_t^T ΔDt=ΔxqtT+DtΔgtwtT
注意hessian是一个对称矩阵,因此我们认为 Δ D t \Delta D_t ΔDt也应该是对称的,又参照上面的两个式子,可以得到
q t T Δ g t = I q_t^T \Delta g_t = I qtTΔgt=I
w t T Δ g t = I w_t^T \Delta g_t = I wtTΔgt=I
不妨设
q t = α t Δ x q_t = \alpha_t \Delta x qt=αtΔx
w t = β t D t Δ g t w_t = \beta_t D_t \Delta g_t wt=βtDtΔgt
带入到上面的式子,可得
α t = 1 Δ x T Δ g t \alpha_t = \frac{1}{\Delta x^T \Delta g_t} αt=ΔxTΔgt1
β t = 1 Δ g t T D t T Δ g t \beta_t= \frac{1}{\Delta g_t^T D_t^T \Delta g_t} βt=ΔgtTDtTΔgt1
代入可得
Δ D t = Δ x t Δ x t T Δ g t T Δ x t − D t Δ g t Δ g t T D t T Δ g t T D t T Δ g t \Delta D_t = \frac{\Delta x_t \Delta x_t^T}{\Delta g_t^T \Delta x_t} - \frac{D_t \Delta g_t \Delta g_t^T D_t^T}{\Delta g_t^T D_t^T \Delta g_t} ΔDt=ΔgtTΔxtΔxtΔxtT−ΔgtTDtTΔgtDtΔgtΔgtTDtT
同理可得BFGS方法的推导。
实现代码如下
import numpy as np
def dfp(f, x0, eps=1e-6, max_iter=100):
"""
DFP法最优化函数
Args:
f: 目标函数
x0: 初始值
eps: 精度
max_iter: 最大迭代次数
Returns:
tuple: 最优化的结果
"""
n = len(x0)
x = x0
H = np.eye(n)
grad = np.ones(n)
k = 0
while np.linalg.norm(grad) > eps and k < max_iter:
grad = np.gradient(f, x)
p = -np.dot(H, grad)
alpha = 1
while f(x + alpha * p) > f(x) + 0.5 * alpha * np.dot(grad, p):
alpha = 0.5 * alpha
s = alpha * p
x_new = x + s
y = np.gradient(f, x_new) - grad
rho = 1 / np.dot(y, s)
A = np.eye(n) - rho * np.outer(s, y)
H_new = np.dot(A, np.dot(H, A.T)) + rho * np.outer(s, s)
x = x_new
H = H_new
k += 1
return x, f(x), k
import numpy as np
def bfgs(f, x0, eps=1e-6, max_iter=100):
"""
BFGS法最优化函数
Args:
f: 目标函数
x0: 初始值
eps: 精度
max_iter: 最大迭代次数
Returns:
tuple: 最优化的结果
"""
n = len(x0)
x = x0
H = np.eye(n)
grad = np.ones(n)
k = 0
while np.linalg.norm(grad) > eps and k < max_iter:
grad = np.gradient(f, x)
p = -np.dot(H, grad)
alpha = 1
while f(x + alpha * p) > f(x) + 0.5 * alpha * np.dot(grad, p):
alpha = 0.5 * alpha
s = alpha * p
x_new = x + s
y = np.gradient(f, x_new) - grad
rho = 1 / np.dot(y, s)
A = np.eye(n) - rho * np.outer(s, y)
H_new = np.dot(A.T, np.dot(H, A)) + rho * np.outer(y, y)
x = x_new
H = H_new
k += 1
return x, f(x), k
之后有空会更新下面的一些算法。