EGU24-12495, updated on 09 Mar 2024
EGU General Assembly 2024
© Author(s) 2024. This work is distributed under
the Creative Commons Attribution 4.0 License.

Hybrid neural differential equation models for atmospheric dynamics

Maximilian Gelbrecht1,2 and Niklas Boers1,2,3
Maximilian Gelbrecht and Niklas Boers
  • 1Department of Aerospace and Geodesy, Technical University of Munich, Munich, Germany (
  • 2Potsdam Institute for Climate Impact Research, Potsdam, Germany
  • 3Department of Mathematics and Global Systems Institute, University of Exeter, Exeter, UK

Combining process-based models in Earth system science with data-driven machine learning methods holds tremendous promise. Can we harness the best of both approaches? In our study, we integrate components of atmospheric models into artificial neural networks (ANN). The resulting hybrid atmospheric model can learn atmospheric dynamics from short trajectories while ensuring robust generalization and stability. We achieve this using the neural differential equations framework, combining ANNs with a differentiable, GPU-enabled version of the well-studied Marshall Molteni Quasigeostrophic Model (QG3). Similar to the approach of many atmospheric models, part of the model is computed in the spherical harmonics domain, and other parts in the grid domain. In our model, ANNs are used as parametrizations in both domains, and form together with the components of the QG3 model the right-hand side of our hybrid model. We showcase the capabilities of our model by demonstrating how it generalizes from the QG3 model to the significantly more complex primitive equation model of SpeedyWeather.jl. 

How to cite: Gelbrecht, M. and Boers, N.: Hybrid neural differential equation models for atmospheric dynamics, EGU General Assembly 2024, Vienna, Austria, 14–19 Apr 2024, EGU24-12495,, 2024.