Saved in:
Bibliographic Details
Main Authors: Janetzky, Pascal, Schlagenhauf, Tobias, Feuerriegel, Stefan
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2411.06916
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915178167140352
author Janetzky, Pascal
Schlagenhauf, Tobias
Feuerriegel, Stefan
author_facet Janetzky, Pascal
Schlagenhauf, Tobias
Feuerriegel, Stefan
contents A common challenge in continual learning (CL) is catastrophic forgetting, where the performance on old tasks drops after new, additional tasks are learned. In this paper, we propose a novel framework called ReCL to slow down forgetting in CL. Our framework exploits an implicit bias of gradient-based neural networks due to which these converge to margin maximization points. Such convergence points allow us to reconstruct old data from previous tasks, which we then combine with the current training data. Our framework is flexible and can be applied on top of existing, state-of-the-art CL methods. We further demonstrate the performance gain from our framework across a large series of experiments, including two challenging CL scenarios (class incremental and domain incremental learning), different datasets (MNIST, CIFAR10, TinyImagenet), and different network architectures. Across all experiments, we find large performance gains through ReCL. To the best of our knowledge, our framework is the first to address catastrophic forgetting by leveraging models in CL as their own memory buffers.
format Preprint
id arxiv_https___arxiv_org_abs_2411_06916
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Slowing Down Forgetting in Continual Learning
Janetzky, Pascal
Schlagenhauf, Tobias
Feuerriegel, Stefan
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
A common challenge in continual learning (CL) is catastrophic forgetting, where the performance on old tasks drops after new, additional tasks are learned. In this paper, we propose a novel framework called ReCL to slow down forgetting in CL. Our framework exploits an implicit bias of gradient-based neural networks due to which these converge to margin maximization points. Such convergence points allow us to reconstruct old data from previous tasks, which we then combine with the current training data. Our framework is flexible and can be applied on top of existing, state-of-the-art CL methods. We further demonstrate the performance gain from our framework across a large series of experiments, including two challenging CL scenarios (class incremental and domain incremental learning), different datasets (MNIST, CIFAR10, TinyImagenet), and different network architectures. Across all experiments, we find large performance gains through ReCL. To the best of our knowledge, our framework is the first to address catastrophic forgetting by leveraging models in CL as their own memory buffers.
title Slowing Down Forgetting in Continual Learning
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.06916