arrow
Return

A stochastic recursive gradient algorithm with inertial extrapolation for non-convex problems and machine learning

delete2025-01-09
delete0
delete
OA
AI
Z
Zhongyu Mo
C
Chen Ouyang
H
Hongtruong Pham
G
Gonglin Yuan *
DOI:10.1007/s13042-024-02524-6delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In recent years, the inertial extrapolation step has gained significant attention due to its capacity to expedite algorithm convergence. This technology has found widespread application across various algorithms. However, within the domain of machine learning, the utilization of extrapolation technology has yielded limited results. Therefore, we apply it to stochastic optimization algorithms to address non-convex and machine learning problems. By integrating the inertial extrapolation step and the modified Barzilai-Borwein (BB) technique into the SARAH framework, we propose an inertial stochastic recurrence gradient method. This method incorporates both the inertial extrapolation step and the improved BB technique. Through theoretical analysis presented in this paper, we demonstrate that the algorithm converges to a global optimum and analyze the linear convergence rate of the non-convex ((lambda) over tilde -gradient-dominated) objective functions. The numerical results obtained from evaluating three widely utilized machine learning problems clearly illustrate the superior performance and practical feasibility of the proposed algorithm.
Keywords:
Extrapolation
Algorithm
Non-convex
Global optimum
Convergence
Numerical results

Journal

International Journal of Machine Learning and Cybernetics cover
International Journal of Machine Learning and Cybernetics
IF:
2.7
Papers:
3.2K
Citations:
5.6K

Organization

T
Thai Nguyen University
Scholars:
1.0K
Papers: 625
Citations: 530
G
guangxi university
Scholars:
3.4W
Papers: 1.8W
Citations: 25
Cited Papers

Cited Papers

errShare
errSave
High-performance organic transistors based on solution-processed rubrene crystals directly grown on a polymeric dielectric
err2018-05-01
err0
PREAI
errKyoseung Sim; Hanah Na; Junyong Park; Junghyun Lee; Junghwan Do; Seungmoon Pyo
errShare
errSave
researcher View more