arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.00845cs.LGcs.DCmath.OC
收藏

协调分布式示例顺序以实现可证明加速的训练

Coordinating Distributed Example Orders for Provably Accelerated Training

Cornell University(康奈尔大学)

A. Feder Cooper, Wentao Guo, Khiem Pham, Tiancheng Yuan, Charlie F. Ruan, Yucheng Lu, Christopher De Sa

AI总结:

针对分布式训练中示例顺序优化问题,提出CD-GraB方法,利用核稀疏化思想将GraB的可证明加速优势扩展到分布式环境,实现线性加速并优于分布式RR。

摘要:

近期关于在线梯度平衡(GraB)的研究揭示,存在基于排列的示例顺序用于SGD,其性能保证优于随机重排(RR)。RR任意排列训练示例,而GraB利用先前轮次的过时梯度来排序示例——实现比RR更快的可证明收敛速率。然而,GraB在设计上存在局限:虽然它在集中式数据上展现出令人瞩目的训练扩展能力,但并未自然扩展到现代分布式机器学习工作负载。因此,我们提出协调分布式GraB(CD-GraB),利用先前关于核稀疏化工作的见解,将基于排列的示例排序的可证明加速优势迁移到分布式环境。在开销可忽略的情况下,CD-GraB在收敛速率上相对于集中式GraB展现出线性加速,并在多种基准任务上优于分布式RR。

英文摘要:

Recent research on online Gradient Balancing (GraB) has revealed that there exist permutation-based example orderings for SGD that are guaranteed to outperform random reshuffling (RR). Whereas RR arbitrarily permutes training examples, GraB leverages stale gradients from prior epochs to order examples -- achieving a provably faster convergence rate than RR. However, GraB is limited by design: while it demonstrates an impressive ability to scale-up training on centralized data, it does not naturally extend to modern distributed ML workloads. We therefore propose Coordinated Distributed GraB (CD-GraB), which uses insights from prior work on kernel thinning to translate the benefits of provably faster permutation-based example ordering to distributed settings. With negligible overhead, CD-GraB exhibits a linear speedup in convergence rate over centralized GraB and outperforms distributed RR on a variety of benchmark tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑