Title: Temporal Subsampling Diminishes Small Spatial Scales in Recurrent Neural Network Emulators of Geophysical Turbulence

Journal Article · · Journal of Advances in Modeling Earth Systems
ORCiD logo [1]; ORCiD logo [2]; ORCiD logo [3]; ORCiD logo [4]
  1. University of Colorado, Boulder, CO (United States); National Oceanic and Atmospheric Administration (NOAA), Boulder, CO (United States)
  2. University of Colorado, Boulder, CO (United States); Sofar Ocean, San Francisco, CA (United States)
  3. University of California San Diego, La Jolla, CA (United States)
  4. Pacific Northwest National Laboratory (PNNL), Richland, WA (United States)

The immense computational cost of traditional numerical weather and climate models has sparked the development of machine learning (ML) based emulators. Because ML methods benefit from long records of training data, it is common to use data sets that are temporally subsampled relative to the time steps required for the numerical integration of differential equations. Here, we investigate how this often overlooked processing step affects the quality of an emulator's predictions. We implement two ML architectures from a class of methods called reservoir computing: (a) a form of Nonlinear Vector Autoregression (NVAR), and (b) an Echo State Network (ESN). Despite their simplicity, it is well documented that these architectures excel at predicting low dimensional chaotic dynamics. We are therefore motivated to test these architectures in an idealized setting of predicting high dimensional geophysical turbulence as represented by Surface Quasi-Geostrophic dynamics. In all cases, subsampling the training data consistently leads to an increased bias at small spatial scales that resembles numerical diffusion. Interestingly, the NVAR architecture becomes unstable when the temporal resolution is increased, indicating that the polynomial based interactions are insufficient at capturing the detailed nonlinearities of the turbulent flow. The ESN architecture is found to be more robust, suggesting a benefit to the more expensive but more general structure. Spectral errors are reduced by including a penalty on the kinetic energy density spectrum during training, although the subsampling related errors persist. Future work is warranted to understand how the temporal resolution of training data affects other ML architectures.

Research Organization:
Pacific Northwest National Laboratory (PNNL), Richland, WA (United States)
Sponsoring Organization:
National Oceanic and Atmospheric Administration (NOAA); Office of Naval Research (ONR); USDOE
Grant/Contract Number:
AC05-76RL01830
OSTI ID:
2278630
Report Number(s):
PNNL-SA-183691
Journal Information:
Journal of Advances in Modeling Earth Systems, Journal Name: Journal of Advances in Modeling Earth Systems Journal Issue: 12 Vol. 15; ISSN 1942-2466
Publisher:
American Geophysical Union (AGU)Copyright Statement
Country of Publication:
United States
Language:
English

References (56)

Statistics for Spatial Data book September 1993
The ERA5 global reanalysis journal June 2020
Data Assimilation Fundamentals book April 2022
Overview of methods for the verification of quantitative precipitation forecasts book January 2008
A Practical Guide to Applying Echo State Networks book January 2012
Scalable gradient–enhanced artificial neural networks for airfoil shape design in the subsonic and transonic regimes journal March 2020
Predicting shallow water dynamics using echo-state networks with transfer learning journal November 2022
A Python surrogate modeling framework with derivatives journal September 2019
DeePr-ESN: A deep projection-encoding echo-state network journal February 2020
Deep reservoir computing: A critical experimental analysis journal December 2017
Investigation of proper orthogonal decomposition for echo state networks journal September 2023
Design of deep echo state networks journal December 2018
Backpropagation algorithms and Reservoir Computing in Recurrent Neural Networks for the forecasting of complex spatiotemporal dynamics journal June 2020
A systematic exploration of reservoir computing for forecasting complex spatiotemporal dynamics journal September 2022
A linear stochastic emulator of the California Current system using balanced truncation journal June 2022
Surface quasi-geostrophic dynamics journal January 1995
Efficient Global Optimization of Expensive Black-Box Functions journal January 1998
Toward Data‐Driven Weather and Climate Forecasting: Approximating a Simple General Circulation Model With Deep Learning journal November 2018
Can Machines Learn to Predict Weather? Using Deep Learning to Predict Gridded 500‐hPa Geopotential Height From Historical Weather Data journal August 2019
A Machine Learning‐Based Global Atmospheric Forecast Model journal May 2020
Improving Data‐Driven Global Weather Prediction Using Deep Convolutional Neural Networks on a Cubed Sphere journal September 2020
Reservoir Computing as a Tool for Climate Predictability Studies journal April 2021
Data‐Driven Medium‐Range Weather Prediction With a Resnet Pretrained on Climate Simulations: A New Model for WeatherBench journal February 2021
Sub‐Seasonal Forecasting With a Large Ensemble of Deep‐Learning Weather Prediction Models journal July 2021
A Comparison of Data‐Driven Approaches to Build Low‐Dimensional Ocean Models journal September 2021
Integrating Recurrent Neural Networks With Data Assimilation for Scalable Data‐Driven State Estimation journal March 2022
PIPs and POPs: The reduction of complex dynamical systems using principal interaction and oscillation patterns journal January 1988
Next generation reservoir computing journal September 2021
Accurate medium-range global weather forecasting with 3D neural networks journal July 2023
SciPy 1.0: fundamental algorithms for scientific computing in Python journal February 2020
Using machine learning to replicate chaotic attractors and calculate Lyapunov exponents from data journal December 2017
Attractor reconstruction by machine learning journal June 2018
Forecasting chaotic systems with very low connectivity reservoir computers
  • Griffith, Aaron; Pomerance, Andrew; Gauthier, Daniel J.
  • Chaos: An Interdisciplinary Journal of Nonlinear Science, Vol. 29, Issue 12 https://doi.org/10.1063/1.5120710
journal December 2019
On explaining the surprising success of reservoir computing forecaster of chaos? The universal machine learning dynamical system with contrast to VAR and DMD journal January 2021
Reducing echo state network size with controllability matrices journal July 2022
Learning spatiotemporal chaos using next-generation reservoir computing journal September 2022
Constraining chaos: Enforcing dynamical invariants in the training of reservoir computers
  • Platt, Jason A.; Penny, Stephen G.; Smith, Timothy A.
  • Chaos: An Interdisciplinary Journal of Nonlinear Science, Vol. 33, Issue 10 https://doi.org/10.1063/5.0156999
journal October 2023
Hierarchical architectures in reservoir computing systems journal August 2021
Can deep learning beat numerical weather prediction?
  • Schultz, M. G.; Betancourt, C.; Gong, B.
  • Philosophical Transactions of the Royal Society A: Mathematical, Physical and Engineering Sciences, Vol. 379, Issue 2194 https://doi.org/10.1098/rsta.2020.0097
journal February 2021
Generalizability of reservoir computing for flux-driven two-dimensional convection journal November 2022
Model-Free Prediction of Large Spatiotemporally Chaotic Systems from Data: A Reservoir Computing Approach journal January 2018
Catch-22s of reservoir computing journal September 2023
Backpropagation-decorrelation: online recurrent learning with O(N) complexity conference January 2004
Memory in reservoirs for high dimensional input conference July 2010
Self-Organizing Maps and Scale-Invariant Maps in Echo State Networks conference November 2011
Multilayered Echo State Machine: A Novel Architecture and Algorithm journal April 2017
Long Waves and Cyclone Waves journal August 1949
Learning skillful medium-range global weather forecasting journal December 2023
Uncertainty Quantification and Polynomial Chaos Techniques in Computational Fluid Dynamics journal January 2009
Real-Time Computing Without Stable States: A New Framework for Neural Computation Based on Perturbations journal November 2002
A Note on the Numerical Representation of Surface Dynamics in Quasigeostrophic Turbulence: Application to the Nonlinear Eady Model journal April 2009
Coupled Data Assimilation for Integrated Earth System Analysis and Prediction journal July 2017
Data-Based Approach for Fast Airfoil Analysis and Optimization journal February 2019
Challenges and design choices for global weather and climate models based on machine learning journal January 2018
Weather and climate forecasting with neural networks: using general circulation models (GCMs) with different complexity as a study ground journal January 2019
The impact of resolving the Rossby radius at mid-latitudes in the ocean: results from a high-resolution version of the Met Office GC2 coupled model journal January 2016