climt-paraformer: Stable Emulation of Convective Parameterization using a Temporal Memory-aware Transformer

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Shuochen, Yadav, Nishant, Monteiro, Joy Merwin, Ganguly, Auroop R.
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911616555024384
author Wang, Shuochen
Yadav, Nishant
Monteiro, Joy Merwin
Ganguly, Auroop R.
author_facet Wang, Shuochen
Yadav, Nishant
Monteiro, Joy Merwin
Ganguly, Auroop R.
contents Accurate representation of moist convective sub-grid-scale processes remains a major challenge in global climate models, as traditional parameterization schemes are both computationally expensive and difficult to scale. Neural network (NN) emulators offer a promising alternative by learning efficient mappings between atmospheric states and convective tendencies while retaining fidelity to the underlying physics. However, most existing NN-based parameterizations are memory-less and rely only on instantaneous inputs, even though convection evolves over time and depends on prior atmospheric states. Recent studies have begun to incorporate convective memory, but they often treat past states as independent features rather than modeling temporal dependencies explicitly. In this work, we develop a temporal memory-aware Transformer emulator for the Emanuel convective parameterization and evaluate it in a single-column climate model (SCM) under both offline and online configurations. The Transformer captures temporal correlations and nonlinear interactions across consecutive atmospheric states. Compared with baseline emulators, including a memory-less multilayer perceptron and a recurrent long short-term memory model, the Transformer achieves lower offline errors. Sensitivity analysis indicates that a memory length of approximately 100 minutes yields the best performance, whereas longer memory degrades performance. We further test the emulator in long-term coupled simulations and show that it remains stable over 10 years. Overall, this study demonstrates the importance of explicit temporal modeling for NN-based parameterizations.
format Preprint
id arxiv_https___arxiv_org_abs_2604_21085
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle climt-paraformer: Stable Emulation of Convective Parameterization using a Temporal Memory-aware Transformer
Wang, Shuochen
Yadav, Nishant
Monteiro, Joy Merwin
Ganguly, Auroop R.
Atmospheric and Oceanic Physics
Machine Learning
Accurate representation of moist convective sub-grid-scale processes remains a major challenge in global climate models, as traditional parameterization schemes are both computationally expensive and difficult to scale. Neural network (NN) emulators offer a promising alternative by learning efficient mappings between atmospheric states and convective tendencies while retaining fidelity to the underlying physics. However, most existing NN-based parameterizations are memory-less and rely only on instantaneous inputs, even though convection evolves over time and depends on prior atmospheric states. Recent studies have begun to incorporate convective memory, but they often treat past states as independent features rather than modeling temporal dependencies explicitly. In this work, we develop a temporal memory-aware Transformer emulator for the Emanuel convective parameterization and evaluate it in a single-column climate model (SCM) under both offline and online configurations. The Transformer captures temporal correlations and nonlinear interactions across consecutive atmospheric states. Compared with baseline emulators, including a memory-less multilayer perceptron and a recurrent long short-term memory model, the Transformer achieves lower offline errors. Sensitivity analysis indicates that a memory length of approximately 100 minutes yields the best performance, whereas longer memory degrades performance. We further test the emulator in long-term coupled simulations and show that it remains stable over 10 years. Overall, this study demonstrates the importance of explicit temporal modeling for NN-based parameterizations.
title climt-paraformer: Stable Emulation of Convective Parameterization using a Temporal Memory-aware Transformer
topic Atmospheric and Oceanic Physics
Machine Learning
url https://arxiv.org/abs/2604.21085