Arthur Douillard, Matthieu Cord, Charles Ollion, Thomas Robert, Eduardo Valle
Lifelong learning has attracted much attention, but existing works still struggle to fight catastrophic forgetting and accumulate knowledge over long stretches of incremental learning. In this work, we propose PODNet, a model inspired by representation learning. By carefully balancing the compromise between remembering the old classes and learning new ones, PODNet fights catastrophic forgetting, even over very long runs of small incremental tasks --a setting so far unexplored by current works. PODNet innovates on existing art with an efficient spatial-based distillation-loss applied throughout the model and a representation comprising multiple proxy vectors for each class. We validate those innovations thoroughly, comparing PODNet with three state-of-the-art models on three datasets: CIFAR100, ImageNet100, and ImageNet1000. Our results showcase a significant advantage of PODNet over existing art, with accuracy gains of 12.10, 6.51, and 2.85 percentage points, respectively. Code is available at https://github.com/arthurdouillard/incremental_learning.pytorch
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Incremental Learning | CIFAR-100 - 50 classes + 10 steps of 5 classes | Average Incremental Accuracy | 63.19 | PODNet (CNN) |
| Incremental Learning | ImageNet-100 - 50 classes + 25 steps of 2 classes | Average Incremental Accuracy | 67.28 | PODNet |
| Incremental Learning | CIFAR-100 - 50 classes + 5 steps of 10 classes | Average Incremental Accuracy | 64.83 | PODNet (CNN) |
| Incremental Learning | CIFAR-100-B0(5steps of 20 classes) | Average Incremental Accuracy | 66.7 | PODNet |
| Incremental Learning | ImageNet - 500 classes + 5 steps of 100 classes | Average Incremental Accuracy | 66.95 | PODNet |
| Incremental Learning | CIFAR-100 - 50 classes + 25 steps of 2 classes | Average Incremental Accuracy | 60.72 | PODNet |
| Incremental Learning | CIFAR-100 - 50 classes + 50 steps of 1 class | Average Incremental Accuracy | 57.98 | PODNet |
| Incremental Learning | ImageNet - 500 classes + 10 steps of 50 classes | Average Incremental Accuracy | 64.13 | PODNet |
| Incremental Learning | ImageNet-100 - 50 classes + 50 steps of 1 class | Average Incremental Accuracy | 62.08 | PODNet |
| Incremental Learning | ImageNet-100 - 50 classes + 10 steps of 5 classes | Average Incremental Accuracy | 73.14 | PODNet |
| Incremental Learning | ImageNet-100 - 50 classes + 5 steps of 10 classes | Average Incremental Accuracy | 75.82 | PODNet |