Using Hindsight to Anchor Past Knowledge in Continual Learning

Arslan Chaudhry, Albert Gordo, Puneet K. Dokania, Philip H. S. Torr, David López-Paz

Proceedings of the AAAI Conference on Artificial Intelligence · 2021 · 182 citations · 39 references

DOIFull text

Open access

Concepts

TL;DR

In continual learning, data streams shift over time and neural networks tend to forget earlier knowledge, so many methods employ experience replay with a small episodic memory to mitigate catastrophic forgetting. The study introduces an anchoring objective that uses bilevel optimization to update the current task while preserving predictions on anchor points from past tasks. Anchor points are learned through gradient‑based optimization that maximizes forgetting, approximated by fine‑tuning the model on past episodic memory. Experiments on several supervised continual learning benchmarks show that the anchoring approach improves accuracy and reduces forgetting compared to standard experience replay across various episodic memory sizes.

Abstract

In continual learning, the learner faces a stream of data whose distribution changes over time. Modern neural networks are known to suffer under this setting, as they quickly forget previously acquired knowledge. To address such catastrophic forgetting, many continual learning methods implement different types of experience replay, re-learning on past data stored in a small buffer known as episodic memory. In this work, we complement experience replay with a new objective that we call ``anchoring'', where the learner uses bilevel optimization to update its knowledge on the current task, while keeping intact the predictions on some anchor points of past tasks. These anchor points are learned using gradient-based optimization to maximize forgetting, which is approximated by fine-tuning the currently trained model on the episodic memory of past tasks. Experiments on several supervised learning benchmarks for continual learning demonstrate that our approach improves the standard experience replay in terms of both accuracy and forgetting metrics and for various sizes of episodic memory.

References

39