Using Neural Emulators and Hamiltonian Monte Carlo to constrain the Epoch of Reionization's History with the Ly$α$ Forest Power Spectrum
Authors:
Diego González-Hernández,
Caitlin Doughty,
Molly Wolfson,
Joseph F. Hennawi,
Zhenyu Jin
Abstract:
The Lyman-alpha (Ly$α$) forest at $z \sim 5$ offers a primary probe to constrain the history of the Epoch of Reionization (EoR), retaining thermal and ionization signatures imprinted by the reionization process. In this work, we present a new inference framework based on JAX that combines forward-modeled Ly$α$ forest observables with differentiable neural emulators and Hamiltonian Monte Carlo (HMC…
▽ More
The Lyman-alpha (Ly$α$) forest at $z \sim 5$ offers a primary probe to constrain the history of the Epoch of Reionization (EoR), retaining thermal and ionization signatures imprinted by the reionization process. In this work, we present a new inference framework based on JAX that combines forward-modeled Ly$α$ forest observables with differentiable neural emulators and Hamiltonian Monte Carlo (HMC). We construct a dataset of 501 low-resolution simulations generated with user-defined reionization histories and compute a set of 1D Ly$α$ power spectra and model-dependent covariance matrices. We then train two independent neural emulators that achieve sub-percent errors across relevant scales and combine them with HMC to efficiently perform parameter estimation. We validate this framework by applying it to a suite of mock observations, demonstrating that the true parameters are reliably recovered. While this work is limited by the low resolution of the simulations used, our results highlight the potential of this method for inferring the reionization history from high-redshift Ly$α$ forest measurements. Future improvements in our reionization models will further enhance its ability to extract constraints from observational datasets.
△ Less
Submitted 16 September, 2025;
originally announced September 2025.
Neural network emulator to constrain the high-$z$ IGM thermal state from Lyman-$α$ forest flux auto-correlation function
Authors:
Zhenyu Jin,
Molly Wolfson,
Joseph F. Hennawi,
Diego González-Hernández
Abstract:
We present a neural network emulator to constrain the thermal parameters of the intergalactic medium (IGM) at $\displaystyle{5.4}\le{z}\le{6.0}$ using the Lyman-$\displaystyleα$ (Ly$\displaystyleα$) forest flux auto-correlation function. Our auto-differentiable JAX-based framework accelerates the surrogate model generation process using approximately 100 sparsely sampled Nyx hydrodynamical simulat…
▽ More
We present a neural network emulator to constrain the thermal parameters of the intergalactic medium (IGM) at $\displaystyle{5.4}\le{z}\le{6.0}$ using the Lyman-$\displaystyleα$ (Ly$\displaystyleα$) forest flux auto-correlation function. Our auto-differentiable JAX-based framework accelerates the surrogate model generation process using approximately 100 sparsely sampled Nyx hydrodynamical simulations with varying combinations of thermal parameters, i.e., the temperature at mean density $\displaystyle{T}_{0}$, the slope of the temperature$\displaystyle-$density relation $\displaystyleγ$, and the mean transmission flux $\displaystyle{\langle}{F}{\rangle}$. We show that this emulator has a typical accuracy of 1.0% across the specified redshift range. Bayesian inference of the IGM thermal parameters, incorporating emulator uncertainty propagation, is further expedited using NumPyro Hamiltonian Monte Carlo. We compare both the inference results and computational cost of our framework with the traditional nearest-neighbor interpolation approach applied to the same set of mock Ly$α$ flux. By examining the credibility contours of the marginalized posteriors for $\displaystyle{T}_{0},γ,\text{and}{\langle}{F}{\rangle}$ obtained using the emulator, the statistical reliability of measurements is established through inference on 100 realistic mock data sets of the auto-correlation function.
△ Less
Submitted 20 December, 2024; v1 submitted 8 October, 2024;
originally announced October 2024.
