arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2011.01349cs.DCcs.NAmath.NA
收藏

基于MPI的分布式机器学习

Distributed Machine Learning for Computational Engineering using MPI

Kailai Xu, Weiqiang Zhu, Eric Darve

摘要:

我们提出了一个训练神经网络,耦合偏微分方程(PDE)在并行计算环境中的框架。与大多数深度神经网络的分布式计算框架不同,我们的重点是在向前和伴随计算中并行化数值求解器和深度神经网络。我们的并行计算模型将数据通信视为数值模拟计算图中的一个节点。我们的模型的优点是,数据通信和计算是干净的分离,从而提供更好的灵活性,模块化和可测试性。我们使用各种大规模问题证明,我们可以通过使用PDE的并行求解器来训练与PDE耦合的深度神经网络,从而实现大幅加速。

英文摘要:

We propose a framework for training neural networks that are coupled with partial differential equations (PDEs) in a parallel computing environment. Unlike most distributed computing frameworks for deep neural networks, our focus is to parallelize both numerical solvers and deep neural networks in forward and adjoint computations. Our parallel computing model views data communication as a node in the computational graph for numerical simulations. The advantage of our model is that data communication and computing are cleanly separated and thus provide better flexibility, modularity, and testability. We demonstrate using various large-scale problems that we can achieve substantial acceleration by using parallel solvers for PDEs in training deep neural networks that are coupled with PDEs.

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

↑