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

Energy-Based Models for Continual Learning

About

We motivate Energy-Based Models (EBMs) as a promising model class for continual learning problems. Instead of tackling continual learning via the use of external memory, growing models, or regularization, EBMs change the underlying training objective to cause less interference with previously learned information. Our proposed version of EBMs for continual learning is simple, efficient, and outperforms baseline methods by a large margin on several benchmarks. Moreover, our proposed contrastive divergence-based training objective can be combined with other continual learning methods, resulting in substantial boosts in their performance. We further show that EBMs are adaptable to a more general continual learning setting where the data distribution changes without the notion of explicitly delineated tasks. These observations point towards EBMs as a useful building block for future continual learning methods.

Shuang Li, Yilun Du, Gido M. van de Ven, Igor Mordatch• 2020

Related benchmarks

TaskDatasetResultRank
Class-incremental learningCIFAR10 (test)
Average Accuracy38.84
59
Class-incremental learningMNIST (test)
Average Accuracy53.12
35
Image ClassificationSplitCIFAR-10 (test)
Accuracy (TInfer Final)0.3884
7
Showing 3 of 3 rows

Other info

Follow for update