解决增量学习中旧知识遗忘和适配器分配不合理的难题
Non-Forgetting Knowledge Allocation with Bi-level Competition for Class-Incremental Learning

- 用递归最小二乘法构建抗遗忘的分配器,避免训练数据丢失
- 引入双层竞争机制,动态调整适配器参与度,提升新旧任务平衡
- 适合需要持续学习且防止知识衰退的模型部署场景
基于预训练模型的类别增量学习(CIL)旨在不遗忘旧知识的前提下,逐步适应新类别。现有适配器方法在推理时对每个适配器采用统一分配策略,忽视任务差异性,导致适配器利用效率低下。同时,在增量学习过程中,分配器易因任务演进而遗忘历史信息。为此,本文提出非遗忘分配与双层竞争机制(NoFA-BC)。通过将分配器训练转化为递归最小二乘问题,构建非遗忘分配器(NFA),其效果等价于使用全部数据训练的结果。在此基础上,设计双层竞争机制(BLC),包含任务内“赢家通吃”(WTA)和任务间“最后者淘汰”(LOF)策略:WTA提取单任务内最具贡献的logit,LOF抑制无关适配器。该机制使每个输入的适配器参与率可动态调整。此外,引入稳定性增强(SE)过程进一步提升旧任务性能。
原文摘要 · Abstract (English)
Class-Incremental Learning (CIL) with pre-trained models (PTMs) aims to sequentially adapt PTMs to new categories without forgetting old knowledge. Built upon PTMs, existing adapter-based methods mainly train models via distinct task-specific adapters, and present a uniform knowledge allocation for each adapter during inference. However, this allocation mechanism ignores the nature of task discrepancy and leads to suboptimal utilization of adapters. Also, under CIL constraint, an allocator is prone to forgetting when tasks evolve. To address these issues, we propose a Non-Forgetting Allocation with Bi-Level Competition (NoFA-BC). NoFA-BC constructs a non-forgetting allocator (NFA) by transforming the allocator training into a recursive least-squares problem and achieves an allocator equivalent to that trained with all data. Based on the NFA, a Bi-Level Competition (BLC) including an intra-task level Winner-Takes-All (WTA) mechanism and inter-task Last-Ones-Fall (LOF) elimination is proposed to provide better allocation of adapter knowledge. WTA extracts the most significant logit within a task to represent the adapter's contribution and LOF suppresses the irrelevant adapters. With BLC, participation ratio of each adapter can be tailored for each input. Moreover, a Stability Enhancement (SE) process is incorporated to further improve the performance of old tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。