arXiv:2505.05926cs.LG2025-05ICML被引 3

用自编码器压缩记忆,实现高效增量学习

Autoencoder-Based Hybrid Replay for Class-Incremental Learning

  • 设计混合自编码器,将样本压缩到低维隐空间存储
  • 内存复杂度降至0.1t,计算开销仍为Ot,性能领先
  • 适合资源受限场景的持续学习,尤其关注记忆效率

在类别增量学习(CIL)中,随着任务数t增加,有效学习策略对缓解任务混淆和灾难性遗忘至关重要。现有样本重放策略内存与计算复杂度均为$/mathcal{O}(t)$。本文提出基于自编码器的混合重放(AHR)策略,利用新型混合自编码器(HAE)作为压缩器,在最坏情况下将内存复杂度降至$/mathcal{O}(0.1 t)$,计算复杂度保持$/mathcal{O}(t)$,同时达到当前最优性能。解码器从隐空间恢复存储的样本,而非原始格式。HAE兼具判别与生成建模能力,分别支持分类与重放。其通过带电粒子系统能量最小化方程与排斥力算法,实现新类别中心在隐空间的增量嵌入与分布。实验表明,AHR在多个基准上持续优于近期基线,且在相同内存/计算预算下运行。

原文摘要 · Abstract (English)

In class-incremental learning (CIL), effective incremental learning strategies are essential to mitigate task confusion and catastrophic forgetting, especially as the number of tasks $t$ increases. Current exemplar replay strategies impose $\mathcal{O}(t)$ memory/compute complexities. We propose an autoencoder-based hybrid replay (AHR) strategy that leverages our new hybrid autoencoder (HAE) to function as a compressor to alleviate the requirement for large memory, achieving $\mathcal{O}(0.1 t)$ at the worst case with the computing complexity of $\mathcal{O}(t)$ while accomplishing state-of-the-art performance. The decoder later recovers the exemplar data stored in the latent space, rather than in raw format. Additionally, HAE is designed for both discriminative and generative modeling, enabling classification and replay capabilities, respectively. HAE adopts the charged particle system energy minimization equations and repulsive force algorithm for the incremental embedding and distribution of new class centroids in its latent space. Our results demonstrate that AHR consistently outperforms recent baselines across multiple benchmarks while operating with the same memory/compute budgets. The source code is included in the supplementary material and will be open-sourced upon publication.

增量学习自编码器记忆压缩

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