用任务专属专家解决预训练模型增量学习中的参数冲突问题
MoTE: Mixture of Task-specific Experts for Pre-Trained ModelBased Class-incremental Learning
- 引入任务感知专家筛选与可靠专家联合推理机制
- 无需示例集即可实现多任务持续学习,性能优于现有方法
- 适合需要长期更新的部署场景,尤其对资源受限环境友好
类别增量学习(CIL)要求模型在持续接收流式数据时不断学习新知识,同时保留旧知识。基于预训练模型(PTM)的CIL近期取得显著进展,但提示方法存在提示覆盖问题,适配器方法则面临任务间维度不匹配难题。尽管混合专家(MoE)结构可缓解维度不一致,但其专家和路由参数在动态环境中仍易被覆盖,难以直接应用于CIL。为此,本文提出任务专属专家混合框架(MoTE),有效缓解因任务间输出维度不一致导致的失准问题。受MoE中加权特征融合与稀疏激活机制启发,引入任务感知专家筛选与可靠专家联合推理策略,在推理阶段模拟路由层行为,避免灾难性遗忘。大量实验表明,该方法无需示例集即可实现优越性能,且专家数量随任务数线性增长。进一步探索适配器扩展与模型性能之间的权衡,提出适配器受限版MoTE。代码已开源。
原文摘要 · Abstract (English)
Class-incremental learning (CIL) requires deep learning models to continuously acquire new knowledge from streaming data while preserving previously learned information. Recently, CIL based on pre-trained models (PTMs) has achieved remarkable success. However, prompt-based approaches suffer from prompt overwriting, while adapter-based methods face challenges such as dimensional misalignment between tasks. While the idea of expert fusion in Mixture of Experts (MoE) can help address dimensional inconsistency, both expert and routing parameters are prone to being overwritten in dynamic environments, making MoE challenging to apply directly in CIL. To tackle these issues, we propose a mixture of task-specific experts (MoTE) framework that effectively mitigates the miscalibration caused by inconsistent output dimensions across tasks. Inspired by the weighted feature fusion and sparse activation mechanisms in MoE, we introduce task-aware expert filtering and reliable expert joint inference during the inference phase, mimicking the behavior of routing layers without inducing catastrophic forgetting. Extensive experiments demonstrate the superiority of our method without requiring an exemplar set. Furthermore, the number of tasks in MoTE scales linearly with the number of adapters. Building on this, we further explore the trade-off between adapter expansion and model performance and propose the Adapter-Limited MoTE. The code is available at https://github.com/Franklilinjie/MoTE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。