arXiv:2602.08818cs.LG2026-02被引 1

让大模型专家可变秩,用更少参数实现更好性能。

FlexMoRE: A Flexible Mixture of Rank-heterogeneous Experts for Efficient Federatedly-trained Large Language Models

  • 专家可选全尺寸或低秩适配器,灵活匹配不同任务需求。
  • 最优秩使平均得分提升至47.18,参数量仅33%(10.75B)。
  • 适合追求高效推理的模型部署与联邦训练场景。

最近的混合专家架构进展表明,单个专家模型可在联邦训练中独立训练,通过共享基础模型实现协调。然而我们假设,并非所有领域都需要全尺寸专家,低秩适配器可能已足够。为此,我们提出FlexMoRE——一种灵活的异构秩混合专家架构,支持全尺寸专家或任意秩的适配器。我们系统评估了专家秩与下游任务性能之间的权衡,共测试6种秩(2⁰到2¹⁴),覆盖150种混合结构(96种双专家,54种七专家),在120项任务上验证。实验基于FlexOlmo,将预训练专家转为低秩版本。回归分析显示,推理类任务的最佳秩显著高于知识类任务。采用最优秩时,FlexMoRE在平均得分47.18上优于基线(45.46),参数量仅10.75B,不足基线33.27B的三分之一。所有代码将公开。

原文摘要 · Abstract (English)

Recent advances in mixture-of-experts architectures have shown that individual experts models can be trained federatedly, i.e., in isolation from other experts by using a common base model to facilitate coordination. However, we hypothesize that full-sized experts may not be necessary for all domains and that instead low-rank adapters may be sufficient. Here, we introduce FlexMoRE, a Flexible Mixture of Rank-heterogenous Experts, which may be either full-sized experts or adapters of a suitable rank. We systematically investigate the trade-off between expert rank and downstream task performance by evaluating $6$ experts with ranks $2^0$ to $2^{14}$ resulting in experiments covering 150 mixtures (96 with 2 experts, 54 with 7 experts) that are evaluated across $120$ tasks. For our experiments, we build on FlexOlmo and turn its pre-trained experts into low-rank versions. Our regression analysis from expert rank to downstream task performance reveals that the best-performing rank is substantially higher for reasoning-heavy benchmarks than for knowledge-heavy benchmarks. These findings on rank sensitivity come with direct implications for memory efficiency: Using optimal ranks, FlexMoRE yields improved downstream task performance (average score $47.18$) compared to the baseline FlexOlmo-style mixture of full-sized experts (average score $45.46$) at less than one third the parameters ($10.75$B for FlexMoRE vs. $33.27$B for FlexOlmo). All code will be made available.

联邦学习模型压缩混合专家低秩适配

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