259
v1v2v3 (latest)

From Continual Learning to SGD and Back: Better Rates for Continual Linear Models

Main:13 Pages
2 Figures
Bibliography:8 Pages
3 Tables
Appendix:29 Pages
Abstract

We study the common continual learning setup where an overparameterized model is sequentially fitted to a set of jointly realizable tasks. We analyze forgetting, defined as the loss on previously seen tasks, after kk iterations. For continual linear models, we prove that fitting a task is equivalent to a single stochastic gradient descent (SGD) step on a modified objective. We develop novel last-iterate SGD upper bounds in the realizable least squares setup and leverage them to derive new results for continual learning. Focusing on random orderings over TT tasks, we establish universal forgetting rates, whereas existing rates depend on problem dimensionality or complexity and become prohibitive in highly overparameterized regimes. In continual regression with replacement, we improve the best existing rate from O((drˉ)/k)O((d-\bar{r})/k) to O(min(1/k4,(drˉ)/k,Trˉ/k))O(\min(1/\sqrt[4]{k}, \sqrt{(d-\bar{r})}/k, \sqrt{T\bar{r}}/k)), where dd is the dimensionality and rˉ\bar{r} the average task rank. Furthermore, we establish the first rate for random task orderings without replacement. The resulting rate O(min(1/T4,(drˉ)/T))O(\min(1/\sqrt[4]{T},\, (d-\bar{r})/T)) shows that randomization alone, without task repetition, prevents catastrophic forgetting in sufficiently long task sequences. Finally, we prove a matching O(1/k4)O(1/\sqrt[4]{k}) forgetting rate for continual linear classification on separable data. Our universal rates extend to broader methods, such as block Kaczmarz and POCS, illuminating their loss convergence under i.i.d. and single-pass orderings.

View on arXiv
Comments on this paper