广义线性模型BFGS拟牛顿方法的统计与计算复杂度
最优化与控制
2024-03-15 v2
摘要
梯度下降(GD)方法已被广泛用于求解广义线性模型(GLMs)中的参数估计,GLMs是当链接函数可为非线性时线性模型的推广。在具有多项式链接函数的GLMs中,已证明在高信噪比(SNR)情形下,由于问题的强凸性与光滑性,GD线性收敛并在对数次迭代内达到最终所需精度。相反,在低SNR设定下,问题变为局部凸,GD以更慢的速率收敛,并需要多项式次迭代才能达到所需精度。尽管牛顿法可用于解决低SNR情形下损失函数的平坦曲率问题,但其计算成本在高维设定下令人望而却步,因为其复杂度为,其中为问题维度。为弥补GD与牛顿法的不足,我们提出使用BFGS拟牛顿方法求解GLMs的参数估计,其每次迭代成本为。当SNR较低时,对于次数为的多项式链接函数GLMs,我们证明BFGS的迭代线性收敛至总体最小二乘损失函数的最优解,且BFGS算法的压缩系数与牛顿法相当。此外,线性速率的压缩因子与问题参数无关,仅依赖于链接函数次数。同时,对于含个样本的经验损失,我们证明在次数为的多项式链接函数GLMs低SNR设定下,BFGS迭代至多经过次迭代达到的最终统计半径。
引用
@article{arxiv.2206.00207,
title = {Statistical and Computational Complexities of BFGS Quasi-Newton Method for Generalized Linear Models},
author = {Qiujiang Jin and Tongzheng Ren and Nhat Ho and Aryan Mokhtari},
journal= {arXiv preprint arXiv:2206.00207},
year = {2024}
}