arXiv:2608.14205cs.AIcs.LG2026-08

通过预测负载提前迁移专家,减少MoE模型推理延迟

FreeBalance: Pre-Routing Online Moe Load Balancing via Residual Workload Prediction

论文配图:FreeBalance: Pre-Routing Online Moe Load Balancing via Residual Workload Prediction
图 1 · 摘自论文原文
  • 用残差特征预测下一轮路由负载,提前规划专家迁移
  • 实测最大与平均负载比降32.8%,预填充延迟减少13.1%
  • 适合高并发多任务推理场景,尤其对延迟敏感应用

负载不均严重制约分布式Mixture-of-Experts(MoE)模型推理效率。最重负载的计算节点会拖慢整体执行,直接增加延迟。尽管离线专家部署可缓解长期不均,但实际多任务服务中路由行为随层和批大小动态变化,在线负载均衡不可或缺。现有方法依赖每次路由后统计结果,导致专家权重迁移必须等待路由决策,将迁移开销置于推理关键路径上。本文观察到,若能提前准确预测路由分布,在线均衡可与前置计算并行。因此提出FreeBalance,一种无损在线负载均衡框架,通过残差网络中跨层隐藏表示的相似性构建轻量负载预测器,实现专家迁移计划在路由前启动,显著重叠权重传输与计算密集型预路由阶段。同时,成本模型控制交换次数,使同步开销完全隐藏在可用窗口内。跨模型与数据集实验表明,FreeBalance将最大/平均节点负载比降低32.8%,端到端预填充延迟减少13.1%。平均每层隐藏5.1个专家的均衡开销,原占关键路径约8.5%的延迟被有效消除。

原文摘要 · Abstract (English)

Load imbalance poses a major bottleneck to the efficiency of expert parallelism in distributed inference of Mixture-of-Experts (MoE) models. The most heavily loaded rank stalls global execution due to skewed routing distributions, directly increasing latency. While offline expert placement can alleviate persistent imbalance, practical multi-task serving workloads exhibit layer- and batch-dependent routing dynamics, making online load balancing indispensable. Existing approaches rely on routing statistics collected after each MoE router, requiring expert weight load or migration to begin only after routing decisions are available, consequently placing migration overhead on the inference critical path. In this work, we observe that online balancing can instead be largely overlapped with computation before target routing (e.g., attention), if routing distributions can be predicted accurately in advance. Therefore, we propose FreeBalance, a lossless online load-balancing framework that overlaps expert migration with preceding computation stages via residual workload prediction. FreeBalance leverages cross-layer similarities in hidden representations within the residual network to build a lightweight workload predictor. This enables proactive expert migration planning before routing decisions are available, creating substantial overlap between weight transfer and computation-heavy pre-routing stages. Furthermore, a cost model constrains the number of swaps to fully hide the synchronization overhead within the available window. Experiments across models and datasets show that FreeBalance reduces the max-to-mean rank load ratio by 32.8% and end-to-end prefill latency by 13.1%. Specifically, our method hides balancing overhead of an average of 5.1 experts per layer, which would otherwise account for about 8.5% of the critical-path latency.

MoE负载均衡推理优化在线调度

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