arXiv · 1301.3641
Training Neural Networks with Stochastic Hessian-Free Optimization
Abstract
Hessian-free (HF) optimization has been successfully used for training deep autoencoders and recurrent networks. HF uses the conjugate gradient algorithm to construct update directions through curvature-vector products that can be computed on the same order of time as gradients. In this paper we exploit this property and study stochastic HF with gradient and curvature mini-batches independent of the dataset size. We modify Martens' HF for these settings and integrate dropout, a method for preventing co-adaptation of feature detectors, to guard against overfitting. Stochastic Hessian-free optimization gives an intermediary between SGD and HF that achieves competitive performance on both classification and deep autoencoder experiments.
Explore related subjects
Keep this discovery
Ryan Kiros. 2013-05-01. Training Neural Networks with Stochastic Hessian-Free Optimization. https://arxiv.org/abs/1301.3641
Cite the original work for its findings. Save a collection to share your selection of sources.