Beyond the Training Data: Confidence-Guided Mixing of Parameterizations in a Hybrid AI-Climate Model

Beyond the Training Data: Confidence-Guided Mixing of Parameterizations in a Hybrid AI-Climate Model*By Helge Heuer, Tom Beucler, Mierk Schwabe, Julien Savre, Manuel Schlund, and Veronika Eyring** This paper presents a **successful proof-of-concept for transferring a machine learning (ML) convection parameterization**—trained on the ClimSim dataset—to the ICON-A climate model. The resulting hybrid ML-physics model achieved stable and accurate simulations in long-term AMIP-style runs lasting at least 20 years.* A core innovation is the **confidence-guided mixing scheme**, which allows the Neural Network (NN) to predict its own error. When the NN's predicted confidence is low (e.g., in moist, unstable regimes or high-variability areas), its prediction is mixed with the conventional Tiedtke convection scheme. This mechanism improves reliability, prevents unphysical outputs by detecting potential extrapolation beyond the training domain, and makes the hybrid model tunable against observations.* The scheme's robustness and accuracy were further enhanced through the **use of a physics-informed loss function**—which encourages adherence to conservation laws like enthalpy and mass—and **noise-augmented training**. These techniques mitigate stability issues commonly faced by ML parameterizations and significantly improve physical consistency compared to purely data-driven models.* In evaluation against observational data, several hybrid configurations **outperformed the default Tiedtke scheme**, demonstrating improved precipitation statistics and showing a better representation of global climate variables. The confidence-guided approach demonstrated a fundamental change in the model's behavior, with the ML component contributing approximately 67% of the convective tendencies on average.