Awesome

Few-Shot Class-Incremental Learning

ArXiv version: https://arxiv.org/abs/2004.10956

Few-Shot Class-Incremental Learning (FSCIL) is a novel problem setting for incremental learning, where a unified classifier is incrementally learned for new classes with very few training samples. In this repository, we provide baseline benchmarks and codes for implementation.

TOPology-preserving knowledge InCrementer (TOPIC)

The TOPIC framework for FSCIL is built with neural gas [1], a seminal algorithm that learns the topology of the data manifold in feature space via competitive Hebbian learning (CHL). Neural gas is capable of preserving the topology of any heterogenous, non-uniform manifold, making it perfect for FSCIL with imbalanced old/new classes. The following animation shows the neural gas + CHL process. You may refer to an online demo of [2] GNG for better understanding the topology learning of neural gas.

The TOPIC code will be released later after the commercial freezing period.

[1] Martinetz Thomas and Schulten Klaus. A "neural-gas" network learns topologies. Artificial Neural Networks, 1991. <br /> [2] Bernd Fritzke. A growing neural gas network learns topologies. Advances in neural information processing systems, 1995.

FSCIL Benchmark Settings

We modify CIFAR100, miniImageNet and CUB200 datasets for FSCIL. For CIFAR100 and miniImageNet, we choose 60 out of 100 classes as the base classes and split the rest 40 classes into 8 incremental learning sessions, each of which has 5 classes and 5 training samples per class. While for CUB200, we choose 100 base classes. The lists of image indexes used in each training session are stored in the directory

data/index_list

Comparison Results

In the following tables, we provide detailed test accuracies of each method under different settings of benchmark datasets and CNN models. The reported results are the mean accuracies averaged over 10 runs.

"Ft-CNN" indicates the baseline finetuning method with only softmax cross-entropy loss. "Joint-CNN" indicates the upper bound that trains the model on the joint set of all encountered training sets. "iCaRL*","EEIL*" and "NCM*" are representative class-incremental learning methods, where we adapt them to FSCIL setting for comparison.

These results are also reported in the figures of the paper.

CIFAR100, QuickNet, 5-way 5-shot (Fig.4 (a))

CIFAR100, ResNet18, 5-way 5-shot (Fig.4 (b))

miniImageNet, QuickNet, 5-way 5-shot (Fig.4 (c))

miniImageNet, ResNet18, 5-way 5-shot (Fig.4 (d))

CUB200, ResNet18, 10-way 5-shot

For CUB200, the comparison methods with their original learning rate settings have much worse test accuracy. We carefully tune their learning rates and greatly boost their original accuracy. In the table below, we use * to denote the settings with the improved accuracy.

Following and Citing FSCIL

FSCIL is an unsolved, challenging but practical incremental learning setting. It still has large research potentials for new solutions and better performances. When you wish to conduct your research using this setting or refer to the baseline results in your paper, please use the following BibTeX entry.

@inproceedings{tao2020fscil,
  author = {Tao, Xiaoyu and Hong, Xiaopeng and Chang, Xinyuan and Dong, Songlin and Wei, Xing and Gong, Yihong},
  title = {Few-Shot Class-Incremental Learning},
  booktitle = {Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  year = {2020}
}