Continual Learning in Low-rank Orthogonal Subspaces

by   Arslan Chaudhry, et al.

In continual learning (CL), a learner is faced with a sequence of tasks, arriving one after the other, and the goal is to remember all the tasks once the continual learning experience is finished. The prior art in CL uses episodic memory, parameter regularization or extensible network structures to reduce interference among tasks, but in the end, all the approaches learn different tasks in a joint vector space. We believe this invariably leads to interference among different tasks. We propose to learn tasks in different (low-rank) vector subspaces that are kept orthogonal to each other in order to minimize interference. Further, to keep the gradients of different tasks coming from these subspaces orthogonal to each other, we learn isometric mappings by posing network training as an optimization problem over the Stiefel manifold. To the best of our understanding, we report, for the first time, strong results over experience-replay baseline with and without memory on standard classification benchmarks in continual learning. The code is made publicly available.



There are no comments yet.


page 1

page 2

page 3

page 4


Energy-Based Models for Continual Learning

We motivate Energy-Based Models (EBMs) as a promising model class for co...

Batch-level Experience Replay with Review for Continual Learning

Continual learning is a branch of deep learning that seeks to strike a b...

Dark Experience for General Continual Learning: a Strong, Simple Baseline

Neural networks struggle to learn continuously, as they forget the old k...

Gradient Projection Memory for Continual Learning

The ability to learn continually without forgetting the past tasks is a ...

CLOPS: Continual Learning of Physiological Signals

Deep learning algorithms are known to experience destructive interferenc...

Compression-aware Continual Learning using Singular Value Decomposition

We propose a compression based continual task learning method that can d...

Continuous Learning of Context-dependent Processing in Neural Networks

Deep artificial neural networks (DNNs) are powerful tools for recognitio...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.