Improving Neural Networks by Preventing Co-Adaptation of Feature Detectors

Venue

Publication Year

Keywords

Computer Science - Computer Vision and Pattern Recognition,Computer Science - Machine Learning,Computer Science - Neural and Evolutionary Computing

Authors

  • Geoffrey E. Hinton
  • Nitish Srivastava
  • Alex Krizhevsky
  • Ilya Sutskever
  • Ruslan R. Salakhutdinov

Abstract

When a large feedforward neural network is trained on a small training set, it typically performs poorly on held-out test data. This "overfitting" is greatly reduced by randomly omitting half of the feature detectors on each training case. This prevents complex co-adaptations in which a feature detector is only helpful in the context of several other specific feature detectors. Instead, each neuron learns to detect a feature that is generally helpful for producing the correct answer given the combinatorially large variety of internal contexts in which it must operate. Random "dropout" gives big improvements on many benchmark tasks and sets new records for speech and object recognition.