Expandable Subspace Ensemble for Pre-Trained Model-Based Class-Incremental Learning
About
Class-Incremental Learning (CIL) requires a learning system to continually learn new classes without forgetting. Despite the strong performance of Pre-Trained Models (PTMs) in CIL, a critical issue persists: learning new classes often results in the overwriting of old ones. Excessive modification of the network causes forgetting, while minimal adjustments lead to an inadequate fit for new classes. As a result, it is desired to figure out a way of efficient model updating without harming former knowledge. In this paper, we propose ExpAndable Subspace Ensemble (EASE) for PTM-based CIL. To enable model updating without conflict, we train a distinct lightweight adapter module for each new task, aiming to create task-specific subspaces. These adapters span a high-dimensional feature space, enabling joint decision-making across multiple subspaces. As data evolves, the expanding subspaces render the old class classifiers incompatible with new-stage spaces. Correspondingly, we design a semantic-guided prototype complement strategy that synthesizes old classes' new features without using any old class instance. Extensive experiments on seven benchmark datasets verify EASE's state-of-the-art performance. Code is available at: https://github.com/sun-hailong/CVPR24-Ease
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | CIFAR-100 | Accuracy87.81 | 302 | |
| Class-incremental learning | CIFAR-100 | Averaged Incremental Accuracy91.51 | 234 | |
| Class-incremental learning | ImageNet-R | Average Accuracy81.74 | 103 | |
| Class-incremental learning | ImageNet A | Average Accuracy65.34 | 86 | |
| Class-incremental learning | CIFAR-100 | Average Accuracy85.27 | 60 | |
| Continual Learning | CIFAR-100 | Accuracy91.1 | 56 | |
| Class-incremental learning | ImageNet-R 10-task | -- | 44 | |
| Class-incremental learning | ObjectNet | Average Accuracy70.84 | 40 | |
| Class-incremental learning | CIFAR-100 B0_Inc5 | Average Accuracy90.79 | 36 | |
| Class-incremental learning | ImageNet-R 20-task | Average Accuracy82.15 | 33 |