This repository contains the code to reproduce the experiments present in our paper titled Loss of Plasticity in Deep Continual Learning. A talk about this work can be found here.
Artificial neural networks, deep-learning methods and the backpropagation algorithm form the foundation of modern machine learning and artificial intelligence. These methods are almost always used in two phases, one in which the weights of the network are updated and one in which the weights are held constant while the network is used or evaluated. This contrasts with natural learning and many applications, which require continual learning. It has been unclear whether deep-learning methods work in continual-learning settings. Here we show that they do not—that standard deep-learning methods gradually lose plasticity in continual-learning settings until they learn no better than a shallow network. We show such loss of plasticity using the classic ImageNet dataset and reinforcement-learning problems across a wide range of variations in the network and the learning algorithm. Plasticity is maintained indefinitely only by algorithms that continually inject diversity into the network, such as our continual backpropagation algorithm, a variation of backpropagation in which a small fraction of less-used units are continually and randomly reinitialized. Our results indicate that methods based on gradient descent are not enough—that sustained deep learning requires a random, non-gradient component to maintain variability and plasticity.
Please cite our work if you find it useful:
@article{dohare2024loss,
title={Loss of Plasticity in Deep Continual Learning},
author={Dohare, Shibhansh and Hernandez-Garcia, J. Fernando and Lan, Qingfeng and Rahman, Parash and Mahmood, A. Ruapm and Sutton, Richard S.},
journal={Nature},
volume={632},
pages={768---774},
year={2024},
publisher={Nature Publishing Group UK London}
}- lop/algos: All the algorithms used in the paper, including our new continual backpropagation algorithm.
- lop/nets: The network architectures used in the paper.
- lop/imagenet: Demonstration and mitigation of loss of plasticity in a task-incremental problem using ImageNet.
- lop/incremental_cifar: Demonstration and mitigation of loss of plasticity in a class-incremental problem.
- lop/slowly_changing_regression: A small problem for quick demonstration of loss of plasticity.
- lop/rl: Loss of plasticity in standard reinforcement learning problems using the PPO algorithm[1].
The README files in each subdirectory contains further information on the contents of the subdirectory.
This package only requires a standard computed with sufficient RAM (8GB+) to reproduce the experimental results. However, a GPU can significantly speed up experiments with larger networks such as the residual networks in lop/incremental_cifar. Internet connection is required to download many of the datasets and packages.
The package has been tested on Ubuntu 20.04 and python3.8. We expect this package to work on all machines that support all the packages listed in requirements.txt
Create a virtual environment
mkdir ~/envs
virtualenv --no-download --python=/usr/bin/python3.8 ~/envs/lop
source ~/envs/lop/bin/activate
pip3 install --no-index --upgrade pipDownload the repository and install the requirements
git clone https://github.com/shibhansh/loss-of-plasticity.git
cd loss-of-plasticity
pip3 install -r requirements.txt
pip3 install -e .Add this lines in your ~/.zshrc or ~/.bashrc
source ~/envs/lop/bin/activateInstallation on a normal laptop with good internet connection should only take a few minutes
[1] Schulman, J., Wolski, F., Dhariwal, P., Radford, A., & Klimov, O. (2017). Proximal policy optimization algorithms. arXiv preprint arXiv:1707.06347.