Stable adaptive momentum for rapid online learning in nonlinear systems

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.

grasch02.pdf
PDF file

In  Proceedings of the International Conference on Neural Networks, ICANN 2002

Publisher  Springer

Details

TypeInproceedings
Pages450–455
Number2415
SeriesLecture Notes in Computer Science
> Publications > Stable adaptive momentum for rapid online learning in nonlinear systems