English

Recurrent neural network training with preconditioned stochastic gradient descent

Machine Learning 2016-12-09 v2 Machine Learning

Abstract

This paper studies the performance of a recently proposed preconditioned stochastic gradient descent (PSGD) algorithm on recurrent neural network (RNN) training. PSGD adaptively estimates a preconditioner to accelerate gradient descent, and is designed to be simple, general and easy to use, as stochastic gradient descent (SGD). RNNs, especially the ones requiring extremely long term memories, are difficult to train. We have tested PSGD on a set of synthetic pathological RNN learning problems and the real world MNIST handwritten digit recognition task. Experimental results suggest that PSGD is able to achieve highly competitive performance without using any trick like preprocessing, pretraining or parameter tweaking.

Keywords

Cite

@article{arxiv.1606.04449,
  title  = {Recurrent neural network training with preconditioned stochastic gradient descent},
  author = {Xi-Lin Li},
  journal= {arXiv preprint arXiv:1606.04449},
  year   = {2016}
}

Comments

Supplemental materials including Matlab code are put at https://sites.google.com/site/lixilinx/home/psgd

R2 v1 2026-06-22T14:25:12.315Z