Spatio-temporal modeling and forecasting with Fourier neural operators
Date:
Authors: P. Nag, A. Zammit-Mangion, S. Singh, N. Cressie
Spatio-temporal process models are often used for modeling dynamical physical and biological phenomena that evolve across space and time. These phenomena may exhibit environmental heterogeneity and complex interactions that are difficult to capture using traditional statistical process models such as Gaussian processes. This work proposes the use of Fourier neural operators (FNOs) for constructing statistical dynamical spatio-temporal models for forecasting. An FNO is a flexible mapping of functions that approximates the solution operator of possibly unknown linear or nonlinear partial differential equations (PDEs) in a computationally efficient manner. It does so using samples of inputs and their respective outputs, and hence explicit knowledge of an underlying PDE is not required. Through a test case where we simulate from a nonlinear PDE with known solution, we compare FNO forecasts to those from state-of-the-art statistical spatio-temporal forecasting methods.