LGJan 23, 2024

The Joint Effect of Task Similarity and Overparameterization on Catastrophic Forgetting -- An Analytical Model

arXiv:2401.12617v226 citationsh-index: 7ICLR
Originality Incremental advance
AI Analysis

This provides an analytical model for understanding forgetting in continual learning, offering insights for researchers but is incremental as it builds on prior separate analyses.

The paper tackles how task similarity and overparameterization jointly affect catastrophic forgetting in continual learning, finding that in highly overparameterized models, intermediate similarity causes the most forgetting, while near the interpolation threshold, forgetting decreases with similarity.

In continual learning, catastrophic forgetting is affected by multiple aspects of the tasks. Previous works have analyzed separately how forgetting is affected by either task similarity or overparameterization. In contrast, our paper examines how task similarity and overparameterization jointly affect forgetting in an analyzable model. Specifically, we focus on two-task continual linear regression, where the second task is a random orthogonal transformation of an arbitrary first task (an abstraction of random permutation tasks). We derive an exact analytical expression for the expected forgetting - and uncover a nuanced pattern. In highly overparameterized models, intermediate task similarity causes the most forgetting. However, near the interpolation threshold, forgetting decreases monotonically with the expected task similarity. We validate our findings with linear regression on synthetic data, and with neural networks on established permutation task benchmarks.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes