Share on Facebook Tweet on Twitter Share on LinkedIn Share by email
Stable adaptive momentum for rapid online learning in nonlinear systems

Thore Graepel and Nicol N. Schraudolph

Abstract

We consider the problem of developing rapid, stable, and scalable stochastic gradient descent algorithms for optimisation of very large nonlinear systems. Based on earlier work by Orr et al. on adaptive momentum—an efficient yet extremely unstable stochastic gradient descent algorithm—we develop a stabilised adaptive momentum algorithm that is suitable for noisy nonlinear optimisation problems. The stability is improved by introducing a forgetting factor lambda between zero and one that smoothes the trajectory and enables adaptation in non-stationary environments. The scalability of the new algorithm follows from the fact that at each iteration the multiplication by the curvature matrix can be achieved in O(n) steps using automatic differentiation tools. We illustrate the behaviour of the new algorithm on two examples: a linear neuron with squared loss and highly correlated inputs, and a multilayer perceptron applied to the four regions benchmark task.

Details

Publication typeInproceedings
Published inProceedings of the International Conference on Neural Networks, ICANN 2002
Pages450–455
Number2415
SeriesLecture Notes in Computer Science
PublisherSpringer
> Publications > Stable adaptive momentum for rapid online learning in nonlinear systems