arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.10015cs.LG
收藏

DiffusionNet:使用深度学习加速时间依赖型偏微分方程的求解

DiffusionNet: Accelerating the solution of Time-Dependent partial differential equations using deep learning

KAIST(韩国科学技术院)

Mahmoud Asem

摘要:

我们提出了我们的深度学习框架来解决和加速一维和二维空间的时间依赖偏微分方程的解。我们通过求解具有Dirichlet边界条件的2D瞬态热传导问题来演示DiffusionNet求解器。该模型在使用交替方向隐式方法计算的解决方案数据上进行训练。我们展示了该模型从七个变量的任何组合来预测解决方案的能力:解决方案的起始时间步长,初始条件,四个边界条件,以及时间步长,扩散系数常数和网格步长的组合变量。为了提高速度,我们利用我们的模型能力来预测多个时间步长后的时间相关PDE的解决方案,以提高解决方案的速度,通过将解决方案分成可并行化的块。我们试图建立一个灵活的架构,能够解决广泛的偏微分方程,以最小的变化。我们证明了我们的模型的灵活性,通过应用我们的模型与用于解决瞬态热传导的相同的网络架构来解决无粘Burgers方程和稳态热传导,然后将我们的模型性能与相关研究进行比较。我们表明,我们的模型减少了调查问题的解决方案的错误。

英文摘要:

We present our deep learning framework to solve and accelerate the Time-Dependent partial differential equation's solution of one and two spatial dimensions. We demonstrate DiffusionNet solver by solving the 2D transient heat conduction problem with Dirichlet boundary conditions. The model is trained on solution data calculated using the Alternating direction implicit method. We show the model's ability to predict the solution from any combination of seven variables: the starting time step of the solution, initial condition, four boundary conditions, and a combined variable of the time step size, diffusivity constant, and grid step size. To improve speed, we exploit our model capability to predict the solution of the Time-dependent PDE after multiple time steps at once to improve the speed of solution by dividing the solution into parallelizable chunks. We try to build a flexible architecture capable of solving a wide range of partial differential equations with minimal changes. We demonstrate our model flexibility by applying our model with the same network architecture used to solve the transient heat conduction to solve the Inviscid Burgers equation and Steady-state heat conduction, then compare our model performance against related studies. We show that our model reduces the error of the solution for the investigated problems.

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

↑