An Empirical Investigation of Catastrophic Forgeting in Gradient-Based Neural Networks

By Ian J. Goodfellow, Mehdi Mirza, Xia Da, Aaron C. Courville, Yoshua Bengio. 2013

1,666 citations recorded by the source corpus.

Catastrophic forgetting is a problem faced by many machine learning models and algorithms. When trained on one task, then trained on a second task, many machine learning models "forget" how to perform the first task. This is widely believed to be a serious problem for neural networks. Here, we investigate the extent to which the catastrophic forgetting problem occurs for modern neural networks, comparing both established and recent gradient-based training algorithms and activation functions. We also examine the effect of the relationship between the first task and the second task on catastrophic forgetting. We find that it is always best to train using the dropout algorithm--the dropout algorithm is consistently best at adapting to the new task, remembering the old task, and has the best tradeoff curve between these two extremes. We find that different tasks and relationships between tasks result in very different rankings of activation function performance. This suggests the choice of activation function should always be cross-validated.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Model Architectures, Optimization & Training

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Yoshua Bengio
  2. Sergey Levine
  3. Mohit Bansal
  4. Dacheng Tao
  5. Jianfeng Gao
  6. Trevor Darrell
  7. Quoc V. Le
  8. Tinne Tuytelaars

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Trevor Darrell
  4. Ming-Hsuan Yang
  5. Xiaogang Wang
  6. Bernt Schiele
  7. Andrea Vedaldi
  8. Andrew Zisserman

In Model Architectures

  1. Yoshua Bengio
  2. Quoc V. Le
  3. Aaron C. Courville
  4. J. Urgen Schmidhuber
  5. Kyunghyun Cho
  6. Ruslan Salakhutdinov
  7. Dacheng Tao
  8. Alan L. Yuille

In Optimization & Training

  1. Yoshua Bengio
  2. Razvan Pascanu
  3. Simon Lacoste-Julien
  4. Frederik Benzing
  5. Stefano Soatto
  6. Quoc V. Le
  7. Christopher Kanan
  8. Peter Richtárik

Citations in the corpus