arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.08802math.NAcs.LGcs.NA
收藏

模拟时间并行算法的Parareal神经网络

Parareal Neural Networks Emulating a Parallel-in-time Algorithm

KAIST(韩国科学技术院);Natural Science Research Institute, KAIST(韩国科学技术院自然科学研究院)

Chang-Ock Lee, Youngkyu Lee, Jongho Park

AI总结:

本文提出将DNN各层视为时间步,通过模拟parareal时间并行算法构建可多GPU并行的神经网络,并在VGG-16和ResNet-1001上验证了加速且精度保持的效果。

摘要:

随着深度神经网络(DNN)变得更深,训练时间也随之增加。从这一视角来看,多GPU并行计算已成为加速DNN训练的关键工具。本文提出了一种新颖的方法,从给定的DNN出发构建一个能够同时利用多个GPU的并行神经网络。我们观察到,DNN的各层可以被解释为时间依赖问题中的时间步,并且可以通过模拟一种称为parareal的时间并行算法来实现并行化。parareal算法由可以并行实现的细结构和为细结构提供合适近似的粗结构组成。通过模拟该算法,DNN的各层被拆分以形成一个并行结构,并使用一个合适的粗网络进行连接。我们报告了将该方法应用于VGG-16和ResNet-1001在多个数据集上的加速且精度保持的结果。

英文摘要:

As deep neural networks (DNNs) become deeper, the training time increases. In this perspective, multi-GPU parallel computing has become a key tool in accelerating the training of DNNs. In this paper, we introduce a novel methodology to construct a parallel neural network that can utilize multiple GPUs simultaneously from a given DNN. We observe that layers of DNN can be interpreted as the time step of a time-dependent problem and can be parallelized by emulating a parallel-in-time algorithm called parareal. The parareal algorithm consists of fine structures which can be implemented in parallel and a coarse structure which gives suitable approximations to the fine structures. By emulating it, the layers of DNN are torn to form a parallel structure which is connected using a suitable coarse network. We report accelerated and accuracy-preserved results of the proposed methodology applied to VGG-16 and ResNet-1001 on several datasets.

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

↑