arXiv:2509.02753cs.LG2025-09被引 6

让专家模型每层自适应激活专家数,提升推理效率。

LExI: Layer-Adaptive Active Experts for Efficient MoE Model Inference

  • 根据模型权重自动计算每层重要性,动态分配活跃专家数。
  • 在H100上实现10%更高准确率的同时保持相同吞吐量。
  • 无需数据即可优化,适合部署时追求高效推理的场景。

混合专家(MoE)模型通过仅激活部分专家实现高效扩展,为密集架构提供计算稀疏替代方案。现有训练后优化方法如跨专家和内专家剪枝虽降低内存占用,但对GPU推理性能提升有限。此外,传统MoE架构在各层统一激活固定数量专家,导致冗余计算且性能不佳。本文首次证明,在vLLM等优化框架下,传统剪枝仅改善内存开销,对推理性能帮助甚微。为此,我们提出LExI,一种无需数据的优化技术,可确定预训练MoE模型中每层最优活跃专家数。LExI仅利用模型权重估计各层相对重要性,并据此自适应分配每层活跃专家数量。在主流语言与视觉MoE基准测试中,LExI显著优于传统剪枝方法,在几乎无精度损失下大幅提升推理效率。例如,使用LExI时,Qwen1.5-MoE在Nvidia H100 GPU上达到与传统剪枝相同吞吐量,但准确率高出10%。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) models scale efficiently by activating only a subset of experts per token, offering a computationally sparse alternative to dense architectures. While prior post-training optimizations, such as inter- and intra-expert pruning, reduce memory usage they provide limited gains in inference-time compute efficiency. Moreover, existing MoE architectures typically activate a fixed number of experts uniformly across all layers, resulting in redundant computation and suboptimal performance. In this work, we first demonstrate that MoE pruning strategies improve only the memory footprint but do not significantly improve inference performance on GPU using optimized frameworks such as vLLM. To address this, we introduce LExI, a data-free optimization technique that determines the optimal number of active experts per layer in a pretrained MoE model. LExI leverages only the model weights to estimate the relative importance of each layer and adaptively assigns the number of active experts accordingly per layer. Experiments on state-of-the-art language and vision MoE benchmarks demonstrate that LExI significantly outperforms traditional MoE pruning approaches in terms of inference efficiency with negligible accuracy loss. For example, using LExI, Qwen1.5-MoE achieves the same throughput on Nvidia H100 GPU with 10% better accuracy than traditional expert pruning.

MoE推理优化专家系统

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