arXiv:2604.08133cs.LGcs.AI2026-04ACL被引 2

通过预算分配优化,让专家模型推理更快更准。

Alloc-MoE: Budget-Aware Expert Activation Allocation for Efficient Mixture-of-Experts Inference

论文配图:Alloc-MoE: Budget-Aware Expert Activation Allocation for Efficient Mixture-of-Experts Inference
图 1 · 摘自论文原文
  • 按层和逐令牌动态分配激活名额,控制计算开销。
  • 在激活数减半时仍保持性能,预填充提速1.15倍,解码提速1.34倍。
  • 适合资源受限场景下的大模型高效部署,尤其对MoE结构有效。

混合专家(MoE)因稀疏激活机制成为扩展大语言模型的主流架构,但在推理阶段大量专家激活造成显著延迟瓶颈,尤其在资源受限场景下。现有减少激活的方法可能导致性能严重下降。本文引入“激活预算”作为激活数量约束,提出统一框架 Alloc-MoE,协同优化层间与令牌级的预算分配,以最小化性能损失。在层级别,提出 Alloc-L,结合敏感性分析与动态规划,确定各层最优激活分配;在令牌级别,提出 Alloc-T,基于路由得分动态重分配激活,不增加延迟。在多个 MoE 模型上的实验表明,Alloc-MoE 在受限激活预算下仍能保持模型性能。特别地,在激活预算减半的情况下,DeepSeek-V2-Lite 实现了 1.15 倍的预填充加速和 1.34 倍的解码加速。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) has become a dominant architecture for scaling large language models due to their sparse activation mechanism. However, the substantial number of expert activations creates a critical latency bottleneck during inference, especially in resource-constrained deployment scenarios. Existing approaches that reduce expert activations potentially lead to severe model performance degradation. In this work, we introduce the concept of \emph{activation budget} as a constraint on the number of expert activations and propose Alloc-MoE, a unified framework that optimizes budget allocation coordinately at both the layer and token levels to minimize performance degradation. At the layer level, we introduce Alloc-L, which leverages sensitivity profiling and dynamic programming to determine the optimal allocation of expert activations across layers. At the token level, we propose Alloc-T, which dynamically redistributes activations based on routing scores, optimizing budget allocation without increasing latency. Extensive experiments across multiple MoE models demonstrate that Alloc-MoE maintains model performance under a constrained activation budget. Especially, Alloc-MoE achieves $1.15\times$ prefill and $1.34\times$ decode speedups on DeepSeek-V2-Lite at half of the original budget.

MoE推理优化预算分配大模型

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