让大模型持续学习不遗忘,专家独立判断任务归属
SEE: Continual Fine-tuning with Sequential Ensemble of Experts
- 每个专家自主决定是否处理当前任务,无需额外路由模块
- 新增任务仅训练新专家,整体系统无需重训,保持性能稳定
- 能识别异常输入并转给通用模型,适合长期演进的AI系统
大语言模型在持续微调中面临灾难性遗忘问题。基于回放的方法虽能缓解,但仍存在性能下降。尽管为每项任务训练独立专家可减轻遗忘,但有效集成仍具挑战。现有方法依赖路由器分配任务,但在持续学习场景下需重新训练才能达到最优。为此,本文提出顺序专家集成框架(SEE)。SEE无需额外路由器,使每个专家可自主判断是否应处理查询。该框架采用分布式路由机制,在持续微调中仅需训练新任务对应的专家,无需重训整个系统。实验表明,SEE在持续微调任务中优于多任务学习等基线方法,且具备出色的泛化能力——专家能有效识别分布外查询,并将其导向更通用模型进行处理。本工作展示了将路由与响应机制内置于每个专家的潜力,为分布式模型集成提供了新方向。
原文摘要 · Abstract (English)
Continual fine-tuning of large language models (LLMs) suffers from catastrophic forgetting. Rehearsal-based methods mitigate this problem by retaining a small set of old data. Nevertheless, they still suffer inevitable performance loss. Although training separate experts for each task can help prevent forgetting, effectively assembling them remains a challenge. Some approaches use routers to assign tasks to experts, but in continual learning, they often require retraining for optimal performance. To address these challenges, we introduce the Sequential Ensemble of Experts (SEE) framework. SEE removes the need for an additional router, allowing each expert to independently decide whether a query should be handled. The framework employs distributed routing, and during continual fine-tuning, SEE only requires the training of new experts for incoming tasks rather than retraining the entire system. Experiments reveal that the SEE outperforms prior approaches, including multi-task learning, in continual fine-tuning. It also demonstrates remarkable generalization ability, as the expert can effectively identify out-of-distribution queries, which can then be directed to a more generalized model for resolution. This work highlights the promising potential of integrating routing and response mechanisms within each expert, paving the way for the future of distributed model ensembling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。