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

OrCo: Towards Better Generalization via Orthogonality and Contrast for Few-Shot Class-Incremental Learning

About

Few-Shot Class-Incremental Learning (FSCIL) introduces a paradigm in which the problem space expands with limited data. FSCIL methods inherently face the challenge of catastrophic forgetting as data arrives incrementally, making models susceptible to overwriting previously acquired knowledge. Moreover, given the scarcity of labeled samples available at any given time, models may be prone to overfitting and find it challenging to strike a balance between extensive pretraining and the limited incremental data. To address these challenges, we propose the OrCo framework built on two core principles: features' orthogonality in the representation space, and contrastive learning. In particular, we improve the generalization of the embedding space by employing a combination of supervised and self-supervised contrastive losses during the pretraining phase. Additionally, we introduce OrCo loss to address challenges arising from data limitations during incremental sessions. Through feature space perturbations and orthogonality between classes, the OrCo loss maximizes margins and reserves space for the following incremental data. This, in turn, ensures the accommodation of incoming classes in the feature space without compromising previously acquired knowledge. Our experimental results showcase state-of-the-art performance across three benchmark datasets, including mini-ImageNet, CIFAR100, and CUB datasets. Code is available at https://github.com/noorahmedds/OrCo

Noor Ahmed, Anna Kukleva, Bernt Schiele• 2024

Related benchmarks

TaskDatasetResultRank
Class-incremental learningCIFAR-100
Averaged Incremental Accuracy46.8
234
Few-Shot Class-Incremental LearningminiImageNet (test)
Accuracy (Session 1)75.32
173
Few-Shot Class-Incremental LearningCIFAR100 (test)
Session 4 Top-1 Acc59.78
122
Few-Shot Class-Incremental LearningCUB200 (test)
Accuracy (Session 1)66.85
92
Few-Shot Class-Incremental LearningCUB-200
Session 1 Accuracy66.85
75
Few-Shot Class-Incremental LearningCIFAR100
Accuracy (S0)80.08
67
Few-Shot Class-Incremental LearningCORe50
BCR99.9
39
Few-Shot Class-Incremental LearningiCubWorld
BCR100
39
Few-Shot Class-Incremental LearningMiniImagenet
Avg Accuracy67.38
31
Few-Shot Class-Incremental LearningCIFAR100 multi-class 5-shot
BCR74.7
26
Showing 10 of 27 rows

Other info

Follow for update