Timezone: »

Learning Controllable Adaptive Simulation for Multi-scale Physics
Tailin Wu · Takashi Maruyama · Qingqing Zhao · Gordon Wetzstein · Jure Leskovec
Event URL: https://openreview.net/forum?id=PhktEpJHU3 »

Simulating the time evolution of physical systems is pivotal in many scientific and engineering problems. An open challenge in simulating such systems is their multi-scale dynamics: a small fraction of the system is extremely dynamic, and requires very fine-grained resolution, while a majority of the system is changing slowly and can be modeled by coarser spatial scales. Typical learning-based surrogate models use a uniform spatial scale, which needs to resolve to the finest required scale and can waste a huge compute to achieve required accuracy. In this work, we introduce Learning controllable Adaptive simulation for Multi-scale Physics (LAMP) as the first full deep learning-based surrogate model that jointly learns the evolution model and optimizes appropriate spatial resolutions that devote more compute to the highly dynamic regions. LAMP consists of a Graph Neural Network (GNN) for learning the forward evolution, and a GNN-based actor-critic for learning the policy of spatial refinement and coarsening. We introduce learning techniques that optimizes LAMP with weighted sum of error and computational cost as objective, which allows LAMP to adapt to varying relative importance of error vs. computation tradeoff at inference time. We test our method in a 1D benchmark of nonlinear PDEs and a challenging 2D mesh-based simulation. We demonstrate that our LAMP outperforms state-of-the-art deep learning surrogate models with up to 39.3% error reduction, and is able to adaptively trade-off computation to improve long-term prediction error.

Author Information

Tailin Wu (Stanford)
Tailin Wu

Tailin Wu is a postdoc researcher in the Department of Computer Science at Stanford University, working with professor Jure Leskovec. His research interests lies in AI for large-scale simulations of complex systems, AI for scientific discovery, and representation learning

Takashi Maruyama (NEC Corporation)

Research engineer @ NEC Corporation Japan.

Qingqing Zhao (Stanford University)
Gordon Wetzstein (Stanford University)
Jure Leskovec (Stanford University/Pinterest)

More from the Same Authors