HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Overcoming catastrophic forgetting in neural networks

James Kirkpatrick; Razvan Pascanu; Neil Rabinowitz; Joel Veness; Guillaume Desjardins; Andrei A. Rusu; Kieran Milan; John Quan; Tiago Ramalho; Agnieszka Grabska-Barwinska; Demis Hassabis; Claudia Clopath; Dharshan Kumaran; Raia Hadsell

Overcoming catastrophic forgetting in neural networks

Abstract

The ability to learn tasks in a sequential fashion is crucial to the development of artificial intelligence. Neural networks are not, in general, capable of this and it has been widely thought that catastrophic forgetting is an inevitable feature of connectionist models. We show that it is possible to overcome this limitation and train networks that can maintain expertise on tasks which they have not experienced for a long time. Our approach remembers old tasks by selectively slowing down learning on the weights important for those tasks. We demonstrate our approach is scalable and effective by solving a set of classification tasks based on the MNIST hand written digit dataset and by learning several Atari 2600 games sequentially.

Code Repositories

srvCodes/continual-learning-benchmark
pytorch
Mentioned in GitHub
xduan7/hat-cl
pytorch
Mentioned in GitHub
catid/never_forget
pytorch
Mentioned in GitHub
daskol/paper-reviews
Mentioned in GitHub
mu-y/lll-tts
pytorch
Mentioned in GitHub
wannabeOG/MAS-PyTorch
pytorch
Mentioned in GitHub
PCJohn/cata-forget
tf
Mentioned in GitHub
dalxndr/NEURO140
tf
Mentioned in GitHub
kamsyn95/CL_DNN
pytorch
Mentioned in GitHub
ariseff/overcoming-catastrophic
tf
Mentioned in GitHub
wjmacro/continualmt
pytorch
Mentioned in GitHub
g-u-n/pycil
pytorch
Mentioned in GitHub
Minhchuyentoancbn/Continual-Learning
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
continual-learning-on-20newsgroup-10-tasksEWC
F1 - macro: 0.9180
continual-learning-on-asc-19-tasksL2
F1 - macro: 0.5243
continual-learning-on-asc-19-tasksEWC
F1 - macro: 0.7452
continual-learning-on-dsc-10-tasksEWC
F1 - macro: 0.6576
continual-learning-on-f-celeba-10-tasksEWC
Acc: 0.6545

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp