通过旋转门机制增强低秩专家混合模型的表达能力。
RotMoLE: Enhancing Mixture of Low-Rank Experts through Rotational Gating Mechanism

- 引入旋转门替代传统标量加权,提升专家选择的灵活性。
- 在多任务和多语言场景下显著提升模型性能,尤其在专家数量有限时效果更优。
- 适合需要高效适配复杂知识的垂直领域应用,如低资源场景下的模型微调。
尽管大语言模型(LLMs)常通过微调以应对特定领域任务,但在包含多样专业知识的复杂场景中仍面临挑战。混合专家(MoE)架构已成为训练LLMs的关键范式,近期研究将MoE与参数高效微调(PEFT)结合,提出低秩专家混合(MoE-LoRA),以增强低秩适配器对复杂知识的学习能力。然而,传统MoE中的门控机制仅对选中专家进行标量重加权,限制了其表示与泛化潜力。受MoE-LoRA低秩结构启发,我们提出RotMoLE,一种面向低秩专家的专用MoE框架,引入额外的旋转门机制。相比简单缩放,每个选中专家均通过旋转操作进行调整,实现更优的专家利用与专业化,尤其在专家候选数有限时表现更佳。在复杂多任务与多语言训练场景中的实证结果验证了其有效性。
原文摘要 · Abstract (English)
While Large Language Models (LLMs) are commonly fine-tuned to handle domain-specific tasks before being applied to vertical applications, adapting them to complex scenarios with diverse specialized knowledge remains challenging. Meanwhile, Mixture-of-Experts (MoE) architecture has risen as a crucial paradigm for training LLMs, and some recent works have also incorporated MoE into Parameter-Efficient Fine-Tuning (PEFT) to propose the Mixture of Low-rank Experts (MoE-LoRA), to enhance the power of low-rank adapters for learning complicated knowledge. However, conventional gating mechanisms in MoE typically apply only a scalar reweighing to selected experts, thereby limiting their underlying capacity of representation and generalization. Motivated and enabled by the low-rank structures in MoE-LoRA, we propose RotMoLE, a specialized MoE framework for low-rank experts featuring an additional rotation gate. Beyond simple scaling, RotMoLE implements a rotation mechanism for each selected expert, enabling superior expert exploitation and specialization for learning diverse data, especially when expert candidates are limited. Empirical results on complex multi-task and multilingual training scenarios validate our effectiveness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。