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

重放可被证明会加剧遗忘

Replay Can Provably Increase Forgetting

Columbia University(哥伦比亚大学);NVIDIA(NVIDIA公司);New York University(纽约大学);Stanford University(斯坦福大学)

Yasaman Mahdaviyeh, James Lucas, Mengye Ren, Andreas S. Tolias, Richard Zemel, Toniann Pitassi

AI总结:

本文从理论和实验上证明,样本重放在持续学习中并非总能减少遗忘,在过参数化线性回归及神经网络中,重放样本选择不当或任务子空间关系不利时反而会加剧遗忘。

摘要:

持续学习旨在使机器学习系统能够顺序地解决不断增加的任务集合。持续学习的一个关键挑战是遗忘,即随着新任务的引入,在先前学习任务上的性能会下降。样本重放是缓解遗忘的常用技术之一,经验表明,通过保留旧任务中的一些样本并将其纳入新的训练阶段,可以减少遗忘。在这项工作中,我们在一个过参数化的持续线性回归设定中对样本重放进行了理论分析,其中每个任务由一个线性子空间给出,并且如果有足够多的重放样本,人们将能够消除遗忘。我们的分析聚焦于样本重放,并强调了重放样本的作用以及任务子空间之间的关系。令人惊讶的是,我们发现,即使在无噪声设定下,遗忘相对于重放样本数量也可能是非单调的。我们给出了在最坏情况设定下重放可能有害的任务,也给出了在分布设定下随机选择样本的重放会在期望上增加遗忘的情况。我们还通过展示配备 SGD 的神经网络表现出类似行为,给出了有害重放并不局限于线性模型训练的经验证据。通过在一个常用基准上的实验,我们提供了额外证据表明,即使在看似良性的场景中,重放的性能也在很大程度上取决于重放样本的选择以及任务之间的关系。

英文摘要:

Continual learning seeks to enable machine learning systems to solve an increasing corpus of tasks sequentially. A critical challenge for continual learning is forgetting, where the performance on previously learned tasks decreases as new tasks are introduced. One of the commonly used techniques to mitigate forgetting, sample replay, has been shown empirically to reduce forgetting by retaining some examples from old tasks and including them in new training episodes. In this work, we provide a theoretical analysis of sample replay in an over-parameterized continual linear regression setting, where each task is given by a linear subspace and with enough replay samples, one would be able to eliminate forgetting. Our analysis focuses on sample replay and highlights the role of the replayed samples and the relationship between task subspaces. Surprisingly, we find that, even in a noiseless setting, forgetting can be non-monotonic with respect to the number of replay samples. We present tasks where replay can be harmful with respect to worst-case settings, and also in distributional settings where replay of randomly selected samples increases forgetting in expectation. We also give empirical evidence that harmful replay is not limited to training with linear models by showing similar behavior for a neural networks equipped with SGD. Through experiments on a commonly used benchmark, we provide additional evidence that, even in seemingly benign scenarios, performance of the replay heavily depends on the choice of replay samples and the relationship between tasks.

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

↑