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

用分布式异步和选择性优化加速神经网络训练(DASO)

Accelerating Neural Network Training with Distributed Asynchronous and Selective Optimization (DASO)

Steinbuch Centre for Computing (SCC)(施泰因布赫计算中心(SCC));Karlsruhe Institute of Technology (KIT)(卡尔斯鲁厄理工学院(KIT));Institute for Software Technology (SC)(软件技术研究所(SC));German Aerospace Center (DLF)(德国航空航天中心(DLF))

Daniel Coquelin, Charlotte Debus, Markus Götz, Fabrice von der Lehr, James Kahn, Martin Siggel, Achim Streit

摘要:

随着数据和模型复杂性的增加,训练神经网络所需的时间变得非常长。为了解决训练时间的指数增长,用户转向数据并行神经网络(DPNN),以利用计算机集群上的大规模分布式资源。目前的DPNN方法通过阻塞通信操作来同步和平均所有进程的梯度来实现网络参数更新。这种同步是算法的核心瓶颈。为了解决这个问题,我们引入了分布式异步和选择性优化(DASO)方法,该方法利用多GPU计算节点架构来加速网络训练。DASO使用分层和异步通信方案,包括节点本地和全局网络,同时在学习过程中调整全局同步速率。我们表明,与其他现有的数据并行训练方法相比,DASO在经典和最先进的网络上的训练时间减少了34%。

英文摘要:

With increasing data and model complexities, the time required to train neural networks has become prohibitively large. To address the exponential rise in training time, users are turning to data parallel neural networks (DPNN) to utilize large-scale distributed resources on computer clusters. Current DPNN approaches implement the network parameter updates by synchronizing and averaging gradients across all processes with blocking communication operations. This synchronization is the central algorithmic bottleneck. To combat this, we introduce the Distributed Asynchronous and Selective Optimization (DASO) method which leverages multi-GPU compute node architectures to accelerate network training. DASO uses a hierarchical and asynchronous communication scheme comprised of node-local and global networks while adjusting the global synchronization rate during the learning process. We show that DASO yields a reduction in training time of up to 34% on classical and state-of-the-art networks, as compared to other existing data parallel training methods.

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

↑