arXiv:2502.00089cs.CLcs.AI2025-02ICLR被引 8

通过分组指令梯度,用低秩适配器实现任务专精,提升大模型多任务表现。

Ensembles of Low-Rank Expert Adapters

  • 按指令梯度聚类,构建专家适配器组,减少训练冲突。
  • 在多个领域任务上优于全数据微调和同类集成方法。
  • 适合需要高效多任务处理的大模型部署场景。

大语言模型的训练与微调常涉及多源异构文本数据,不同数据带来的梯度方向冲突会阻碍优化并影响模型泛化能力,降低下游性能。近期研究发现,对特定任务子集进行微调可达到甚至超过全数据微调的效果。基于此,我们提出低秩专家适配器集成框架(ELREA),通过按梯度方向聚类训练指令,划分不同专业领域,减少优化过程中的冲突。每个聚类训练一个低秩适配器(LoRA),保证训练效率与模型可扩展性。推理时,根据输入数据与各聚类梯度的相似性,选择最相关的专家适配器进行组合,实现任务自适应。实验表明,该方法在多个领域任务上优于全数据微调的基线LoRA及同类集成方法,且训练与推理复杂度相当。

原文摘要 · Abstract (English)

The training and fine-tuning of large language models (LLMs) often involve diverse textual data from multiple sources, which poses challenges due to conflicting gradient directions, hindering optimization and specialization. These challenges can undermine model generalization across tasks, resulting in reduced downstream performance. Recent research suggests that fine-tuning LLMs on carefully selected, task-specific subsets of data can match or even surpass the performance of using the entire dataset. Building on these insights, we propose the Ensembles of Low-Rank Expert Adapters (ELREA) framework to improve the model's capability to handle diverse tasks. ELREA clusters the training instructions based on their gradient directions, representing different areas of expertise and thereby reducing conflicts during optimization. Expert adapters are then trained on these clusters, utilizing the low-rank adaptation (LoRA) technique to ensure training efficiency and model scalability. During inference, ELREA combines predictions from the most relevant expert adapters based on the input data's gradient similarity to the training clusters, ensuring optimal adapter selection for each task. Experiments show that our method outperforms baseline LoRA adapters trained on the full dataset and other ensemble approaches with similar training and inference complexity across a range of domain-specific tasks.

大模型微调低秩适配任务集成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。