arXiv:2503.23362cs.CLcs.AI2025-03被引 4

通过多路由协同提升模型微调效果,性能平均提升1%。

Mixture of Routers

  • 用多个子路由联合选专家,主路由学习权重分配。
  • 在多数任务上相比基线平均提升1%的性能。
  • 参数高效、即插即用,适合各类下游应用。

监督微调(SFT)是使大语言模型与人类指令对齐并适应下游任务的重要里程碑。低秩适配(LoRA)因其参数高效而广受关注,但对大模型性能的提升有限。近期研究发现,将LoRA与专家混合(MoE)结合可显著提升微调性能。MoE通过动态选择最合适的专家来应对数据集的多样性和复杂性,从而提高任务准确率和效率。然而,现有研究揭示了MoE路由机制中存在的错误分配和专家负载不均问题。受冗余与容错理论启发,我们创新性地将专家混合思想融入路由机制,提出一种高效的微调方法——混合路由(MoR)。MoR采用多个子路由进行联合选择,并通过可学习的主路由确定子路由的权重。实验结果表明,MoR在多数任务上优于基线模型,平均性能提升1%。该方法可作为即插即用、参数高效的微调方案,适用于广泛的应用场景。代码已公开:https://anonymous.4open.science/r/MoR-DFC6。

原文摘要 · Abstract (English)

Supervised fine-tuning (SFT) is a milestone in aligning large language models with human instructions and adapting them to downstream tasks. In particular, Low-Rank Adaptation (LoRA) has gained widespread attention due to its parameter efficiency. However, its impact on improving the performance of large models remains limited. Recent studies suggest that combining LoRA with Mixture-of-Experts (MoE) can significantly enhance fine-tuning performance. MoE adapts to the diversity and complexity of datasets by dynamically selecting the most suitable experts, thereby improving task accuracy and efficiency. Despite impressive results, recent studies reveal issues in the MoE routing mechanism, such as incorrect assignments and imbalanced expert allocation. Inspired by the principles of Redundancy and Fault Tolerance Theory. We innovatively integrate the concept of Mixture of Experts into the routing mechanism and propose an efficient fine-tuning method called Mixture of Routers (MoR). It employs multiple sub-routers for joint selection and uses a learnable main router to determine the weights of the sub-routers. The results show that MoR outperforms baseline models on most tasks, achieving an average performance improvement of 1%. MoR can serve as a plug-and-play, parameter-efficient fine-tuning method suitable for a wide range of applications. Our code is available here: https://anonymous.4open.science/r/MoR-DFC6.

微调MoELoRA路由

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