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

细粒度梯度限制:一种缓解灾难性遗忘的简单方法

Fine-Grained Gradient Restriction: A Simple Approach for Mitigating Catastrophic Forgetting

The University of Texas at Austin(德克萨斯大学奥斯汀分校);Sony AI(索尼人工智能)

Bo Liu, Mao Ye, Peter Stone, Qiang Liu

AI总结:

本文分析GEM中记忆强度超参数的作用机制,发现其通过改善泛化能力实现新旧任务权衡,并提出两种更灵活的细粒度梯度限制方法,在帕累托前沿上一致优于记忆强度,同时给出计算高效的近似求解方案。

摘要:

持续学习中的一个根本性挑战是在学习新任务与记住先前获得的知识之间取得平衡。梯度情景记忆(Gradient Episodic Memory, GEM)通过利用过去训练样本的一个子集来限制模型参数的更新方向,从而实现这种平衡。在本工作中,我们首先分析GEM中一个常被忽视的超参数——记忆强度,该参数通过进一步约束更新方向来提升经验性能。我们表明,记忆强度之所以有效,主要因为它改善了GEM的泛化能力,从而带来更有利的权衡。基于这一发现,我们提出了两种更灵活地约束更新方向的方法。与使用记忆强度相比,我们的方法能够在记住旧知识和学习新知识方面取得一致更优的帕累托前沿。我们进一步提出了一种计算高效的方法,用于近似求解具有更多约束的优化问题。

英文摘要:

A fundamental challenge in continual learning is to balance the trade-off between learning new tasks and remembering the previously acquired knowledge. Gradient Episodic Memory (GEM) achieves this balance by utilizing a subset of past training samples to restrict the update direction of the model parameters. In this work, we start by analyzing an often overlooked hyper-parameter in GEM, the memory strength, which boosts the empirical performance by further constraining the update direction. We show that memory strength is effective mainly because it improves GEM's generalization ability and therefore leads to a more favorable trade-off. By this finding, we propose two approaches that more flexibly constrain the update direction. Our methods are able to achieve uniformly better Pareto Frontiers of remembering old and learning new knowledge than using memory strength. We further propose a computationally efficient method to approximately solve the optimization problem with more constraints.

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

↑