arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2405.17583cs.LG

理解线性回归中持续学习的遗忘现象

Understanding Forgetting in Continual Learning with Linear Regression

State University of New York at Buffalo(纽约州立大学布法罗分校);King Abdullah University of Science and Technology(阿卜杜拉国王科技大学)

Meng Ding, Kaiyi Ji, Di Wang, Jinhui Xu

AI总结:

本文通过SGD对线性回归模型进行理论分析,揭示任务序列与步长对遗忘的影响,并在线性回归和DNN模拟中验证了结论。

摘要:

持续学习专注于顺序学习多个任务,近年来受到了广泛关注。尽管过去取得了巨大进展,但理论理解,尤其是导致灾难性遗忘的因素,仍相对未被探索。在本文中,我们通过随机梯度下降(SGD)对线性回归模型中的遗忘进行了通用理论分析,该分析适用于欠参数化和过参数化两种情形。我们的理论框架揭示了任务序列与算法参数之间复杂关系的一些有趣见解,而先前研究因其限制性假设未能完全捕捉这一方面。具体而言,我们证明,在数据量足够大的情况下,任务在序列中的排列——即其总体数据协方差矩阵具有较大特征值的任务被较晚训练——往往会导致遗忘增加。此外,我们的发现强调,适当选择步长将有助于在欠参数化和过参数化设置中减轻遗忘。为验证我们的理论分析,我们在线性回归模型和深度神经网络(DNNs)上进行了模拟实验。这些模拟的结果证实了我们的理论发现。

英文摘要:

Continual learning, focused on sequentially learning multiple tasks, has gained significant attention recently. Despite the tremendous progress made in the past, the theoretical understanding, especially factors contributing to catastrophic forgetting, remains relatively unexplored. In this paper, we provide a general theoretical analysis of forgetting in the linear regression model via Stochastic Gradient Descent (SGD) applicable to both underparameterized and overparameterized regimes. Our theoretical framework reveals some interesting insights into the intricate relationship between task sequence and algorithmic parameters, an aspect not fully captured in previous studies due to their restrictive assumptions. Specifically, we demonstrate that, given a sufficiently large data size, the arrangement of tasks in a sequence, where tasks with larger eigenvalues in their population data covariance matrices are trained later, tends to result in increased forgetting. Additionally, our findings highlight that an appropriate choice of step size will help mitigate forgetting in both underparameterized and overparameterized settings. To validate our theoretical analysis, we conducted simulation experiments on both linear regression models and Deep Neural Networks (DNNs). Results from these simulations substantiate our theoretical findings.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑