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

用于类增量学习的压缩原型重放

Condensed Prototype Replay for Class Incremental Learning

College of William and Mary(威廉与玛丽学院);University of Maryland, College Park(马里兰大学帕克分校)

Jiangtao Kong, Zhenyu Zong, Tianyi Zhou, Huajie Shao

AI总结:

本文提出 YONO 方法,通过注意力均值漂移和样本向原型聚拢学习每类一个压缩原型,最大化类边界以减少遗忘,并扩展 YONO+ 生成合成重放数据,在增量学习中超越现有方法。

摘要:

增量学习(IL)在学习新任务时会遭受对旧任务的灾难性遗忘。可以通过重放存储在内存中的先前任务数据来缓解这一问题,但这种方式通常受到内存大小限制并存在隐私泄露风险。近期研究仅存储类中心作为原型,并用高斯噪声增强这些原型以生成用于重放的合成数据。然而,这些方法无法有效避免原型边界附近的类间干扰,从而导致遗忘。此外,注入的噪声扭曲了真实数据与原型之间的丰富结构,因此甚至对增量学习有害。在本文中,我们提出 YONO,即每个类只需要重放一个压缩原型,该方法首次能够超越内存开销高昂的样本重放方法。为此,我们开发了一种新颖的原型学习方法,该方法(1)通过注意力均值漂移算法在高密度区域中搜索更具代表性的原型,并(2)将每个类中的样本移动到其原型附近,以形成远离其他类的紧凑簇。由此,类边界被最大化,从而有效减少导致未来遗忘的干扰。此外,我们将 YONO 扩展为 YONO+,通过在表示空间中每个原型的邻域内随机采样来生成合成重放数据。我们证明合成数据可以进一步提升 YONO 的性能。在增量学习基准上的大量实验表明,YONO/YONO+ 在准确率和遗忘率方面均优于现有的增量学习方法。

英文摘要:

Incremental learning (IL) suffers from catastrophic forgetting of old tasks when learning new tasks. This can be addressed by replaying previous tasks' data stored in a memory, which however is usually prone to size limits and privacy leakage. Recent studies store only class centroids as prototypes and augment them with Gaussian noises to create synthetic data for replay. However, they cannot effectively avoid class interference near their margins that leads to forgetting. Moreover, the injected noises distort the rich structure between real data and prototypes, hence even detrimental to IL. In this paper, we propose YONO that You Only Need to replay One condensed prototype per class, which for the first time can even outperform memory-costly exemplar-replay methods. To this end, we develop a novel prototype learning method that (1) searches for more representative prototypes in high-density regions by an attentional mean-shift algorithm and (2) moves samples in each class to their prototype to form a compact cluster distant from other classes. Thereby, the class margins are maximized, which effectively reduces interference causing future forgetting. In addition, we extend YONO to YONO+, which creates synthetic replay data by random sampling in the neighborhood of each prototype in the representation space. We show that the synthetic data can further improve YONO. Extensive experiments on IL benchmarks demonstrate the advantages of YONO/YONO+ over existing IL methods in terms of both accuracy and forgetting.

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

↑