SLIM让大模型在微调中学得多、忘得少,通过动态路由和自适应融合提升性能与记忆保持。
SLIM: Let LLM Learn More and Forget Less with Soft LoRA and Identity Mixture
- 引入软LoRA与身份混合机制,实现适配器与主干路径的动态选择。
- 在多个下游任务上表现媲美顶尖方法,遗忘率显著降低。
- 适合需要长期学习新任务且不丢失通用能力的场景。
尽管已有诸多努力,但在许多应用中仍难以平衡训练预算、下游性能与大模型的通用能力。全量微调成本高且易引发灾难性遗忘;虽参数高效微调(PEFT)可降低开销,但仍存在遗忘问题并限制下游学习。为此,我们提出基于软LoRA与身份混合的专家混合框架SLIM,支持适配器与跳连路径间的动态路由,有效抑制遗忘。采用滑动聚类加权策略增强跨域区分能力以优化路由。还将低秩适配器混合转化为模型合并形式,引入快速动态合并机制,保持基模型的通用能力。大量实验表明,SLIM在下游任务表现上接近最先进PEFT方法,同时在缓解灾难性遗忘方面达到领先水平。
原文摘要 · Abstract (English)
Although many efforts have been made, it is still a challenge to balance the training budget, downstream performance, and the general capabilities of the LLMs in many applications. Training the whole model for downstream tasks is expensive, and could easily result in catastrophic forgetting. By introducing parameter-efficient fine-tuning (PEFT), the training cost could be reduced, but it still suffers from forgetting, and limits the learning on the downstream tasks. To efficiently fine-tune the LLMs with less limitation to their downstream performance while mitigating the forgetting of general capabilities, we propose a novel mixture of expert (MoE) framework based on Soft LoRA and Identity Mixture (SLIM), that allows dynamic routing between LoRA adapters and skipping connection, enables the suppression of forgetting. We adopt weight-yielding with sliding clustering for better out-of-domain distinguish to enhance the routing. We also propose to convert the mixture of low-rank adapters to the model merging formulation and introduce fast dynamic merging of LoRA adapters to keep the general capabilities of the base model. Extensive experiments demonstrate that the proposed SLIM is comparable to the state-of-the-art PEFT approaches on the downstream tasks while achieving the leading performance in mitigating catastrophic forgetting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。