arXiv:2601.02523math.OCcs.DC2026-01被引 4

提出首个在异构数据下理论最优的异步SGD算法,突破训练效率瓶颈。

First Provably Optimal Asynchronous SGD for Homogeneous and Heterogeneous Data

  • 通过筛选过时更新,实现同构数据下的最优时间复杂度
  • 引入结构化梯度表,使异构数据场景也达理论最优
  • 自适应分配任务,显著降低实际运行时间,适合大规模分布式训练

人工智能依赖大规模神经网络训练,需数千个GPU/TPU协同运行数周,消耗巨大算力与能源。然而优化算法进展缓慢,多数训练仍采用同步方法,需等待最慢设备,造成资源浪费。去同步看似简单,但会引入更新延迟(staleness),导致分析困难,尤其当延迟源于系统随机性而非算法设计时,异步方法的时间复杂度长期不明。本文构建了异步一阶随机优化的严格框架,聚焦异构工作节点速度差异问题。研究表明,合理设计下异步SGD可达到与同步方法相同的最优时间复杂度。提出Ringmaster ASGD,在同构数据下通过丢弃过时更新实现最优;Ringleader ASGD扩展至异构数据(如联邦学习),采用结构化梯度表机制;ATA则通过学习工作节点计算时间分布,动态分配任务,以更少计算量逼近最优墙钟时间。三者共同确立异步优化的理论可行性与实践高效性,证明无需同步即可实现最优协调。

原文摘要 · Abstract (English)

Artificial intelligence has advanced rapidly through large neural networks trained on massive datasets using thousands of GPUs or TPUs. Such training can occupy entire data centers for weeks and requires enormous computational and energy resources. Yet the optimization algorithms behind these runs have not kept pace. Most large scale training still relies on synchronous methods, where workers must wait for the slowest device, wasting compute and amplifying the effects of hardware and network variability. Removing synchronization seems like a simple fix, but asynchrony introduces staleness, meaning updates computed on outdated models. This makes analysis difficult, especially when delays arise from system level randomness rather than algorithmic choices. As a result, the time complexity of asynchronous methods remains poorly understood. This dissertation develops a rigorous framework for asynchronous first order stochastic optimization, focusing on the core challenge of heterogeneous worker speeds. Within this framework, we show that with proper design, asynchronous SGD can achieve optimal time complexity, matching guarantees previously known only for synchronous methods. Our first contribution, Ringmaster ASGD, attains optimal time complexity in the homogeneous data setting by selectively discarding stale updates. The second, Ringleader ASGD, extends optimality to heterogeneous data, common in federated learning, using a structured gradient table mechanism. Finally, ATA improves resource efficiency by learning worker compute time distributions and allocating tasks adaptively, achieving near optimal wall clock time with less computation. Together, these results establish asynchronous optimization as a theoretically sound and practically efficient foundation for distributed learning, showing that coordination without synchronization can be both feasible and optimal.

异步训练分布式学习优化算法联邦学习

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