BFGS算法作为一种优化方法,通过避免直接计算和求逆海森矩阵,相较于牛顿法等方法实现了改进。然而,BFGS算法仍需要存储和更新海森矩阵(或其逆矩阵)的一个近似,这通常是一个稠密的 d×d 矩阵,d 是参数 (parameter)的数量。对于拥有数百万甚至数十亿参数的机器学习 (machine learning)模型,由于内存限制(O(d2) 内存),存储一个 d×d 矩阵是完全不可行的。在这种情况下,有限内存BFGS (L-BFGS) 算法变得必要。
核心思想:以内存换计算
L-BFGS巧妙地避开了标准BFGS的 O(d2) 内存需求。L-BFGS不存储稠密的 d×d 海森近似矩阵 Hk,它只存储少量(例如 m 个)最近的校正对 (si,yi)。回顾一下:
- sk=xk+1−xk (参数 (parameter)变化)
- yk=∇f(xk+1)−∇f(xk) (梯度变化)
通常,m 是一个小的整数,常在5到20之间,这与问题维度 d 无关。这使得内存需求大幅降低到 O(md),其与参数数量 d 成线性关系,因此即使对于非常大的模型也能应对。
其权衡之处在于,L-BFGS不直接访问矩阵 Hk,而是仅利用存储的 (si,yi) 对隐式计算矩阵向量 (vector)积 Hk∇fk(这是确定搜索方向 pk=−Hk∇fk 所需的)。这种计算涉及一个通常被称为“L-BFGS两循环递归”的过程。
L-BFGS两循环递归
L-BFGS的核心在于它如何在不显式形成矩阵 Hk 的情况下重构搜索方向 pk=−Hk∇fk。它利用BFGS更新公式,该公式可以递归表示。该算法使用最近的 m 个对 {(si,yi)}i=k−mk−1 来计算乘积 Hk∇fk。
过程包含:
- 初始化: 从初始海森近似 Hk0 开始。一个常见的选择是 Hk0=γkI,其中 I 是单位矩阵,γk 是一个缩放因子,用于近似真实海森矩阵的大小。一个常用选择是:
γk=yk−1Tyk−1sk−1Tyk−1
这种缩放有助于确保搜索方向具有适当的大小。
- 第一个循环(反向传递): 从 i=k−1 向下迭代到 k−m。在此循环中,算法计算中间值 αi 并更新梯度向量 (vector) q=∇fk。
- ρi=1/(yiTsi)
- αi=ρisiTq
- q=q−αiyi
- 缩放: 将中间向量 q 乘以初始海森近似 Hk0:z=Hk0q。
- 第二个循环(正向传递): 从 i=k−m 向上迭代到 k−1。此循环使用存储的 αi 值和 (si,yi) 对来重构最终方向。
- βi=ρiyiTz
- z=z+si(αi−βi)
- 结果: 最终向量 z 是期望的乘积,z=Hk∇fk。然后搜索方向是 pk=−z。
这种两循环过程只需向量操作(点积、标量-向量乘法、向量加法),每迭代的计算成本约为 O(md),远低于标准BFGS更新的 O(d2) 成本或牛顿法的 O(d3) 成本。
L-BFGS算法概述
L-BFGS算法的一个典型迭代如下所示:
- 选择起始点 x0。
- 设置内存大小 m。
- 对于 k=0,1,2,... 直到收敛:
a. 计算梯度 ∇fk=∇f(xk)。
b. 使用最近的 m 对 (si,yi) 和初始海森近似 Hk0,通过两循环递归计算搜索方向 pk=−Hk∇fk。
c. 使用线搜索过程(例如,满足Wolfe条件的Armijo线搜索)找到合适的步长 αk>0。
d. 更新参数 (parameter):xk+1=xk+αkpk。
e. 计算新梯度 ∇fk+1=∇f(xk+1)。
f. 计算 sk=xk+1−xk 和 yk=∇fk+1−∇fk。
g. 曲率检查: 如果 skTyk>ϵ (对于某个小的 ϵ>0),则存储对 (sk,yk)。如果存储对的数量超过 m,则丢弃最旧的对 (sk−m,yk−m)。
h. 检查收敛标准(例如,∣∣∇fk+1∣∣<容差)。
L-BFGS的优点
- 内存效率高: 其主要优点。O(md) 的内存使其适用于参数 (parameter)数量庞大、存储 O(d2) 不可行的问题。
- 计算效率高: 对于大的 d,每次迭代 O(md) 的计算速度远快于标准BFGS或牛顿法。
- 收敛性好: 通常达到超线性收敛速度,在许多确定性问题上,其收敛速度(以迭代次数计)通常比梯度下降 (gradient descent)或共轭梯度等一阶方法快得多。
缺点与考量
- 超参数 (parameter) (hyperparameter) m: 内存缓冲的大小 m 影响性能。较大的 m 存储更多曲率信息,可能带来更好的搜索方向和更快的收敛(更少迭代次数),但会增加每次迭代的内存使用和计算成本。典型值较小(例如,5-20)。
- 线搜索: 仍需要线搜索算法来寻找步长 αk,这有时本身就计算量大,每次迭代需要多次函数和梯度评估。
- 曲率条件: 更新依赖于条件 skTyk>0,这意味着函数在 sk 方向上的曲率为正。在非凸区域,这可能不成立。如果此条件不满足,实现通常会跳过更新。
- 批量方法: L-BFGS,与标准BFGS和牛顿法一样,基本是为确定性(批量)优化设计的,其中真实梯度是使用整个数据集计算的。将其有效调整到深度学习 (deep learning)中常见的随机设置(使用小批量)具有挑战性,尽管存在一些变体。
L-BFGS在实践中
L-BFGS是许多科学计算和优化范畴中的主力算法。在机器学习 (machine learning)中,它通常对以下方面有效:
- 训练那些计算完整梯度可行(例如,中等规模数据集上的逻辑回归、线性SVM)的模型。
- 需要在最优点附近高精度的问题。
- 有时在Adam或SGD等随机方法初步训练后,用作最后的微调 (fine-tuning)步骤。
尽管像Adam这样的自适应一阶方法因其与小批量处理的兼容性以及在复杂非凸地形中寻找最优点的能力,通常更受青睐用于训练大型深度神经网络 (neural network),但了解L-BFGS能提供有价值的认识,关于如何在内存受限下将曲率信息应用于优化。它代表了连接一阶算法简洁性与二阶算法高效性的强大桥梁。在SciPy (scipy.optimize.minimize(method='L-BFGS-B')) 等库中可以方便地找到其实现,并且一些机器学习框架内部也使用它。