arXiv:2510.17898cs.LGcs.AI2025-10被引 2

用低秩适配器构建可端到端训练的轻量级专家混合模型。

L-MoE: End-to-End Training of a Lightweight Mixture of Low-Rank Adaptation Experts

  • 将专家设为低秩适配器,通过门控网络动态组合参数。
  • 实现端到端训练,支持高效微调与动态技能组合。
  • 适合追求高效微调与模型轻量化的研究者使用。

Mixture of Experts(MoE)架构通过为每个输入激活稀疏权重子集,使大型语言模型(LLMs)规模扩展至万亿参数,同时保持推理时计算成本恒定。与此同时,低秩适配(LoRA)已成为高效微调专用任务的主流技术。本文提出一种新型端到端可训练框架L-MoE:轻量级LoRA专家混合模型。L-MoE将专家重构为任务专用的低秩适配器,而非密集前馈网络。一个轻量门控网络与专家联合训练,通过计算各适配器参数的加权平均,为每个输入词元动态组合适配器。该组合过程完全可微,允许自回归语言建模目标的梯度反向传播至整个架构,同步优化专家适配器与路由策略。该方法构建了高度参数高效的MoE模型,具有模块化设计、动态技能组合能力,并支持端到端训练。论文给出L-MoE的形式化数学框架,详述可微路由机制与联合优化目标,为构建更高效、可扩展、专业化语言模型提供新路径。

原文摘要 · Abstract (English)

The Mixture of Experts (MoE) architecture enables the scaling of Large Language Models (LLMs) to trillions of parameters by activating a sparse subset of weights for each input, maintaining constant computational cost during inference. Concurrently, Low-Rank Adaptation (LoRA) has emerged as a dominant technique for parameter-efficiently fine-tuning LLMs on specialized tasks. In this work, we unify these two paradigms into a novel, end-to-end trainable framework named L-MoE: a Lightweight Mixture of LoRA Experts. L-MoE redefines MoE experts not as dense feed-forward networks, but as a collection of task-specialized, low-rank adapters. A lightweight gating network, trained jointly with the experts, learns to dynamically compose these LoRA adapters by computing a weighted average of their parameters for each input token. This composition is fully differentiable, allowing gradients from a standard auto-regressive language modeling objective to flow back through the entire architecture, simultaneously refining both the expert adapters and the routing strategy. This approach creates a highly parameter-efficient MoE model that is modular by design, allows for dynamic skill composition, and is trainable from end-to-end. We present the formal mathematical framework for L-MoE, detailing the differentiable routing mechanism and the joint optimization objective, thereby providing a new path toward building more efficient, scalable, and specialized language models.

MoELoRA参数高效端到端

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