English

Machine learning emulation of gravity wave drag in numerical weather forecasting

Atmospheric and Oceanic Physics 2021-08-11 v2 Computational Physics

Abstract

We assess the value of machine learning as an accelerator for the parameterisation schemes of operational weather forecasting systems, specifically the parameterisation of non-orographic gravity wave drag. Emulators of this scheme can be trained to produce stable and accurate results up to seasonal forecasting timescales. Generally, more complex networks produce more accurate emulators. By training on an increased complexity version of the existing parameterisation scheme we build emulators that produce more accurate forecasts. {For medium range forecasting we find evidence our emulators are more accurate} than the version of the parametrisation scheme that is used for operational predictions. Using the current operational CPU hardware our emulators have a similar computational cost to the existing scheme, but are heavily limited by data movement. On GPU hardware our emulators perform ten times faster than the existing scheme on a CPU.

Keywords

Cite

@article{arxiv.2101.08195,
  title  = {Machine learning emulation of gravity wave drag in numerical weather forecasting},
  author = {Matthew Chantry and Sam Hatfield and Peter Duben and Inna Polichtchouk and Tim Palmer},
  journal= {arXiv preprint arXiv:2101.08195},
  year   = {2021}
}
R2 v1 2026-06-23T22:21:29.385Z