Formalizing Forgetting: Theory of Forgetting for Online Continual Learning
Organizations: Boston College · Sun Yat-sen University
Abstract
Online continual learning (OCL) requires a learner to acquire a sequence of tasks in a single pass over the data, and catastrophic forgetting of earlier tasks remains its central challenge. Despite the many methods proposed to mitigate forgetting, the theoretical understanding of forgetting in OCL is still limited: existing analyses mostly focus on linear, overparameterized, or neural tangent kernel regimes, and it remains unclear which factors govern forgetting and how they affect it. To fill this gap, we analyze single-pass Shuffle SGD under strongly convex and general convex objectives and derive explicit upper bounds on the expected final forgetting. Further analysis of these bounds shows how the per-task data amount, task similarity, and task ordering enter them, and that their effects can differ across objective classes. Moreover, experiments with neural networks on Split-CIFAR-100 and PMNIST show that the trends suggested by our theory also appear in practice, indicating that these insights extend beyond the convex setting.
Figures & tables
| Order | , | , | |
|---|---|---|---|
| 1 | |||
| 2 | |||
| 3 | |||
| 4 | |||
| 5 |
| PMNIST | Split CIFAR-100 | ||||
|---|---|---|---|---|---|
| ACC(%) | BWT(%) | ACC(%) | BWT(%) | ||
| 1500 | 72.52 | 2.98 | 44.43 | -18.01 | |
| 2500 | 68.71 | -7.28 | 42.51 | -24.79 | |
| 5000 | 57.89 | -15.56 | 39.46 | -25.29 | |
| PMNIST | Split CIFAR-100 | ||||
|---|---|---|---|---|---|
| ACC(%) | BWT(%) | ACC(%) | BWT(%) | ||
| 0% | 75.13 | 0.64 | 17.71 | -22.35 | |
| 40% | 71.03 | -4.51 | 16.90 | -23.68 | |
| 100% | 70.20 | -5.46 | 14.32 | -26.53 | |
| PMNIST | Split CIFAR-100 | ||||
|---|---|---|---|---|---|
| ACC(%) | BWT(%) | ACC(%) | BWT(%) | ||
| 1 | 75.42 | 5.86 | 23.59 | 2.02 | |
| 3 | 74.24 | 5.78 | 23.24 | 1.81 | |
| 5 | 73.62 | 5.28 | 21.74 | -0.49 | |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Number of tasks and the task index set . | |
| Training set of task , with samples drawn i.i.d. from the distribution . | |
| Finite-sum objective of task and its -th component, . | |
| A fixed minimizer of . | |
| Parameter obtained after training on task ; initialization . | |
| Random permutation used by Shuffle SGD on task . | |
| Model error on task , and forgetting after learning task . |
Online Continual Learning
Online continual learning adapts models to sequential data streams while preserving useful knowledge and limiting forgetting.
See all 104 papers