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

Model Zoo: A Growing "Brain" That Learns Continually

About

This paper argues that continual learning methods can benefit by splitting the capacity of the learner across multiple models. We use statistical learning theory and experimental analysis to show how multiple tasks can interact with each other in a non-trivial fashion when a single model is trained on them. The generalization error on a particular task can improve when it is trained with synergistic tasks, but can also deteriorate when trained with competing tasks. This theory motivates our method named Model Zoo which, inspired from the boosting literature, grows an ensemble of small models, each of which is trained during one episode of continual learning. We demonstrate that Model Zoo obtains large gains in accuracy on a variety of continual learning benchmark problems. Code is available at https://github.com/grasp-lyrl/modelzoo_continual.

Rahul Ramesh, Pratik Chaudhari• 2021

Related benchmarks

TaskDatasetResultRank
Continual LearningCIFAR100 Split
Average Per-Task Accuracy94.99
85
Continual LearningPermuted MNIST
Mean Test Accuracy98.03
44
Continual LearningSplit MNIST
Mean Test Accuracy99.98
19
Continual LearningRotated-MNIST
Accuracy99.66
13
Continual LearningSplit Mini-ImageNet
Avg Per-Task Accuracy96.84
11
Continual LearningCIFAR100 Coarse
Avg Per-Task Accuracy84.27
9
Continual LearningSplit-CIFAR10
Avg Per-Task Accuracy98.68
6
Showing 7 of 7 rows

Other info

Code

Follow for update