动态分配参数预算,防止大模型持续学习时遗忘旧知识。
Adaptive Budget Allocation for Orthogonal-Subspace Adapter Tuning in LLMs Continual Learning
- 通过动态调整每个任务的参数空间大小,实现自适应预算分配。
- 在标准基准上准确率更高,参数量减少58.5%。
- 适合需要高效更新的大模型持续学习场景。
大型语言模型在持续学习中常因灾难性遗忘导致旧任务性能显著下降。尽管基于正交子空间的方法可缓解任务干扰,但通常采用固定预算分配,忽略任务与层间的复杂度差异。此外,现有自适应预算方法多为多阶段解耦设计,导致优化与预算分配不一致,限制其在持续学习中的应用。为此,我们提出OA-Adapter,一种统一动态预算适配与正交子空间学习的端到端参数高效方法。该方法引入动态瓶颈维度调节机制,在不产生偏差的前提下同时完成预算分配与任务优化。为有效保留已有知识,对当前任务参数子空间与历史任务动态分配子空间施加正交约束。实验表明,OA-Adapter在多个持续学习基准上优于现有最优方法:在标准基准上平均准确率更高,参数量减少58.5%;在含15个任务的两个更大基准上仍保持优势。
原文摘要 · Abstract (English)
Large language models (LLMs) often suffer from catastrophic forgetting in continual learning (CL) scenarios, where performance on previously learned tasks degrades severely while training on sequentially arriving tasks. Although pioneering CL approaches using orthogonal subspaces can mitigate task interference, they typically employ fixed budget allocation, neglecting the varying complexity across tasks and layers. Besides, recent budget-adaptive tuning methods for LLMs often adopt multi-stage paradigms that decouple optimization and budget allocation. Such decoupling results in potential misalignment, which hinders those approaches' practical application in CL scenarios. To address these limitations, we propose OA-Adapter, a novel parameter-efficient approach for continual learning in LLMs that unifies dynamic budget adaptation with orthogonal subspace learning in an end-to-end training stage. Specifically, OA-Adapter introduces a dynamic bottleneck dimension adaptation mechanism that simultaneously allocates an efficient parameter budget and optimizes task objectives without misalignment.To effectively preserve previously acquired knowledge while coordinating with the dynamic budget allocation, orthogonal constraints are applied specifically between the parameter subspace of the current task and the dynamically allocated parameter subspaces of historical tasks. Experimental results on continual learning benchmarks demonstrate that OA-Adapter outperforms state-of-the-art methods in both accuracy and parameter efficiency. OA-Adapter achieves higher average accuracy while using 58.5% fewer parameters on the standard CL benchmark, and maintains its advantages on two larger benchmarks comprising 15 tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。