PreprintCited by 129Domain Adaptation and Few-Shot LearningAdvanced Neural Network ApplicationsNeural Networks and Applications

Gradient Projection Memory for Continual Learning

Open access
Saha, Gobinda, Garg, Isha, Roy, Kaushik·2021·arXiv

Abstract

The ability to learn continually without forgetting the past tasks is a desired attribute for artificial learning systems. Existing approaches to enable such learning in artificial neural networks usually rely on network growth, importance based weight update or replay of old data from the memory. In contrast, we propose a novel approach where a neural network learns new tasks by taking gradient steps in the orthogonal direction to the gradient subspaces deemed important for the past tasks. We find the bases of these subspaces by analyzing network representations (activations) after learning each task with Singular Value Decomposition (SVD) in a single shot manner and store them in the memory as Gradient Projection Memory (GPM). With qualitative and quantitative analyses, we show that such orthogonal gradient descent induces minimum to no interference with the past tasks, thereby mitigates forgetting. We evaluate our algorithm on diverse image classification datasets with short and long sequences of tasks and report better or on-par performance compared to the state-of-the-art approaches.

Integrity signals

Registry

DataCite · 10.48550/arxiv.2103.09762

Cites

Statistics

Recomputing p-values needs the paper’s full text.

Claims the paper makes

We couldn’t retrieve this paper’s full text, so its claims and self-check weren’t run. The registry and citation checks still apply.

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • The gradient update occurs within the space spanned by the input x.

    “Consequently, the gradient update occurs within the space spanned by the input x (a detailed proof is provided in Appendix B.1).”
  • The gradient updates of a linear layer always lie within the space spanned by its input; equivalently, each column of the gradient of the weight matrix lies within the space spanned by the input vector.

    “Based on existing research(Saha et al., 2021; Zhang et al., 2021), the following conclusion can be drawn: the gradient updates of a linear layer always lie within the space spanned by its input.”
  • GPM analyzes network activations to identify the basis of task subspaces and uses gradient projections to reduce interference between new and old tasks.

    “GPM (Saha et al., 2021) analyzes network activations to identify the basis of task subspaces and uses gradient projections to reduce interference between new and old tasks.”
  • Taking gradient steps in the orthogonal direction to the gradient subspaces associated with past tasks can effectively mitigate catastrophic forgetting.

    “By taking gradient steps in the orthogonal direction to the gradient subspaces associated with past tasks, we can effectively mitigate catastrophic forgetting as it prevents interference with the past task loss functions.”

    From Orthogonal Subspace Learning for Language Model Continual Learning (2023) · cites this paper as Saha et al., 2021

The citation lists haven’t been fetched yet

References 33 · Cited by 129

Authors
Saha, G., Garg, I., & Roy, K.
Affiliations
Purdue University West Lafayette (US)
Venue
arXiv
Publisher
arXiv
Published
2021
ISSN
2331-8422 (ISSN-L)
OpenAlex
W3123155632
Cited by
129
References
33
Citations by year
2025: 33 · 2024: 37 · 2023: 32 · 2022: 17 · 2021: 10
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University)
  • International Conference on Learning Representations
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
  • Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
  • Physical Sciences › Computer Science › Artificial Intelligence › Neural Networks and Applications
Keywords
continual learning, catastrophic forgetting, singular value decomposition, instance-based learning, image classification, task sequences

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review