通过引导向量优化专家数量与等级分配,提升LoRA-MoE性能
GuiLoMo: Allocating Expert Number and Rank for LoRA-MoE via Bilevel Optimization with GuidedSelection Vectors
- 用双层优化学习引导向量,动态分配每层专家数与秩
- 在三个模型、多个基准上均超越或媲美现有方法
- 揭示不同层和任务下最优专家配置差异,适合进阶PEFT研究者
参数高效微调(PEFT)方法,特别是低秩适应(LoRA),以较低计算成本实现了大语言模型的适配。然而其性能受限于可训练参数数量较少。近期工作将LoRA与混合专家(MoE)结合,形成LoRA-MoE以增强模型容量,但仍存在两个瓶颈:1)下游任务对专家数量分配的影响未被充分考虑;2)所有LoRA专家采用统一秩,限制了表征多样性。为此,我们提出GuiLoMo,一种基于引导选择向量(GSVs)的细粒度逐层专家数量与秩分配策略。GSVs通过前期双层优化过程学习,捕捉模型与任务特定需求,并用于指导最优专家数量与秩的分配。在三个主干模型及多种基准上的实验表明,GuiLoMo在性能上持续优于或媲美所有基线方法。进一步分析揭示了不同层与任务中专家数量与秩的变化规律,凸显自适应配置的优势。代码已公开于 https://github.com/Liar406/Gui-LoMo.git。
原文摘要 · Abstract (English)
Parameter-efficient fine-tuning (PEFT) methods, particularly Low-Rank Adaptation (LoRA), offer an efficient way to adapt large language models with reduced computational costs. However, their performance is limited by the small number of trainable parameters. Recent work combines LoRA with the Mixture-of-Experts (MoE), i.e., LoRA-MoE, to enhance capacity, but two limitations remain in hindering the full exploitation of its potential: 1) the influence of downstream tasks when assigning expert numbers, and 2) the uniform rank assignment across all LoRA experts, which restricts representational diversity. To mitigate these gaps, we propose GuiLoMo, a fine-grained layer-wise expert numbers and ranks allocation strategy with GuidedSelection Vectors (GSVs). GSVs are learned via a prior bilevel optimization process to capture both model- and task-specific needs, and are then used to allocate optimal expert numbers and ranks. Experiments on three backbone models across diverse benchmarks show that GuiLoMo consistently achieves superior or comparable performance to all baselines. Further analysis offers key insights into how expert numbers and ranks vary across layers and tasks, highlighting the benefits of adaptive expert configuration. Our code is available at https://github.com/Liar406/Gui-LoMo.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。