STENCIL-NET for equation-free forecasting from data

Research output: Contribution to journalResearch articleContributedpeer-review

Contributors

  • Suryanarayana Maddu - , Chair of Scientific Computing for Systems Biology, Max Planck Institute of Molecular Cell Biology and Genetics, Center for Systems Biology Dresden (CSBD), Center for Scalable Data Analytics and Artificial Intelligence (ScaDS.AI) Dresden/Leipzig (Author)
  • Dominik Sturm - , Helmholtz-Zentrum Dresden-Rossendorf, Center for Advanced Systems Understanding (CASUS) (Author)
  • Bevan L. Cheeseman - , Chair of Scientific Computing for Systems Biology, Max Planck Institute of Molecular Cell Biology and Genetics, Center for Systems Biology Dresden (CSBD) (Author)
  • Christian L. Müller - , Ludwig Maximilian University of Munich, Helmholtz Zentrum München - German Research Center for Environmental Health, Flatiron Institute (Author)
  • Ivo F. Sbalzarini - , Chair of Scientific Computing for Systems Biology, Max Planck Institute of Molecular Cell Biology and Genetics, Center for Systems Biology Dresden (CSBD), Center for Scalable Data Analytics and Artificial Intelligence (ScaDS.AI) Dresden/Leipzig (Author)

Abstract

We present an artificial neural network architecture, termed STENCIL-NET, for equation-free forecasting of spatiotemporal dynamics from data. STENCIL-NET works by learning a discrete propagator that is able to reproduce the spatiotemporal dynamics of the training data. This data-driven propagator can then be used to forecast or extrapolate dynamics without needing to know a governing equation. STENCIL-NET does not learn a governing equation, nor an approximation to the data themselves. It instead learns a discrete propagator that reproduces the data. It therefore generalizes well to different dynamics and different grid resolutions. By analogy with classic numerical methods, we show that the discrete forecasting operators learned by STENCIL-NET are numerically stable and accurate for data represented on regular Cartesian grids. A once-trained STENCIL-NET model can be used for equation-free forecasting on larger spatial domains and for longer times than it was trained for, as an autonomous predictor of chaotic dynamics, as a coarse-graining method, and as a data-adaptive de-noising method, as we illustrate in numerical experiments. In all tests, STENCIL-NET generalizes better and is computationally more efficient, both in training and inference, than neural network architectures based on local (CNN) or global (FNO) nonlinear convolutions.

Details

Original languageEnglish
Article number12787
JournalScientific reports
Volume13
Issue number1
Publication statusPublished - Dec 2023
Peer-reviewedYes

External IDs

PubMed 37550328
ORCID /0000-0003-4414-4340/work/159608267

Keywords

ASJC Scopus subject areas