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

Abstract

We present a neural network emulator to constrain the thermal parameters of the intergalactic medium (IGM) at $5.4\le z\le 6.0$ using the Lyman-$\alpha$ (Ly$\alpha$) 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 $T_0$, the slope of the temperature$-$density relation $\gamma$, and the mean transmission flux $⟨F⟩$. 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$\alpha$ flux. By examining the credibility contours of the marginalized posteriors for $T_0,\gamma ,\text{and}⟨F⟩$ obtained using the emulator, the statistical reliability of measurements is established through inference on 100 realistic mock data sets of the auto-correlation function.

References in corpus (26)

• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• On the difficulty of training Recurrent Neural Networks
• Evidence of patchy hydrogen reionization from an extreme Ly$\alpha$ trough below redshift six
• Hydrogen reionisation ends by $z=5.3$: Lyman-$\alpha$ optical depth measured by the XQR-30 sample
• Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro
• The mean free path of ionizing photons at 5 < z < 6: evidence for rapid evolution near reionization
• Minimally Parametric Power Spectrum Reconstruction from the Lyman-alpha Forest
• Constraint on neutrino masses from SDSS-III/BOSS Ly$\alpha$ forest and other cosmological probes
• Emulating the CFHTLenS Weak Lensing data: Cosmological Constraints from moments and Minkowski functionals
• The thermal history of the intergalactic medium down to redshift z=1.5: a new curvature measurement
• Self-Consistent Modeling of Reionization in Cosmological Hydrodynamical Simulations
• The DAWES review 10: The impact of deep learning for the analysis of galaxy surveys
• Simulating intergalactic gas for DESI-like small scale Lymanα forest observations
• Machine Learning for Observational Cosmology
• PRIYA: A New Suite of Lyman-alpha Forest Simulations for Cosmology
• A neural network emulator for the Lyman-$\alpha$ 1D flux power spectrum
• Fast, high-fidelity Lyman $\alpha$ forests with convolutional neural networks
• Possible evidence for a large-scale enhancement in the Lyman-$\alpha$ forest power spectrum at redshift $\text{z}\ge 4$
• A Multi-Fidelity Emulator for the Lyman-$\alpha$ Forest Flux Power Spectrum
• Improving IGM temperature constraints using wavelet analysis on high-redshift quasars
• Bayesian error propagation for neural-net based parameter inference
• Deep Forest: Neural Network reconstruction of intergalactic medium temperature
• Forecasting constraints on the high-z IGM thermal state from the Lyman-$\alpha$ forest flux auto-correlation function
• Precisely Measuring the Cosmic Reionization History from IGM Damping Wings Towards Quasars
• Machine learning in parameter estimation of nonlinear systems
• Deep Learning the Intergalactic Medium using Lyman-alpha Forest at $4\le z\le 5$