arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2311.04499cs.DC
收藏

采用重叠感知梯度压缩的近线性扩展数据并行训练

Near-Linear Scaling Data Parallel Training with Overlapping-Aware Gradient Compression

Lin Meng, Yuzhong Sun, Weimin Li

AI总结:

针对数据并行训练通信瓶颈,本文提出 COVAP 梯度压缩方案,以近零开销粗粒度过滤、自适应压缩率、张量分片和误差反馈实现通信计算近完全重叠,并在阿里云实验中取得近线性扩展。

摘要:

深度神经网络(DNN)现有的数据并行(DP)训练往往因巨大的通信开销而在加速比方面表现出有限的可扩展性。虽然 Overlapping(重叠)技术能够通过在 DP 中并行执行通信与计算来缓解这一问题,但其有效性受到 DP 训练任务较高通信计算比(CCR)的制约。梯度压缩(GC)是一项很有前景的技术,可通过直接减少通信量来降低 CCR。然而,将 GC 应用于 Overlapping 以获得真实性能提升颇具挑战,原因在于:(1)传统 GC 因较高压缩开销而产生严重性能损失;(2)GC 方案中可能存在的数据依赖会导致 Overlapping 收益下降。本文提出 COVAP,这是一种新型 GC 方案,设计了新的粗粒度过滤器,使压缩开销接近于零。COVAP 通过采用针对特定训练任务定制的自适应压缩率和张量分片,确保通信与计算几乎完全重叠。COVAP 还采用改进的误差反馈机制来维持训练精度。实验在阿里云 ECS 实例上进行,使用真实应用中的不同 DNN。结果表明,COVAP 在求解时间方面优于现有 GC 方案 1.92 倍至 15.39 倍,并展现出近线性扩展。此外,在四种不同集群规模的实验中,COVAP 实现了最佳可扩展性。

英文摘要:

Existing Data Parallel (DP) trainings for deep neural networks (DNNs) often experience limited scalability in speedup due to substantial communication overheads. While Overlapping technique can mitigate such problem by paralleling communication and computation in DP, its effectiveness is constrained by the high communication-to-computation ratios (CCR) of DP training tasks. Gradient compression (GC) is a promising technique to obtain lower CCR by reducing communication volume directly. However, it is challenging to obtain real performance improvement by applying GC into Overlapping because of (1) severe performance penalties in traditional GCs caused by high compression overhead and (2) decline of Overlapping benefit owing to the possible data dependency in GC schemes. In this paper, we propose COVAP, a novel GC scheme designing a new coarse-grained filter, makes the compression overhead close to zero. COVAP ensures an almost complete overlap of communication and computation by employing adaptive compression ratios and tensor sharding tailored to specific training tasks. COVAP also adopts an improved error feedback mechanism to maintain training accuracy. Experiments are conducted on Alibaba Cloud ECS instances with different DNNs of real-world applications. The results illustrate that COVAP outperforms existent GC schemes in time-to-solution by 1.92x-15.39x and exhibits near-linear scaling. Furthermore, COVAP achieves best scalability under experiments on four different cluster sizes.

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

↑