Approximating the Solution of Surface Wave Propagation Using Deep Neural Networks

Abstract

Partial differential equations formalise the understanding of the behaviour of the physical world that humans acquire through experience and observation. Through their numerical solution, such equations are used to model and predict the evolution of dynamical systems. However, such techniques require extensive computational resources and assume the physics are prescribed a priori. Here, we propose a neural network capable of predicting the evolution of a specific physical phenomenon: propagation of surface waves enclosed in a tank, which, mathematically, can be described by the Saint-Venant equations. The existence of reflections and interference makes this problem non-trivial. Forecasting of future states (i.e. spatial patterns of rendered wave amplitude) is achieved from a relatively small set of initial observations. Using a network to make approximate but rapid predictions would enable the active, real-time control of physical systems, often required for engineering design. We used a deep neural network comprising of three main blocks: an encoder, a propagator with three parallel Long Short-Term Memory layers, and a decoder. Results on a novel, custom dataset of simulated sequences produced by a numerical solver show reasonable predictions for as long as 80 time steps into the future on a hold-out dataset. Furthermore, we show that the network is capable of generalising to two other initial conditions that are qualitatively different from those seen at training time.

Supported by the Rosetrees trust.

W. E. Sorteberg and S. Garasto—These authors are joint first authors.