Origin & Evolution of Life

Bridging Machine Learning And Cosmological Simulations: Using Neural Operators To Emulate Chemical Evolution

By Keith Cowing
Status Report
astro-ph.IM
March 17, 2025
Filed under , , , , , ,
Bridging Machine Learning And Cosmological Simulations: Using Neural Operators To Emulate Chemical Evolution
The computational expense of solving non-equilibrium chemistry equations in astrophysical simulations poses a significant challenge, particularly in high-resolution, large-scale cosmological models. — astro-ph.IM)

In this work, we explore the potential of machine learning, specifically Neural Operators, to emulate the Grackle chemistry solver, which is widely used in cosmological hydrodynamical simulations.

Neural Operators offer a mesh-free, data-driven approach to approximate solutions to coupled ordinary differential equations governing chemical evolution, gas cooling, and heating. We construct and train multiple Neural Operator architectures (DeepONet variants) using a dataset derived from cosmological simulations to optimize accuracy and efficiency. Our results demonstrate that the trained models accurately reproduce Grackle’s outputs with an average error of less than 0.6 dex in most cases, though deviations increase in highly dynamic chemical environments.

Compared to Grackle, the machine learning models provide computational speedups of up to a factor of six in large-scale simulations, highlighting their potential for reducing computational bottlenecks in astrophysical modeling. However, challenges remain, particularly in iterative applications where accumulated errors can lead to numerical instability.

Additionally, the performance of these machine learning models is constrained by their need for well-represented training datasets and the limited extrapolation capabilities of deep learning methods.

While promising, further development is required for Neural Operator-based emulators to be fully integrated into astrophysical simulations. Future work should focus on improving stability over iterative timesteps and optimizing implementations for hardware acceleration. This study provides an initial step toward the broader adoption of machine learning approaches in astrophysical chemistry solvers.

Pelle van de Bor, John Brennan, John A. Regan, Jonathan Mackey

Comments: 11 pages, 9 figures
Subjects: Instrumentation and Methods for Astrophysics (astro-ph.IM); Cosmology and Nongalactic Astrophysics (astro-ph.CO); Computational Physics (physics.comp-ph)
Cite as: arXiv:2503.10736 [astro-ph.IM] (or arXiv:2503.10736v1 [astro-ph.IM] for this version)
https://doi.org/10.48550/arXiv.2503.10736
Focus to learn more
Submission history
From: Pelle Van De Bor
[v1] Thu, 13 Mar 2025 15:47:04 UTC (1,864 KB)
https://arxiv.org/abs/2503.10736

Astrobiology,

Explorers Club Fellow, ex-NASA Space Station Payload manager/space biologist, Away Teams, Journalist, Lapsed climber, Synaesthete, Na’Vi-Jedi-Freman-Buddhist-mix, ASL, Devon Island and Everest Base Camp veteran, (he/him) 🖖🏻