提出SEDEG框架,分两阶段提升编码器与解码器的泛化能力,缓解小内存下的灾难性遗忘。
SEDEG:Sequential Enhancement of Decoder and Encoder's Generality for Class Incremental Learning with Small Memory
- 分阶段优化编码器和解码器泛化能力,先特征增强后知识蒸馏压缩
- 在三个基准数据集上实现优于现有方法的分类准确率,小内存下表现更优
- 适合资源受限场景下的持续学习应用,尤其关注长期知识保持
在增量学习中,提升知识泛化能力对适应动态数据至关重要,有助于构建更具泛化的表征或更均衡的决策边界,从而防止长期知识退化并缓解灾难性遗忘。一些新兴方法采用编码器-解码器架构并取得良好效果。在此架构中,同时提升编码器与解码器的泛化能力极为关键,可兼顾旧知识保留与对新异构数据的适应性。然而,多数现有方法仅关注单一组件的改进,限制了其在缓解灾难性遗忘方面的效果,尤其在小内存场景(仅能存储有限历史样本)下表现更差。为此,本文提出SEDEG,一种面向视觉变压器(ViT)的两阶段训练框架,旨在顺序增强解码器与编码器的泛化能力。第一阶段通过特征增强训练集成编码器,学习通用表征,进而提升解码器泛化性并平衡分类器;第二阶段采用知识蒸馏(KD)策略压缩集成编码器,生成新的更泛化的编码器,结合平衡式KD与特征级KD实现高效知识迁移。大量实验在三个基准数据集上验证了SEDEG的优越性能,消融实验确认各组件有效性。代码已开源。
原文摘要 · Abstract (English)
In incremental learning, enhancing the generality of knowledge is crucial for adapting to dynamic data inputs. It can develop generalized representations or more balanced decision boundaries, preventing the degradation of long-term knowledge over time and thus mitigating catastrophic forgetting. Some emerging incremental learning methods adopt an encoder-decoder architecture and have achieved promising results. In the encoder-decoder achitecture, improving the generalization capabilities of both the encoder and decoder is critical, as it helps preserve previously learned knowledge while ensuring adaptability and robustness to new, diverse data inputs. However, many existing continual methods focus solely on enhancing one of the two components, which limits their effectiveness in mitigating catastrophic forgetting. And these methods perform even worse in small-memory scenarios, where only a limited number of historical samples can be stored. To mitigate this limitation, we introduces SEDEG, a two-stage training framework for vision transformers (ViT), focusing on sequentially improving the generality of both Decoder and Encoder. Initially, SEDEG trains an ensembled encoder through feature boosting to learn generalized representations, which subsequently enhance the decoder's generality and balance the classifier. The next stage involves using knowledge distillation (KD) strategies to compress the ensembled encoder and develop a new, more generalized encoder. This involves using a balanced KD approach and feature KD for effective knowledge transfer. Extensive experiments on three benchmark datasets show SEDEG's superior performance, and ablation studies confirm the efficacy of its components. The code is available at https://github.com/ShaolingPu/CIL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。