Online Test of a Neural Network Deep Convection Parameterization in ARP-GEM1

Blanka Balogh a CNRM, Université de Toulouse, Météo-France, CNRS, Toulouse, France

Search for other papers by Blanka Balogh in
Current site
Google Scholar
PubMed
Close
,
David Saint-Martin a CNRM, Université de Toulouse, Météo-France, CNRS, Toulouse, France

Search for other papers by David Saint-Martin in
Current site
Google Scholar
PubMed
Close
, and
Olivier Geoffroy a CNRM, Université de Toulouse, Météo-France, CNRS, Toulouse, France

Search for other papers by Olivier Geoffroy in
Current site
Google Scholar
PubMed
Close
Open access

Abstract

In this study, we integrate a neural network-based parameterization into the global atmospheric model ARP-GEM1 using the Python interface of the OASIS coupler. This setup enables the exchange of fields between the Fortran-based ARP-GEM1 model and a Python component implementing the neural network inference. The Python component was deployed on a separate partition from the general circulation model, using GPUs. As a proof-of-concept, we trained a neural network to emulate ARP-GEM1’s deep convection parameterization. Leveraging the flexible Fortran/Python interface, we successfully replaced ARP-GEM1’s deep convection scheme with the neural network emulator. To evaluate its online performance, we realized a 30-year ARP-GEM1 simulation using the neural network for deep convection. The evaluation of the averaged fields showed good agreement with the output of an ARP-GEM1 simulation using the physics-based deep convection scheme.

© 2025 American Meteorological Society. This is an Author Accepted Manuscript distributed under the terms of the default AMS reuse license. For information regarding reuse and general copyright information, consult the AMS Copyright Policy (www.ametsoc.org/PUBSReuseLicenses).

Corresponding author: Blanka Balogh, blanka.balogh@meteo.fr

Abstract

In this study, we integrate a neural network-based parameterization into the global atmospheric model ARP-GEM1 using the Python interface of the OASIS coupler. This setup enables the exchange of fields between the Fortran-based ARP-GEM1 model and a Python component implementing the neural network inference. The Python component was deployed on a separate partition from the general circulation model, using GPUs. As a proof-of-concept, we trained a neural network to emulate ARP-GEM1’s deep convection parameterization. Leveraging the flexible Fortran/Python interface, we successfully replaced ARP-GEM1’s deep convection scheme with the neural network emulator. To evaluate its online performance, we realized a 30-year ARP-GEM1 simulation using the neural network for deep convection. The evaluation of the averaged fields showed good agreement with the output of an ARP-GEM1 simulation using the physics-based deep convection scheme.

© 2025 American Meteorological Society. This is an Author Accepted Manuscript distributed under the terms of the default AMS reuse license. For information regarding reuse and general copyright information, consult the AMS Copyright Policy (www.ametsoc.org/PUBSReuseLicenses).

Corresponding author: Blanka Balogh, blanka.balogh@meteo.fr
Save