提出轻量级MoE结构DeaMoE,提升小批量推理效率
DeaMoE: Efficient MoE Structure for Fast Small-Batch Decoding
- 专家分组共享参数,仅保留少量私有参数
- 两阶段路由避免冗余加载,降低50.9%权重开销
- 适合实时交互场景,显著提速推理性能
混合专家(MoE)模型广泛应用于代码助手、音视频实时交互等低延迟场景。为满足极低响应延迟要求,实践中常采用小批量推理,但此时MoE推理受专家权重加载限制,成为内存瓶颈。现有方案如训练后压缩或细粒度专家设计,或降低精度,或引入额外计算与通信开销。为此,我们提出DeaMoE——一种高效解码的MoE架构:将专家按专业领域分组,同组专家共享大部分参数,每专家保留少量私有参数以体现独特性;并设计定制化两阶段路由策略,避免冗余加载。相比原始MoE,DeaMoE在A40上使7B模型每步加载权重减少50.9%,端到端 TPOT 提升1.33倍;在DeepSeek-V3上,微基准测试中分别实现A40上2.00倍、H100上1.97倍峰值加速。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) models have been widely adopted in real-time interactive applications such as coding assistants, real-time audio-video interaction systems. To meet the extremely low response latency requirements of these scenarios, practitioners commonly employ small-batch decoding, under which MoE inference becomes memory-bound and is severely bottlenecked by expert weight loading. However, this bottleneck has received limited attention, and existing solutions such as post-training weight compression or fine-grained expert design during pre-training either degrade model accuracy or introduce additional computation and communication overhead. To tackle this issue, we propose DeaMoE, a decoding-efficient MoE architecture, in which the experts are grouped into several departments, and the experts belonging to the same department share most parameters since they come from the same professional field, and additionally each expert contains a few private parameters to reflect its uniqueness. Moreover, we design customized two-stage routing strategy for DeaMoE to avoid redundant loading, under which DeaMoE greatly improves the efficiency during LLM decoding. Compared with vanilla MoE, DeaMoE reduces per-step loaded weights by up to 50.9% and achieves up to 1.33 end-to-end TPOT speedup for the pre-trained 7B model on A40, and up to 2.00x and 1.97x peak speedup for DeepSeek-V3 on A40 and H100 in microbenchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。