arXiv:2410.15372cs.LG2024-10被引 4

混合真实与生成数据缓解增量学习遗忘问题

Hybrid Memory Replay: Blending Real and Distilled Data for Class Incremental Learning

  • 用历史模型快照生成合成数据,结合真实样本优化记忆缓冲
  • 在小缓冲区下显著降低灾难性遗忘,准确率提升10%以上
  • 兼容主流重放类增量学习模型,适合资源受限场景

增量学习(IL)旨在学习新任务知识的同时保留旧任务知识。基于重放的方法通过缓冲区存储过往任务的实例进行回放,但缓冲区大小有限,难以存储足够多的真实实例。相比之下,数据蒸馏(DD)可将大量真实数据压缩为更紧凑的合成实例,减少缓冲区占用。然而,随着合成实例数量增加,其性能优势迅速消失。为此,本文提出混合记忆机制,同时利用真实与合成数据。创新性地,从历史模型快照滑动窗口中蒸馏合成数据;并基于合成数据优化真实实例选择,以补充蒸馏目标。该混合记忆结合了两类数据优势,在缓冲区受限时有效缓解类别增量学习(CIL)中的灾难性遗忘。实验表明,该方法在多个基准上显著优于现有基线。

原文摘要 · Abstract (English)

Incremental learning (IL) aims to acquire new knowledge from current tasks while retaining knowledge learned from previous tasks. Replay-based IL methods store a set of exemplars from previous tasks in a buffer and replay them when learning new tasks. However, there is usually a size-limited buffer that cannot store adequate real exemplars to retain the knowledge of previous tasks. In contrast, data distillation (DD) can reduce the exemplar buffer's size, by condensing a large real dataset into a much smaller set of more information-compact synthetic exemplars. Nevertheless, DD's performance gain on IL quickly vanishes as the number of synthetic exemplars grows. To overcome the weaknesses of real-data and synthetic-data buffers, we instead optimize a hybrid memory including both types of data. Specifically, we propose an innovative modification to DD that distills synthetic data from a sliding window of checkpoints in history (rather than checkpoints on multiple training trajectories). Conditioned on the synthetic data, we then optimize the selection of real exemplars to provide complementary improvement to the DD objective. The optimized hybrid memory combines the strengths of synthetic and real exemplars, effectively mitigating catastrophic forgetting in Class IL (CIL) when the buffer size for exemplars is limited. Notably, our method can be seamlessly integrated into most existing replay-based CIL models. Extensive experiments across multiple benchmarks demonstrate that our method significantly outperforms existing replay-based baselines.

增量学习数据蒸馏记忆回放分类任务

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。