arXiv:2506.04531cs.LG2025-06ICML被引 6

HALoS通过分层异步训练,加速跨地域大模型训练

HALoS: Hierarchical Asynchronous Local SGD over Slow Networks for Geo-Distributed Large Language Model Training

  • 分区域设置本地参数服务器,减少跨区通信开销
  • 比同步方法快7.5倍,比现有异步方法快2.1倍
  • 适合异构、跨地域的大模型训练场景

大语言模型(LLM)训练日益依赖地理分布的加速器,导致跨区域通信成本高昂且硬件利用率不均。本文提出HALoS,一种分层异步优化框架:每个区域内部署本地参数服务器(LPS),由全局参数服务器(GPS)合并跨区域更新。该设计显著降低昂贵的跨区通信,缓解慢节点影响,并利用快速的区域内链路。我们为非凸目标下的HALoS提供了严格的收敛分析,包括层级动量在异步训练中作用的理论保证。实验表明,HALoS在地理分布式LLM训练中相比同步基线实现最高7.5倍的加速,优于现有异步方法达2.1倍。关键的是,其模型质量与全同步SGD相当或更优,在标准语言建模和下游任务上保持高准确率,同时大幅缩短总训练时间。结果表明,分层服务器端更新累积与全局模型融合是新世代大模型在异构、地理分布环境下高效可扩展训练的强大工具。

原文摘要 · Abstract (English)

Training large language models (LLMs) increasingly relies on geographically distributed accelerators, causing prohibitive communication costs across regions and uneven utilization of heterogeneous hardware. We propose HALoS, a hierarchical asynchronous optimization framework that tackles these issues by introducing local parameter servers (LPSs) within each region and a global parameter server (GPS) that merges updates across regions. This hierarchical design minimizes expensive inter-region communication, reduces straggler effects, and leverages fast intra-region links. We provide a rigorous convergence analysis for HALoS under non-convex objectives, including theoretical guarantees on the role of hierarchical momentum in asynchronous training. Empirically, HALoS attains up to 7.5x faster convergence than synchronous baselines in geo-distributed LLM training and improves upon existing asynchronous methods by up to 2.1x. Crucially, HALoS preserves the model quality of fully synchronous SGD-matching or exceeding accuracy on standard language modeling and downstream benchmarks-while substantially lowering total training time. These results demonstrate that hierarchical, server-side update accumulation and global model merging are powerful tools for scalable, efficient training of new-era LLMs in heterogeneous, geo-distributed environments.

大模型训练异步优化分布式系统跨区域通信

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