论文标题
深度神经网络多机求解器的结构保存
Structure Preservation for the Deep Neural Network Multigrid Solver
论文作者
论文摘要
部分微分方程的模拟是数值分析的核心主题,也是科学,工程和相关领域必不可少的工具。现有的方法(例如有限元素)提供了(高度)的工具,但在过去几年中出现了基于神经网络的深层技术,作为替代方案,结果非常有希望。我们研究了两种方法的组合,用于近似Navier-Stokes方程以及在何种程度上可以并且应该尊重诸如Divergence Freedom之类的结构特性。我们的工作基于DNN-MG,这是一种深层神经网络多机技术,我们最近介绍了该技术,并使用神经网络代表了几何多机有限元元素求解器无法解决的细网格波动。尽管DNN-MG提供了非常准确的解决方案,并且在计算上具有高效,但我们注意到基于神经网络的校正实质上违反了速度矢量场的差异自由。在这项贡献中,我们讨论了这些发现,并分析了解决该问题的三种方法:鼓励网络输出分歧自由的罚款术语;校正速度场的罚款;以及一个学习流函数的网络,即无差异速度矢量场的标量电势,从而通过施工无差校正来屈服。我们的实验结果表明,基于流函数的第三种方法优于其他两个方法,不仅可以改善差异自由,而且尤其是模拟的总体保真度。
The simulation of partial differential equations is a central subject of numerical analysis and an indispensable tool in science, engineering and related fields. Existing approaches, such as finite elements, provide (highly) efficient tools but deep neural network-based techniques emerged in the last few years as an alternative with very promising results. We investigate the combination of both approaches for the approximation of the Navier-Stokes equations and to what extent structural properties such as divergence freedom can and should be respected. Our work is based on DNN-MG, a deep neural network multigrid technique, that we introduced recently and which uses a neural network to represent fine grid fluctuations not resolved by a geometric multigrid finite element solver. Although DNN-MG provides solutions with very good accuracy and is computationally highly efficient, we noticed that the neural network-based corrections substantially violate the divergence freedom of the velocity vector field. In this contribution, we discuss these findings and analyze three approaches to address the problem: a penalty term to encourage divergence freedom of the network output; a penalty term for the corrected velocity field; and a network that learns the stream function, i.e. the scalar potential of the divergence free velocity vector field and which hence yields by construction divergence free corrections. Our experimental results show that the third approach based on the stream function outperforms the other two and not only improves the divergence freedom but in particular also the overall fidelity of the simulation.