arXiv · 1709.04546
Normalized Direction-preserving Adam
Abstract
Adaptive optimization algorithms, such as Adam and RMSprop, have shown better optimization performance than stochastic gradient descent (SGD) in some scenarios. However, recent studies show that they often lead to worse generalization performance than SGD, especially for training deep neural networks (DNNs). In this work, we identify the reasons that Adam generalizes worse than SGD, and develop a variant of Adam to eliminate the generalization gap. The proposed method, normalized direction-preserving Adam (ND-Adam), enables more precise control of the direction and step size for updating weight vectors, leading to significantly improved generalization performance. Following a similar rationale, we further improve the generalization performance in classification tasks by regularizing the softmax logits. By bridging the gap between SGD and Adam, we also hope to shed light on why certain optimization algorithms generalize better than others.
Explore related subjects
Keep this discovery
Zijun Zhang, Lin Ma, Zongpeng Li, Chuan Wu. 2017-09-13. Normalized Direction-preserving Adam. https://arxiv.org/abs/1709.04546
Cite the original work for its findings. Save a collection to share your selection of sources.