论文标题
教不可压缩的Navier-Stokes方程以在3D中快速神经替代模型
Teaching the Incompressible Navier-Stokes Equations to Fast Neural Surrogate Models in 3D
论文作者
论文摘要
物理上合理的流体模拟在现代计算机图形和工程中起着重要作用。但是,为了实现实时性能,需要以身体准确性进行计算速度。基于神经网络的替代流体模型具有实现快速液体模拟和高物理精度的潜力。但是,这些方法依赖大量的培训数据,需要复杂的管道来进行培训和推理,或者不推广到新的流体域。 在这项工作中,我们为最近提出的深度学习框架提供了重大扩展,该框架解决了上述2D中提到的挑战。我们从2D到3D,并提出了一个有效的体系结构,以应对记忆和计算复杂性方面的3D网格需求。此外,我们将神经流体模型调节于有关流体粘度和密度的其他信息,这些信息允许基于相同的替代模型模拟层流以及湍流。 我们的方法允许训练流体模型,而无需事先进行流体模拟数据。推断是快速而简单的,因为流体模型暂时将流体状态和边界条件直接映射到T+DT处的随后的流体状态。我们在128x64x64网格上获得了实时流体模拟,其中包括各种流体现象,例如Magnus效应或Karman Vortex街道,并推广到训练期间未考虑的域几何形状。我们的方法表明,与当前基于3D NN的流体模型相比,准确性,速度和概括能力方面有很大改善。
Physically plausible fluid simulations play an important role in modern computer graphics and engineering. However, in order to achieve real-time performance, computational speed needs to be traded-off with physical accuracy. Surrogate fluid models based on neural networks have the potential to achieve both, fast fluid simulations and high physical accuracy. However, these approaches rely on massive amounts of training data, require complex pipelines for training and inference or do not generalize to new fluid domains. In this work, we present significant extensions to a recently proposed deep learning framework, which addresses the aforementioned challenges in 2D. We go from 2D to 3D and propose an efficient architecture to cope with the high demands of 3D grids in terms of memory and computational complexity. Furthermore, we condition the neural fluid model on additional information about the fluid's viscosity and density which allows simulating laminar as well as turbulent flows based on the same surrogate model. Our method allows to train fluid models without requiring fluid simulation data beforehand. Inference is fast and simple, as the fluid model directly maps a fluid state and boundary conditions at a moment t to a subsequent fluid state at t+dt. We obtain real-time fluid simulations on a 128x64x64 grid that include various fluid phenomena such as the Magnus effect or Karman vortex streets and generalize to domain geometries not considered during training. Our method indicates strong improvements in terms of accuracy, speed and generalization capabilities over current 3D NN-based fluid models.