QNG:一种用于大规模统计学习的准自然梯度方法

QNG: A Quasi-Natural Gradient Method for Large-Scale Statistical Learning

SIAM Journal on Optimization · 2022
被引 0
ABS 3

中文导读

提出一种准自然梯度方法,通过秩一模型近似Fisher信息矩阵,降低大规模统计学习中的计算成本,在收敛速度和鲁棒性上优于随机梯度下降和在线有限内存BFGS方法。

Abstract

Natural gradient method provides a powerful paradigm for training statistical models and offers several appealing theoretic benefits. It constructs the Fisher information matrix to correct the ordinary gradients, and thus, the cost may become prohibitively expensive in the large-scale setting. This paper proposes a quasi-natural gradient method to address this computational issue. It employs a rank-one model to capture the second-order information from the underlying statistical manifold and to iteratively update the Fisher approximations. A three-loop procedure is designed to implement the updating formulas. This procedure resembles the classical two-loop procedure in the limited-memory BFGS method but saves half of the memory usage while it can be made faster. The resulting method retains the convergence rate advantages of existing stochastic optimization methods and has inherent ability in handling nonconvexity. Numerical studies conducted on several machine learning tasks demonstrate the reduction in convergence time and the robustness in tackling nonconvexity relative to stochastic gradient descent and the online limited-memory BFGS method.

机器学习统计学习优化算法随机梯度下降