arXiv:2605.20866cs.LGcs.DC2026-05

提出可重叠通信计算的稀疏聚合算法,提升分布式训练效率。

LOSCAR-SGD: Local SGD with Communication-Computation Overlap and Delay-Corrected Sparse Model Averaging

论文配图:LOSCAR-SGD: Local SGD with Communication-Computation Overlap and Delay-Corrected Sparse Model Averaging
图 1 · 摘自论文原文
  • 通信时并行本地优化,只传模型部分参数
  • 延迟修正聚合机制,避免信息丢失
  • 适合异构设备、慢网络的分布式训练场景

分布式学习中通信是主要瓶颈,尤其在大规模训练和联邦学习的慢链路环境下。现有方法通常结合通信压缩、本地训练和通信-计算重叠三种策略,但三者协同的理论分析仍不足。本文研究异构计算场景下各工作节点执行不同本地步数的情况,提出LOSCAR-SGD:仅传输模型部分坐标,并在通信期间持续本地优化。核心是延迟修正聚合规则,保留重叠阶段的更新进度,不丢弃延迟同步的信息。对光滑非凸目标函数给出了收敛性保证,揭示了稀疏性、重叠性和节点异构性对收敛速率的影响。据我们所知,这是首个针对该组合策略的理论分析。实验表明,通信-计算重叠显著缩短训练时间,延迟修正聚合优于简单覆盖策略。

原文摘要 · Abstract (English)

Communication is a major bottleneck in distributed learning, especially in large-scale settings and in federated learning environments with slow links. Three standard ways to reduce this cost are communication compression, local training, and communication-computation overlap. Methods that combine these ingredients are used in practice and have been found to be effective for large-scale training, but there is little theory for methods that combine all three. We study a heterogeneous-compute setting in which different workers may take different numbers of local steps, and we propose LOSCAR-SGD, a Local SGD method that communicates only a sparse subset of model coordinates and continues optimizing while communication is in flight. A key ingredient is a delay-corrected merge rule that incorporates delayed synchronized information without discarding the progress made during the overlap phase. We give convergence guarantees for smooth non-convex objectives and show how sparsity, overlap, and worker heterogeneity affect the rate. To the best of our knowledge, this is the first theory for this combination of ingredients. Experiments further show that communication-computation overlap reduces training time and that the delay-corrected merge outperforms naive overwriting.

分布式训练通信优化异构计算

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