MACE: A Machine learning Approach to Chemistry Emulation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Maes, S., De Ceuster, F., Van de Sande, M., Decin, L.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916236647989248
author Maes, S.
De Ceuster, F.
Van de Sande, M.
Decin, L.
author_facet Maes, S.
De Ceuster, F.
Van de Sande, M.
Decin, L.
contents The chemistry of an astrophysical environment is closely coupled to its dynamics, the latter often found to be complex. Hence, to properly model these environments a 3D context is necessary. However, solving chemical kinetics within a 3D hydro simulation is computationally infeasible for a even a modest parameter study. In order to develop a feasible 3D hydro-chemical simulation, the classical chemical approach needs to be replaced by a faster alternative. We present mace, a Machine learning Approach to Chemistry Emulation, as a proof-of-concept work on emulating chemistry in a dynamical environment. Using the context of AGB outflows, we have developed an architecture that combines the use of an autoencoder (to reduce the dimensionality of the chemical network) and a set of latent ordinary differential equations (that are solved to perform the temporal evolution of the reduced features). Training this architecture with an integrated scheme makes it possible to successfully reproduce a full chemical pathway in a dynamical environment. mace outperforms its classical analogue on average by a factor 26. Furthermore, its efficient implementation in PyTorch results in a sub-linear scaling with respect to the number of hydrodynamical simulation particles.
format Preprint
id arxiv_https___arxiv_org_abs_2405_03274
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MACE: A Machine learning Approach to Chemistry Emulation
Maes, S.
De Ceuster, F.
Van de Sande, M.
Decin, L.
Computational Physics
Astrophysics of Galaxies
Instrumentation and Methods for Astrophysics
Solar and Stellar Astrophysics
The chemistry of an astrophysical environment is closely coupled to its dynamics, the latter often found to be complex. Hence, to properly model these environments a 3D context is necessary. However, solving chemical kinetics within a 3D hydro simulation is computationally infeasible for a even a modest parameter study. In order to develop a feasible 3D hydro-chemical simulation, the classical chemical approach needs to be replaced by a faster alternative. We present mace, a Machine learning Approach to Chemistry Emulation, as a proof-of-concept work on emulating chemistry in a dynamical environment. Using the context of AGB outflows, we have developed an architecture that combines the use of an autoencoder (to reduce the dimensionality of the chemical network) and a set of latent ordinary differential equations (that are solved to perform the temporal evolution of the reduced features). Training this architecture with an integrated scheme makes it possible to successfully reproduce a full chemical pathway in a dynamical environment. mace outperforms its classical analogue on average by a factor 26. Furthermore, its efficient implementation in PyTorch results in a sub-linear scaling with respect to the number of hydrodynamical simulation particles.
title MACE: A Machine learning Approach to Chemistry Emulation
topic Computational Physics
Astrophysics of Galaxies
Instrumentation and Methods for Astrophysics
Solar and Stellar Astrophysics
url https://arxiv.org/abs/2405.03274