HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

On Tiny Episodic Memories in Continual Learning

Arslan Chaudhry; Marcus Rohrbach; Mohamed Elhoseiny; Thalaiyasingam Ajanthan; Puneet K. Dokania; Philip H. S. Torr; Marc'Aurelio Ranzato

On Tiny Episodic Memories in Continual Learning

Abstract

In continual learning (CL), an agent learns from a stream of tasks leveraging prior experience to transfer knowledge to future tasks. It is an ideal framework to decrease the amount of supervision in the existing learning algorithms. But for a successful knowledge transfer, the learner needs to remember how to perform previous tasks. One way to endow the learner the ability to perform tasks seen in the past is to store a small memory, dubbed episodic memory, that stores few examples from previous tasks and then to replay these examples when training for future tasks. In this work, we empirically analyze the effectiveness of a very small episodic memory in a CL setup where each training example is only seen once. Surprisingly, across four rather different supervised learning benchmarks adapted to CL, a very simple baseline, that jointly trains on both examples from the current task as well as examples stored in the episodic memory, significantly outperforms specifically designed CL approaches with and without episodic memory. Interestingly, we find that repetitive training on even tiny memories of past tasks does not harm generalization, on the contrary, it improves it, with gains between 7\% and 17\% when the memory is populated with a single example per class.

Code Repositories

chandar-lab/Lifelong-Hanabi
pytorch
Mentioned in GitHub
chandar-lab/CMAL_Hanabi
pytorch
Mentioned in GitHub
facebookresearch/agem
Official
tf
Mentioned in GitHub
firehose-dataset/congrad
pytorch
Mentioned in GitHub
yaqianzhang/repeatedaugmentedrehearsal
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
class-incremental-learning-on-cifar100EMR
10-stage average accuracy: 48.66

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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp