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 | 248 | |
| Class-incremental learning | CIFAR-100 | Average Accuracy91.13 | 116 | |
| Class-incremental learning | ImageNet-R | Average Accuracy81.74 | 112 | |
| Class-incremental learning | CIFAR-100 10 (test) | Average Top-1 Accuracy92.1 | 105 | |
| Class-incremental learning | ImageNet A | Average Accuracy65.34 | 86 | |
| Continual Learning | CIFAR100 Split | -- | 85 | |
| Class-incremental learning | ImageNet-R B0 Inc20 | Last Accuracy77.45 | 79 | |
| Class-incremental learning | CUB200 10 Tasks | FN (Final Acc)6.27 | 59 | |
| Continual Learning | CIFAR-100 | Accuracy91.1 | 56 |