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

求解偏微分方程的深度生成模型:用于训练大型无数据模型的分布式计算

Deep Generative Models that Solve PDEs: Distributed Computing for Training Large Data-Free Models

RocketML Inc.(RocketML 公司);New York University (NYU)(纽约大学);Iowa State University(爱荷华州立大学);RocketML Inc(RocketML 公司)

Sergio Botelho, Ameya Joshi, Biswajit Khara, Soumik Sarkar, Chinmay Hegde, Santi Adavani, Baskar Ganapathysubramanian

摘要:

科学机器学习(SciML)的最新进展开辟了训练解决复杂偏微分方程(PDE)的新型神经网络架构的可能性。最近有几种(几乎无数据)方法成功地解决了PDE,例如深度前馈网络,生成网络和深度编码器-解码器网络。然而,这些方法的实际采用受到训练这些模型的困难的限制,特别是在大输出分辨率($\geq 1024 \times 1024$)下进行预测。在这里,我们报告了一个用于数据并行分布式深度学习的软件框架,它解决了训练这些大型SciML模型的双重挑战-在合理的时间内训练以及分配存储需求。我们的框架提供了几个开箱即用的功能,包括(a)损失的完整性独立于进程的数量,(b)同步批量归一化,(c)分布式高阶优化方法。我们展示了这个框架在云和HPC集群上的出色可扩展性,并报告了带宽,网络拓扑和裸机与云之间的相互作用。我们部署这种方法来训练生成模型的大小迄今不可能,表明神经PDE求解器可以可行地训练实际应用。我们还证明了分布式高阶优化方法比基于随机梯度的方法快$2-3\times$,并提供最小的收敛漂移,具有更高的批量大小。

英文摘要:

Recent progress in scientific machine learning (SciML) has opened up the possibility of training novel neural network architectures that solve complex partial differential equations (PDEs). Several (nearly data free) approaches have been recently reported that successfully solve PDEs, with examples including deep feed forward networks, generative networks, and deep encoder-decoder networks. However, practical adoption of these approaches is limited by the difficulty in training these models, especially to make predictions at large output resolutions ($\geq 1024 \times 1024$). Here we report on a software framework for data parallel distributed deep learning that resolves the twin challenges of training these large SciML models - training in reasonable time as well as distributing the storage requirements. Our framework provides several out of the box functionality including (a) loss integrity independent of number of processes, (b) synchronized batch normalization, and (c) distributed higher-order optimization methods. We show excellent scalability of this framework on both cloud as well as HPC clusters, and report on the interplay between bandwidth, network topology and bare metal vs cloud. We deploy this approach to train generative models of sizes hitherto not possible, showing that neural PDE solvers can be viably trained for practical applications. We also demonstrate that distributed higher-order optimization methods are $2-3\times$ faster than stochastic gradient-based methods and provide minimal convergence drift with higher batch-size.

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

↑