Abstract:The study of galaxy formation and evolution critically depends on our understanding of the complex photo-chemical processes that govern the evolution and thermodynamics of the interstellar medium (ISM). In a computational sense, resolving the chemistry is among the weightiest tasks in cosmological and astrophysical simulations. Astrophysical simulations can include photo-chemical models that allow for a wide range of densities ($n$), abundances of different species ($n_i/n$) and temperature ($T$), and plausible evolution scenarios of the ISM under the action of a radiation field ($ F $) with different spectral shapes and intensities. The evolution of such a non-equilibrium photo-chemical network relies on implicit, precise, computationally costly, ordinary differential equations (ODE) solvers. Here, we aim to substitute such procedural solvers with fast, pre-trained emulators based on neural operators. We emulated a non-equilibrium chemical network up to H$_2$ formation (9 species, 52 reactions) by adopting the DeepONet formalism, namely: by splitting the ODE solver operator that maps the initial conditions and time evolution into a tensor product of two neural networks (named branch and trunk). We used $ KROME $ to generate a training set, spanning $-2 K < 0$, and adopting an incident radiation field, $ F $ sampled in ten energy bins with a continuity prior. We separately trained the solver for $T$ and each $n_i$ for $ GPUhrs$. Compared with the reference solutions obtained by $ KROME $ for single-zone models, the typical precision obtained is of the order of $10^ $, that is, it is 10 times better when using a training that is 40 times less costly, with respect to previous emulators that only considered a fixed $ F $. DeepONet also performs well for $T$ and $n_i$ outside the range of the training sample. Furthermore, the emulator aptly reproduces the ion and temperature profiles of photo dissociation regions as well; namely, by giving errors that are comparable to the typical difference between various photo-ionization codes. The present model achieves a speed-up of a factor of $128 with respect to stiff ODE solvers. Our neural emulator represents a significant leap forward in the modelling of ISM chemistry, offering a good balance of precision, versatility, and computational efficiency. Nevertheless, further work is required to address the challenges represented by the extrapolation beyond the training time domain and the removal of potential outliers.

As Simple as Possible but No Simpler: Optimizing the Performance of Neural Net Emulators for Galaxy SED Fitting

Neural Network Emulation of Spatio-temporal Data Using Linear and Nonlinear Dimensionality Reduction

SPECULATOR: Emulating stellar population synthesis for fast and accurate galaxy spectra and photometry

Emulating Radiative Transfer with Artificial Neural Networks

Fast emulation of two-point angular statistics for photometric galaxy surveys

A neural network emulator for the Lyman-$α$ 1D flux power spectrum

Validating posteriors obtained by an emulator when jointly-fitting mock data of the global 21-cm signal and high-z galaxy UV luminosity function

Attention-based Neural Network Emulators for Multi-Probe Data Vectors Part I: Forecasting the Growth-Geometry split

A fast neural emulator for interstellar chemistry

Emulators for stellar profiles in binary population modeling

High-accuracy emulators for observables in $Λ$CDM, $N_\mathrm{eff}$, $Σm_ν$, and $w$ cosmologies

Galaxy clustering multi-scale emulation

A GPR-based emulator for semi-numerical reionization code SCRIPT: parameter inference from 21 cm data

On the road to percent accuracy II: calibration of the non-linear matter power spectrum for arbitrary cosmologies

Computer Model Emulation with High-Dimensional Functional Output in Large-Scale Observing System Uncertainty Experiments

Neural network based emulation of galaxy power spectrum covariances -- A reanalysis of BOSS DR12 data

Emulating the interstellar medium chemistry with neural operators

High-accuracy emulators for observables in ΛCDM, N eff, Σ m ν, and w cosmologies

Neural network emulator to constrain the high-$z$ IGM thermal state from Lyman-$α$ forest flux auto-correlation function

Neural Network-based model of galaxy power spectrum: Fast full-shape galaxy power spectrum analysis