ML4WX

Hybrid physics-ML & parameterization

Machine learning emulation of gravity wave drag in numerical weather forecasting

Matthew Chantry, Sam Hatfield, Peter Duben, Inna Polichtchouk, Tim Palmer

Journal of Advances in Modeling Earth Systems · · ECMWF, University of Oxford

PaperPDF

In one sentence. Neural emulators of the non-orographic gravity wave drag scheme run stably to seasonal timescales, improve medium-range forecasts and run ten times faster on GPUs than the CPU scheme.

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.

Labels

Topic
Hybrid physics-ML & parameterization, Global medium-range NWP
Type
New model or method
Method
Other
Time scale
3–15 days, Seasonal to annual
Domain
Global
Data
NWP analysis or forecasts (IFS, GFS, ...)
Status
Pre-operational / experimental
Identifiers
arXiv:2101.08195 DOI:10.1029/2021ms002477 OpenAlex:W3123510158
Added
2026-10-22

Labels and summary are generated by Claude and reviewed weekly. Wrong? Edit this entry.