Evaluating the prediction of subgrid eddy kinetic energy from a machine learning model
Abstract
Global climate model resolution is still too coarse to fully resolve mesoscale processes, therefore subgrid closures are required to approximate the feedback of eddies onto the resolved flow. With high-resolution datasets now readily available, there has been a rapid growth in the use of machine learning to develop new eddy closures that could help reduce model biases and uncertainty. In this work we employ an offline learning approach with a convolutional neural network to predict subgrid eddy kinetic energy (KEe), laying the initial groundwork for a machine-learned scale-aware GEOMETRIC eddy closure for the UK's Earth System Model. We address the question of whether global parameterizations are needed or if regional closures are sufficient by evaluating a number of neural network models through region and feature ablation experiments. ML parameterization models generalize to unseen regions when their training data share similar feature and target distributions with those regions. Models that fail to generalize incur distributional shift, or employ different filtering scales for the true eddy kinetic energy field. When holding out features, there is improvement among the generalizable ML models, with some feature spaces leading to improvements in some regions but degrading performance in others. Generally, power spectral density of KEe tend to diverge at small scales, though some feature spaces are able to minimize this divergence. Overall, the best performing model trained on all regions achieves time-mean and anomaly R2 scores of R2≈ 0.95 and R2≈ 0.8 − 0.9, respectively.
Related
- Butterfly Effect and the Kinetic Energy Cascade in Probabilistic Machine Learning Weather Prediction Models
- Integrated experimental and machine learning-driven multi-objective optimization for phenol removal via silica aerogel/ZSM-5 nanocomposite: Isotherm and kinetic analysis, thermodynamic insights and pareto-based process design
- Machine-learning prediction of freezing temperature in Pb-contaminated soils remediated with biochar
- FastNet: Improving the Physical Consistency of Machine Learning Weather Prediction Models through Loss Function Design
- AIFS-DOP: End-to-End Medium-Range Weather Prediction from Observations Alone with Machine Learning
BibTeX
@misc{wilder2026evaluating,
title = {{Evaluating the prediction of subgrid eddy kinetic energy from a machine learning model}},
author = {Thomas Wilder and Ranjini Swaminathan and David Storkey and Till Kuhlbrodt},
howpublished = {arXiv},
year = {2026},
month = nov,
doi = {10.48550/arxiv.2611.06539},
eprint = {2611.06539},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2611.06539},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - Evaluating the prediction of subgrid eddy kinetic energy from a machine learning model
AU - Thomas Wilder
AU - Ranjini Swaminathan
AU - David Storkey
AU - Till Kuhlbrodt
PY - 2026
DA - 2026/11/04
PB - arXiv
DO - 10.48550/arxiv.2611.06539
UR - https://arxiv.org/abs/2611.06539
AB - Global climate model resolution is still too coarse to fully resolve mesoscale processes, therefore subgrid closures are required to approximate the feedback of eddies onto the resolved flow. With high-resolution datasets now readily available, there has been a rapid growth in the use of machine learning to develop new eddy closures that could help reduce model biases and uncertainty. In this work we employ an offline learning approach with a convolutional neural network to predict subgrid eddy kinetic energy (KE_e), laying the initial groundwork for a machine-learned scale-aware GEOMETRIC eddy closure for the UK's Earth System Model. We address the question of whether global parameterizations are needed or if regional closures are sufficient by evaluating a number of neural network models through region and feature ablation experiments. ML parameterization models generalize to unseen regions when their training data share similar feature and target distributions with those regions. Models that fail to generalize incur distributional shift, or employ different filtering scales for the true eddy kinetic energy field. When holding out features, there is improvement among the generalizable ML models, with some feature spaces leading to improvements in some regions but degrading performance in others. Generally, power spectral density of KE_e tend to diverge at small scales, though some feature spaces are able to minimize this divergence. Overall, the best performing model trained on all regions achieves time-mean and anomaly R2 scores of R2≈ 0.95 and R2≈ 0.8 − 0.9, respectively.
ER -