arXiv · 2004.09740
AdaX: Adaptive Gradient Descent with Exponential Long Term Memory
Abstract
Although adaptive optimization algorithms such as Adam show fast convergence in many machine learning tasks, this paper identifies a problem of Adam by analyzing its performance in a simple non-convex synthetic problem, showing that Adam's fast convergence would possibly lead the algorithm to local minimums. To address this problem, we improve Adam by proposing a novel adaptive gradient descent algorithm named AdaX. Unlike Adam that ignores the past gradients, AdaX exponentially accumulates the long-term gradient information in the past during training, to adaptively tune the learning rate. We thoroughly prove the convergence of AdaX in both the convex and non-convex settings. Extensive experiments show that AdaX outperforms Adam in various tasks of computer vision and natural language processing and can catch up with Stochastic Gradient Descent.
Explore related subjects
Keep this discovery
Wenjie Li, Zhaoyang Zhang, Xinjiang Wang, Ping Luo. 2020-04-21. AdaX: Adaptive Gradient Descent with Exponential Long Term Memory. https://arxiv.org/abs/2004.09740
Cite the original work for its findings. Save a collection to share your selection of sources.