Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Achieving a Better Stability-Plasticity Trade-off via Auxiliary Networks in Continual Learning

About

In contrast to the natural capabilities of humans to learn new tasks in a sequential fashion, neural networks are known to suffer from catastrophic forgetting, where the model's performances on old tasks drop dramatically after being optimized for a new task. Since then, the continual learning (CL) community has proposed several solutions aiming to equip the neural network with the ability to learn the current task (plasticity) while still achieving high accuracy on the previous tasks (stability). Despite remarkable improvements, the plasticity-stability trade-off is still far from being solved and its underlying mechanism is poorly understood. In this work, we propose Auxiliary Network Continual Learning (ANCL), a novel method that applies an additional auxiliary network which promotes plasticity to the continually learned model which mainly focuses on stability. More concretely, the proposed framework materializes in a regularizer that naturally interpolates between plasticity and stability, surpassing strong baselines on task incremental and class incremental scenarios. Through extensive analyses on ANCL solutions, we identify some essential principles beneath the stability-plasticity trade-off.

Sanghwan Kim, Lorenzo Noci, Antonio Orvieto, Thomas Hofmann• 2023

Related benchmarks

TaskDatasetResultRank
Class-incremental learningCIFAR-100
Averaged Incremental Accuracy63.15
234
Incremental LearningTinyImageNet
Avg Incremental Accuracy63.79
83
Class-incremental learningCIFAR100 (test)
Avg Acc55.01
76
Continual LearningCIFAR-100
Accuracy81.5
56
Continual LearningImageNet mini
BWT-1.8
35
Continual LearningImageNet tiny
BWT-150
35
Class-incremental learningTinyImageNet (test)
Accuracy34.26
29
Task-Incremental LearningCIFAR-100
Accuracy79.99
24
Depth CompletionIndoor sequence NYUv2 → ScanNet → VOID
AF MAE9.73
24
Depth CompletionOutdoor sequence KITTI → Waymo → VKITTI--
18
Showing 10 of 13 rows

Other info

Code

Follow for update