Abstract:Enabling a neural network to sequentially learn multiple tasks is of great significance for expanding the applicability of neural networks in real-world applications. However, artificial neural networks face the well-known problem of catastrophic forgetting. What is worse, the degradation of previously learned skills becomes more severe as the task sequence increases, known as the long-term catastrophic forgetting. It is due to two facts: first, as the model learns more tasks, the intersection of the low-error parameter subspace satisfying for these tasks becomes smaller or even does not exist; second, when the model learns a new task, the cumulative error keeps increasing as the model tries to protect the parameter configuration of previous tasks from interference. Inspired by the memory consolidation mechanism in mammalian brains with synaptic plasticity, we propose a confrontation mechanism in which Adversarial Neural Pruning and synaptic Consolidation (ANPyC) is used to overcome the long-term catastrophic forgetting issue. The neural pruning acts as long-term depression to prune task-irrelevant parameters, while the novel synaptic consolidation acts as long-term potentiation to strengthen task-relevant parameters. During the training, this confrontation achieves a balance in that only crucial parameters remain, and non-significant parameters are freed to learn subsequent tasks. ANPyC avoids forgetting important information and makes the model efficient to learn a large number of tasks. Specifically, the neural pruning iteratively relaxes the current task's parameter conditions to expand the common parameter subspace of the task; the synaptic consolidation strategy, which consists of a structure-aware parameter-importance measurement and an element-wise parameter updating strategy, decreases the cumulative error when learning new tasks. Our approach encourages the synapse to be sparse and polarized, which enables long-term learning and memory. ANPyC exhibits effectiveness and generalization on both image classification and generation tasks with multiple layer perceptron, convolutional neural networks, and generative adversarial networks, and variational autoencoder. The full source code is available at https://github.com/GeoX-Lab/ANPyC.

Hard ASH: Sparsity and the right optimizer make a continual learner

Efficient Spiking Neural Networks with Sparse Selective Activation for Continual Learning

Overcoming Long-Term Catastrophic Forgetting Through Adversarial Neural Pruning and Synaptic Consolidation

Elephant Neural Networks: Born to Be a Continual Learner

Sparsity and Heterogeneous Dropout for Continual Learning in the Null Space of Neural Activations

Representations learnt by SGD and Adaptive learning rules: Conditions that vary sparsity and selectivity in neural networks

ASP: Learning to Forget with Adaptive Synaptic Plasticity in Spiking Neural Networks

ALRS: an Attention Loss Function Based on Row-Sparsity for Incremental Learning.

AdaHAT: Adaptive Hard Attention to the Task in Task-Incremental Learning

Learning Neural Networks with Sparse Activations

Convergence of a Relaxed Variable Splitting Method for Learning Sparse Neural Networks via $\ell_1, \ell_0$, and transformed-$\ell_1$ Penalties

Does the Adam Optimizer Exacerbate Catastrophic Forgetting?

Pushing the Limits of Sparsity: A Bag of Tricks for Extreme Pruning

SNAP: Stopping Catastrophic Forgetting in Hebbian Learning with Sigmoidal Neuronal Adaptive Plasticity

EXACFS -- A CIL Method to mitigate Catastrophic Forgetting

DSD$^2$: Can We Dodge Sparse Double Descent and Compress the Neural Network Worry-Free?

SSN: Learning Sparse Switchable Normalization via SparsestMax

Optimizing Spca-based Continual Learning: A Theoretical Approach.

Learning Sparse Representations Incrementally in Deep Reinforcement Learning

Learning Activation Functions for Sparse Neural Networks

Wide Neural Networks Forget Less Catastrophically