arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1705.09786cs.LGcs.AIcs.DCstat.ML
收藏

AMPNet:动态神经网络的异步模型并行训练

AMPNet: Asynchronous Model-Parallel Training for Dynamic Neural Networks

Google Brain(谷歌大脑);Microsoft Research(微软研究院)

Alexander L. Gaunt, Matthew A. Johnson, Maik Riechert, Daniel Tarlow, Ryota Tomioka, Dimitrios Vytiniotis, Sam Webster

摘要:

正在开发和进入市场的新型机器学习硬件有望以与GPU一样深刻的方式彻底改变深度学习。然而,现有的用于深度学习的软件框架和训练算法尚未发展到完全利用新一波硅的能力。我们已经看到了现有算法的局限性,这些算法通过复杂的和依赖于实例的控制流来利用结构化输入,这就禁止了minimizing。我们提出了一个异步模型并行(AMP)的训练算法,是专门由互联设备的网络上的训练。通过在多核CPU上的实现,我们表明AMP训练在相似数量的时期内收敛到与传统同步训练算法相同的精度,但即使对于小的minibatch大小,也可以更有效地利用可用硬件,从而显着缩短整体训练时间。我们的框架为扩展一类新的深度学习模型打开了大门,这些模型目前无法有效训练。

英文摘要:

New types of machine learning hardware in development and entering the market hold the promise of revolutionizing deep learning in a manner as profound as GPUs. However, existing software frameworks and training algorithms for deep learning have yet to evolve to fully leverage the capability of the new wave of silicon. We already see the limitations of existing algorithms for models that exploit structured input via complex and instance-dependent control flow, which prohibits minibatching. We present an asynchronous model-parallel (AMP) training algorithm that is specifically motivated by training on networks of interconnected devices. Through an implementation on multi-core CPUs, we show that AMP training converges to the same accuracy as conventional synchronous training algorithms in a similar number of epochs, but utilizes the available hardware more efficiently even for small minibatch sizes, resulting in significantly shorter overall training times. Our framework opens the door for scaling up a new class of deep learning models that cannot be efficiently trained today.

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

↑