arXiv:2507.17346cs.LG2025-07

动态调整压缩与延迟同步,显著提升分布式训练效率

Taming Latency and Bandwidth: A Theoretical Framework and Adaptive Algorithm for Communication-Constrained Training

  • 提出理论框架分解通信优化问题,分离出可分析的噪声项
  • 发现延迟同步会指数放大压缩损失,影响收敛速度
  • 设计自适应算法DeCo-SGD,实时响应网络条件,提速达5.07倍

单个数据中心的区域能源限制制约了大规模模型训练的发展。当模型规模可控时,单中心训练可行,但模型与计算需求的指数增长使其难以为继。将训练分布到多个广域网数据中心是自然选择,虽可整合资源,却面临高延迟和低且波动的带宽,严重降低吞吐量。联合使用梯度压缩与延迟聚合可缓解通信瓶颈,但引入压缩率、延迟步数(滞后性)与收敛速度之间的复杂三重权衡。现有工作缺乏理论指导,仅能采用固定策略,无法适应计算与通信条件变化。本文提出新的理论工具,将联合优化问题分解为传统过程与多个可分析的噪声项。分析首次给出该场景下的收敛速率,并揭示:增加滞后性会指数级放大压缩的负面影响。基于此,提出DeCo-SGD算法,根据实时通信与计算条件动态选择压缩率与滞后步数。在高延迟和低且波动带宽网络中,DeCo-SGD相比分布式SGD和静态策略分别实现最高5.07倍和1.37倍加速。

原文摘要 · Abstract (English)

Regional energy caps limit the growth of any single data center used for large-scale model training. This single-center training paradigm works when model size remains manageable, but exponential growth in the model size and computational demand challenges it. A natural alternative is to distribute training across multiple data centers over wide-area networks. This pools distributed resources, but suffers from high latency and low, time-varying bandwidth, sharply reducing throughout. Employing jointly gradient compression and delayed aggregation can alleviate communication problems, but introduces a complex three-way trade-off among compression ratio, staleness (delayed synchronization steps), and convergence rate. Existing work lacks theoretical guidance and can only propose fixed strategies, insensitive to computation and communication conditions. We address this with a new theoretical tool, decomposing the joint optimization problem into a traditional process plus multiple analyzable noise terms. Our analysis yields the first convergence rate for this setting and shows that increasing staleness exponentially amplifies the detrimental effect of compression. Leveraging these insights, we propose DeCo-SGD, which dynamically selects the compression ratio and staleness based on the real-time communication and computation conditions. DeCo-SGD achieves up to $5.07\times$ and $1.37\times$ speed-ups over distributed SGD and static strategy in high-latency and low, varying bandwidth networks, respectively.

分布式训练通信优化自适应算法

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