Command Palette
Search for a command to run...
PODNet: Pooled Outputs Distillation for Small-Tasks Incremental Learning
Arthur Douillard Matthieu Cord Charles Ollion Thomas Robert Eduardo Valle
Abstract
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
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| incremental-learning-on-cifar-100-50-classes | PODNet | Average Incremental Accuracy: 57.98 |
| incremental-learning-on-cifar-100-50-classes-1 | PODNet | Average Incremental Accuracy: 60.72 |
| incremental-learning-on-cifar-100-50-classes-2 | PODNet (CNN) | Average Incremental Accuracy: 63.19 |
| incremental-learning-on-cifar-100-50-classes-3 | PODNet (CNN) | Average Incremental Accuracy: 64.83 |
| incremental-learning-on-cifar-100-b0-5steps | PODNet | Average Incremental Accuracy: 66.70 |
| incremental-learning-on-imagenet-100-50 | PODNet | Average Incremental Accuracy: 62.08 |
| incremental-learning-on-imagenet-100-50-1 | PODNet | Average Incremental Accuracy: 67.28 |
| incremental-learning-on-imagenet-100-50-2 | PODNet | Average Incremental Accuracy: 73.14 |
| incremental-learning-on-imagenet-100-50-3 | PODNet | Average Incremental Accuracy: 75.82 |
| incremental-learning-on-imagenet-500-classes-1 | PODNet | Average Incremental Accuracy: 66.95 |
| incremental-learning-on-imagenet-500-classes-2 | PODNet | Average Incremental Accuracy: 64.13 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.