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 | 281 | |
| Class-incremental learning | CIFAR-100 | Average Accuracy91.13 | 150 | |
| Class-incremental learning | ImageNet-R | Last Accuracy77.75 | 147 | |
| Continual Learning | CIFAR100 Split | -- | 117 | |
| Class-incremental learning | ImageNet A | Average Accuracy65.34 | 110 | |
| Class-incremental learning | CIFAR-100 10 (test) | Average Top-1 Accuracy92.1 | 105 | |
| Class-incremental learning | ImageNet-R B0 Inc20 | Last Accuracy77.45 | 98 | |
| Image Classification | ImageNet A | -- | 73 | |
| Class-incremental learning | CIFAR100 10 Tasks | Accuracy86.95 | 66 |