Submitted as: development and technical paper16 Feb 2021
Submitted as: development and technical paper | 16 Feb 2021
Review status: this preprint is currently under review for the journal GMD.
A machine learning-guided adaptive algorithm to reduce the computational cost of
atmospheric chemistry in Earth System models: application to GEOS-Chem
versions 12.0.0 and 12.9.1
Lu Shen1,Daniel J. Jacob1,Mauricio Santillana2,3,Kelvin Bates1,Jiawei Zhuang1,and Wei Chen4Lu Shen et al.Lu Shen1,Daniel J. Jacob1,Mauricio Santillana2,3,Kelvin Bates1,Jiawei Zhuang1,and Wei Chen4
Received: 15 Dec 2020 – Accepted for review: 15 Feb 2021 – Discussion started: 16 Feb 2021
Abstract. Atmospheric composition plays a crucial role in determining the evolution of the atmosphere, but the high computational cost has been the major barrier to include atmospheric chemistry into Earth system models. Here we present an adaptive and efficient algorithm that can remove this barrier. Our approach is inspired by unsupervised machine learning clustering techniques and traditional asymptotic analysis ideas. We first partition species into 13 blocks, using a novel machine learning approach that analyzes the species network structures and their production and loss rates. Building on these blocks, we pre-select 20 submechanisms, as defined by unique assemblages of the species blocks, and then pick locally on the fly which submechanism to use based on local chemical conditions. In each submechanism, we isolate slow species and unimportant reactions from the coupled system. Application to a global 3-D model shows that we can cut the computational costs of the chemical integration by 50 % with accuracy losses smaller than 1 % that do not propagate in time. Tests show that this algorithm is highly chemically coherent making it easily portable to new models without compromising its performance. Our algorithm will significantly ease the computational bottleneck and will facilitate the development of next generation of earth system models.
Replication Data for: A machine learning-guided and accurate algorithm to halve the computational cost of atmospheric chemistry in Earth System modelsShen et al. https://doi.org/10.7910/DVN/KASQOC