Scalable Learning with Incremental Probabilistic PCA

Incremental class learning is the classification problem of learning a model where instances from new object classes are added sequentially, and it is desired that the model be retrained only on the new classes with minimal training on the old classes. One major problem facing class incremental learning is catastrophic forgetting, where the updated model forgets the old classes and focuses only on the new classes. This paper proposes a simple and novel incremental class learning method that uses a self-supervised pretrained feature extractor to obtain meaningful features and trains Probabilistic PCA models on the extracted features for each class separately. The Mahalanobis distance is used to obtain the classification result, and an equivalent equation is derived to make the approach computationally affordable. Experiments on standard and large datasets show that the proposed approach outperforms existing state of the art incremental learning methods by a large margin. The fact that the model is trained on each class separately makes it applicable to training on very large datasets such as the whole ImageNet with more than 10,000 classes.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Benchmark
Class Incremental Learning CIFAR-100 - 50 classes + 10 steps of 5 classes PPCA-CLIP Final Accuracy 69.71 # 2
Class Incremental Learning CIFAR-100 - 50 classes + 10 steps of 5 classes PPCA-SWSL Final Accuracy 77.07 # 1
Class Incremental Learning CIFAR-100 - 50 classes + 5 steps of 10 classes PPCA-SWSL Final Accuracy 77.07 # 1
Incremental Learning CIFAR-100 - 50 classes + 5 steps of 10 classes PPCA-SWSL Final Accuracy 77.07 # 1
Incremental Learning CIFAR-100 - 50 classes + 5 steps of 10 classes PPCA-CLIP Final Accuracy 69.71 # 2
Class Incremental Learning CIFAR-100 - 50 classes + 5 steps of 10 classes PPCA-CLIP Final Accuracy 69.71 # 2
Incremental Learning ImageNet-10k - 5225 classes + 5 steps of 1045 classes PPCA-CLIP Final Accuracy 35.42 # 1
Incremental Learning ImageNet - 500 classes + 10 steps of 50 classes PPCA-CLIP Final Accuracy 71.25 # 1
Incremental Learning ImageNet - 500 classes + 5 steps of 100 classes PPCA-CLIP Final Accuracy 71.25 # 1

Methods