arXiv:2504.03871cs.DCcs.LG2025-04被引 12

让新旧显卡协同训练专家模型,效率提升近两倍。

HeterMoE: Efficient Training of Mixture-of-Experts Models on Heterogeneous GPUs

  • 分离注意力与专家计算,旧卡专跑专家模块。
  • 采用交错并行和非对称分配,减少显卡空闲时间。
  • 旧卡替换一半仍保持95%吞吐,适合混合硬件集群。

混合专家(MoE)架构已成为扩展大语言模型的重要方法。为降低成本,已有方案利用包含新旧代显卡的集群进行训练。但现有方法未考虑不同模型组件(注意力与专家)在异构硬件上的性能差异,未能充分发挥各显卡的算力。本文提出HeterMoE系统,高效训练异构显卡上的MoE模型。核心思想是:新显卡在注意力计算上显著优于旧卡,而旧卡在专家计算上仍具效率。HeterMoE将注意力与专家计算拆分,仅将专家模块分配给旧卡。通过提出的斑马并行机制,实现跨显卡计算重叠,并结合非对称专家分配策略,实现细粒度负载均衡,最大限度减少显卡闲置。评估显示,相较于现有MoE训练系统,HeterMoE最高提速2.3倍;相比最优平衡的异构感知方案,提速1.4倍。当同质A40集群中一半显卡被V100替换时,仍可维持平均95%的训练吞吐率。

原文摘要 · Abstract (English)

The Mixture-of-Experts (MoE) architecture has become increasingly popular as a method to scale up large language models (LLMs). To save costs, heterogeneity-aware training solutions have been proposed to utilize GPU clusters made up of both newer and older-generation GPUs. However, existing solutions are agnostic to the performance characteristics of different MoE model components (i.e., attention and expert) and do not fully utilize each GPU's compute capability. In this paper, we introduce HeterMoE, a system to efficiently train MoE models on heterogeneous GPUs. Our key insight is that newer GPUs significantly outperform older generations on attention due to architectural advancements, while older GPUs are still relatively efficient for experts. HeterMoE disaggregates attention and expert computation, where older GPUs are only assigned with expert modules. Through the proposed zebra parallelism, HeterMoE overlaps the computation on different GPUs, in addition to employing an asymmetric expert assignment strategy for fine-grained load balancing to minimize GPU idle time. Our evaluation shows that HeterMoE achieves up to 2.3x speed-up compared to existing MoE training systems, and 1.4x compared to an optimally balanced heterogeneity-aware solution. HeterMoE efficiently utilizes older GPUs by maintaining 95% training throughput on average, even with half of the GPUs in a homogeneous A40 cluster replaced with V100.

MoE异构计算训练加速显卡优化

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