Skip to content

Implementation of "Overcoming catastrophic forgetting in neural networks" in Tensorflow

Notifications You must be signed in to change notification settings

spragunr/overcoming-catastrophic

 
 

Repository files navigation

This is an implementation of "Overcoming catastrophic forgetting in neural networks" (https://arxiv.org/abs/1612.00796) for supervised learning in TensorFlow.

model.py defines a simple fully-connected network and methods to compute the diagonal of the Fisher information matrix.

experiment.ipynb trains and tests a single network on three MNIST classification tasks sequentially (i.e., once the network begins training on a given task, it is never exposed to previous task training data again).

About

Implementation of "Overcoming catastrophic forgetting in neural networks" in Tensorflow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 78.3%
  • Python 21.5%
  • Shell 0.2%