A new inexact stochastic recursive gradient descent algorithm with Barzilai–Borwein step size in machine learning

Yi-ming Yang,Fu-sheng Wang,Jin-xiang Li,Yuan-yuan Qin
DOI: https://doi.org/10.1007/s11071-022-07987-2
IF: 5.741
2022-10-24
Nonlinear Dynamics
Abstract:The inexact SARAH (iSARAH) algorithm as a variant of SARAH algorithm for variance reduction has recently surged into prominence for solving large-scale optimization problems in the context of machine learning. The performance of the iSARAH significantly depends on the choice of step-size sequence. In this paper, we develop a new algorithm called iSARAH-BB, which employs the Barzilai–Borwein (BB) method to automatically compute step size based on SARAH. By introducing this adaptive step size in the design of the new algorithm, iSARAH-BB can take better advantages of both iSARAH and BB methods. Finally, we analyze the convergence rate and the complexity of the new algorithm under the usual assumptions. Numerical experiments on standard datasets indicate that our proposed iSARAH-BB algorithm is robust to the selection of the initial step size, and it is effective and more competitive than the existing algorithms.
engineering, mechanical,mechanics
What problem does this paper attempt to address?