The Improved Adaptive Algorithm of Deep Learning with Barzilai-Borwein Step Size

PubDate: Aug 2022

Teams:  Wenzhou University；Hangzhou Normal University

Writers: Zhi-Jun Wang; He-Bei Gao; Bin-Shuang Zhang

PDF: The Improved Adaptive Algorithm of Deep Learning with Barzilai-Borwein Step Size

Abstract

To solve the problem that it is difficult to determine the learning rate when training a neural network model, this paper proposes an improved adaptive algorithm based on the Barzilai-Borwein (BB) step size. In this paper, the new algorithm accelerates the model’s training through the second-order momentum and adapts the learning rate according to the BB step size. We also set an adequate range for the learning rate to ensure the stability of adaptive adjustment and reduce the error of step size. Compared with different algorithms in a series of popular models, the new algorithm significantly avoids the tediousness of manually adjusting the learning rate and helps to improve the convergence speed. The results show that the new algorithm is feasible and effective.