Bayesian Structure Adaptation for Continual Learning
About
Continual Learning is a learning paradigm where learning systems are trained with sequential or streaming tasks. Two notable directions among the recent advances in continual learning with neural networks are ($i$) variational Bayes based regularization by learning priors from previous tasks, and, ($ii$) learning the structure of deep networks to adapt to new tasks. So far, these two approaches have been orthogonal. We present a novel Bayesian approach to continual learning based on learning the structure of deep neural networks, addressing the shortcomings of both these approaches. The proposed model learns the deep structure for each task by learning which weights to be used, and supports inter-task transfer through the overlapping of different sparse subsets of weights learned by different tasks. Experimental results on supervised and unsupervised benchmarks shows that our model performs comparably or better than recent advances in continual learning setting.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Class-incremental learning | CIFAR-100 20 tasks | -- | 58 | |
| Task-Incremental Learning | Tiny-ImageNet 20 tasks | Average Accuracy57.1 | 54 | |
| Task-Incremental Learning | CIFAR-100 10 tasks | Backward Transfer-6.4 | 44 | |
| Task-Incremental Learning | CIFAR-100 (20-split) | Accuracy76.3 | 27 | |
| Task-Incremental Learning | CelebA 20 binary attributes | Average Accuracy87.1 | 15 | |
| Task-Incremental Learning | EMNIST 20 tasks | Average Accuracy87.8 | 15 | |
| Task-Incremental Learning | ImageNet 100 tasks | Average Accuracy54.8 | 15 | |
| Task-Incremental Learning | CIFAR-100 (10-splits) | Average Accuracy69.3 | 15 | |
| Task-Incremental Learning | CelebA 20 tasks | Backward Transfer0.9 | 11 | |
| Task-Incremental Learning | EMNIST 20 tasks | Backward Transfer0.3 | 11 |