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

DER: Dynamically Expandable Representation for Class Incremental Learning

About

We address the problem of class incremental learning, which is a core step towards achieving adaptive vision intelligence. In particular, we consider the task setting of incremental learning with limited memory and aim to achieve better stability-plasticity trade-off. To this end, we propose a novel two-stage learning approach that utilizes a dynamically expandable representation for more effective incremental concept modeling. Specifically, at each incremental step, we freeze the previously learned representation and augment it with additional feature dimensions from a new learnable feature extractor. This enables us to integrate new visual concepts with retaining learned knowledge. We dynamically expand the representation according to the complexity of novel concepts by introducing a channel-level mask-based pruning strategy. Moreover, we introduce an auxiliary loss to encourage the model to learn diverse and discriminate features for novel concepts. We conduct extensive experiments on the three class incremental learning benchmarks and our method consistently outperforms other methods with a large margin.

Shipeng Yan, Jiangwei Xie, Xuming He• 2021

Related benchmarks

TaskDatasetResultRank
Class-incremental learningCIFAR-100
Averaged Incremental Accuracy75.92
234
Class-incremental learningCIFAR100 (test)
Avg Acc75.36
76
Class-incremental learningCIFAR-100 10 (test)
Average Top-1 Accuracy75.36
75
Class-incremental learningImageNet-100
Avg Acc80.53
74
Class-incremental learningCIFAR100 B50 (test)
Average Accuracy74.61
67
Class-incremental learningCIFAR-100
Average Accuracy73
60
Class-incremental learningImageNet-100 B=50, C=10 1.0
Avg Incremental Acc85.17
42
Class-incremental learningCIFAR100-B0 (test)
Accuracy76.8
40
Incremental LearningCIFAR100 10 steps
Final Step Performance65.22
39
Incremental LearningCIFAR100 50 steps
Last Accuracy59.76
36
Showing 10 of 76 rows
...

Other info

Code

Follow for update