The great learning ability of deep learning models facilitates us to comprehend the real physical world, making learning to simulate complicated particle systems a promising endeavour. However, the complex laws of the physical world pose significant challenges to the learning based simulations, such as the varying spatial dependencies between interacting particles and varying temporal dependencies between particle system states in different time stamps, which dominate particles' interacting behaviour and the physical systems' evolution patterns. Existing learning based simulation methods fail to fully account for the complexities, making them unable to yield satisfactory simulations. To better comprehend the complex physical laws, this paper proposes a novel learning based simulation model- Graph Networks with Spatial-Temporal neural Ordinary Equations (GNSTODE)- that characterizes the varying spatial and temporal dependencies in particle systems using a united end-to-end framework. Through training with real-world particle-particle interaction observations, GNSTODE is able to simulate any possible particle systems with high precisions. We empirically evaluate GNSTODE's simulation performance on two real-world particle systems, Gravity and Coulomb, with varying levels of spatial and temporal dependencies. The results show that the proposed GNSTODE yields significantly better simulations than state-of-the-art learning based simulation methods, which proves that GNSTODE can serve as an effective solution to particle simulations in real-world application.
翻译:深度学习模型的强大学习能力有助于我们理解真实物理世界,使学习模拟复杂粒子系统成为一项前景广阔的任务。然而,物理世界的复杂规律给基于学习的模拟带来了巨大挑战,例如相互作用粒子间时空依赖关系的动态变化,以及不同时间步粒子系统状态间的时变依赖性,这些因素主导着粒子的相互作用行为与物理系统的演化模式。现有基于学习的模拟方法未能充分考虑这些复杂性,难以生成令人满意的模拟结果。为更深入理解复杂物理规律,本文提出了一种新型基于学习的模拟模型——时空神经常微分方程图网络(GNSTODE),通过统一的端到端框架刻画粒子系统中动态变化的时空依赖关系。通过真实粒子对相互作用观测数据的训练,GNSTODE能够高精度模拟任意可能的粒子系统。我们在具有不同时空依赖程度的真实物理系统(重力系统与库仑系统)上对GNSTODE的模拟性能进行了实证评估。结果表明,所提出的GNSTODE相比最先进的基于学习的模拟方法实现了显著更优的模拟效果,证明其可作为真实应用中粒子模拟的有效解决方案。