通过解耦专家参数与优化器状态,实现高效稀疏激活模型训练。
SYMI: Efficient Mixture-of-Experts Training via Model and Optimizer State Decoupling
- 将专家参数与优化器状态分离,静态分配优化器,动态调整参数位置。
- 每轮迭代可精准调配资源,减少负载不均导致的性能损失。
- 适合大规模MoE模型训练,显著提升收敛速度,降低资源迁移开销。
混合专家(MoE)模型已成为在不线性增加计算量的前提下继续扩展模型规模的常用方案。在训练过程中,每个输入标记动态路由到每一层Transformer中的部分专家——稀疏激活的前馈网络。专家间标记分配分布随训练过程快速变化,导致负载严重不均。现有系统要么丢弃热门专家的标记以避免过载,影响收敛;要么频繁根据热度重新分配资源,带来高状态迁移开销。为突破这一性能-精度权衡,本文提出SYMI,一种自适应的MoE训练系统。其核心思想是将专家参数放置与其大尺寸优化器状态解耦。SYMI将每个专家的优化器静态地跨所有训练节点划分,同时利用已有权重更新动态调整专家参数位置,避免迁移开销。由此,SYMI可在每轮迭代中精准配置GPU资源,最小化开销。相比最先进的MoE训练系统DeepSpeed和FlexMoE,SYMI分别实现了30.5%和25.9%的更快收敛速度。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) models have become a widely-adopted solution to continue scaling model sizes without a corresponding linear increase in compute. During MoE model training, each input token is dynamically routed to a subset of experts -- sparsely-activated feed-forward networks -- within each transformer layer. The distribution of tokens assigned to each expert varies widely and rapidly over the course of training. To handle the wide load imbalance across experts, current systems are forced to either drop tokens assigned to popular experts, degrading convergence, or frequently rebalance resources allocated to each expert based on popularity, incurring high state migration overheads. To break this performance-accuracy tradeoff, we introduce SYMI, an adaptive MoE training system. The key insight of SYMI is to decouple the placement of expert parameters from their large optimizer state. SYMI statically partitions the optimizer of each expert across all training nodes. Meanwhile, SYMI dynamically adjusts the placement of expert parameters by repurposing existing weight updates, avoiding migration overheads. In doing so, SYMI right-sizes the GPU resources allocated to each expert, on a per-iteration basis, with minimal overhead. Compared to state-of-the-art MoE training systems, DeepSpeed and FlexMoE, SYMI is able to achieve a 30.5% and 25.9% faster time-to-convergence, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。