Learning curves for stochastic gradient descent in linear feedforward networks

Justin Werfel, Xiaohui Xie, H. Sebastian Seung

Research output: Contribution to journalArticlepeer-review

40 Scopus citations

Abstract

Gradient-following learning methods can encounter problems of implementation in many applications, and stochastic variants are sometimes used to overcome these difficulties. We analyze three online training methods used with a linear perceptron: direct gradient descent, node perturbation, and weight perturbation. Learning speed is defined as the rate of exponential decay in the learning curves. When the scalar parameter that controls the size of weight updates is chosen to maximize learning speed, node perturbation is slower than direct gradient descent by a factor equal to the number of output units; weight perturbation is slower still by an additional factor equal to the number of input units. Parallel perturbation allows faster learning than sequential perturbation, by a factor that does not depend on network size. We also characterize how uncertainty in quantities used in the stochastic updates affects the learning curves. This study suggests that in practice, weight perturbation may be slow for large networks, and node perturbation can have performance comparable to that of direct gradient descent when there are few output units. However, these statements depend on the specifics of the learning problem, such as the input distribution and the target function, and are not universally applicable.

Original languageEnglish (US)
Pages (from-to)2699-2718
Number of pages20
JournalNeural computation
Volume17
Issue number12
DOIs
StatePublished - Dec 2005
Externally publishedYes

All Science Journal Classification (ASJC) codes

  • Arts and Humanities (miscellaneous)
  • Cognitive Neuroscience

Fingerprint Dive into the research topics of 'Learning curves for stochastic gradient descent in linear feedforward networks'. Together they form a unique fingerprint.

Cite this