Conjugate-gradient-based Adam for stochastic optimization and its application to deep learning
Abstract
This paper proposes a conjugate-gradient-based Adam algorithm blending Adam with nonlinear conjugate gradient methods and shows its convergence analysis. Numerical experiments on text classification and image classification show that the proposed algorithm can train deep neural network models in fewer epochs than the existing adaptive stochastic optimization algorithms can.
- Publication:
-
arXiv e-prints
- Pub Date:
- February 2020
- DOI:
- 10.48550/arXiv.2003.00231
- arXiv:
- arXiv:2003.00231
- Bibcode:
- 2020arXiv200300231K
- Keywords:
-
- Mathematics - Optimization and Control;
- Computer Science - Machine Learning