提出BADIT方法,让大模型任务间互不干扰,提升多任务学习效果。
Decomposing the Basic Abilities of Large Language Models: Mitigating Cross-Task Interference in Multi-Task Instruct-Tuning
- 将大模型参数分解为正交的基础能力模块,用低秩适配器表示
- 在6个大模型上测试,显著降低任务间干扰,性能超越现有方法
- 适合需要多任务协同训练的场景,如智能助手、通用AI系统
近期大语言模型的优异表现主要得益于多任务指令微调。然而,该范式存在关键问题——任务间参数共享导致梯度冲突,即跨任务干扰。已有方法通过引入任务专属参数(如任务特定神经元选择或专家混合)缓解此问题,但实证发现,由于仍存在大量跨任务共享参数,干扰依然存在。本文提出一种新方案BADIT:基于实证发现,部分参数始终共激活,并自然形成基础组块,暗示大模型编码了若干正交的基本能力,任意任务可表示为这些能力的线性组合。因此,我们设计BADIT,将模型参数分解为代表基础能力的正交高奇异值LoRA专家,并通过秩1组件的球面聚类动态强制其正交性。在包含6个大模型的SuperNI基准上进行广泛实验,结果表明BADIT能有效缓解跨任务干扰,性能优于当前最优方法。
原文摘要 · Abstract (English)
Recently, the prominent performance of large language models (LLMs) has been largely driven by multi-task instruct-tuning. Unfortunately, this training paradigm suffers from a key issue, named cross-task interference, due to conflicting gradients over shared parameters among different tasks. Some previous methods mitigate this issue by isolating task-specific parameters, e.g., task-specific neuron selection and mixture-of-experts. In this paper, we empirically reveal that the cross-task interference still exists for the existing solutions because of many parameters also shared by different tasks, and accordingly, we propose a novel solution, namely Basic Abilities Decomposition for multi-task Instruct-Tuning (BADIT). Specifically, we empirically find that certain parameters are consistently co-activated, and that co-activated parameters naturally organize into base groups. This motivates us to analogize that LLMs encode several orthogonal basic abilities, and that any task can be represented as a linear combination of these abilities. Accordingly, we propose BADIT that decomposes LLM parameters into orthogonal high-singular-value LoRA experts representing basic abilities, and dynamically enforces their orthogonality during training via spherical clustering of rank-1 components. We conduct extensive experiments on the SuperNI benchmark with 6 LLMs, and empirical results demonstrate that BADIT can outperform SOTA methods and mitigate the degree of cross-task interference.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。