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 (82)

• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• emcee: The MCMC Hammer
• Decoupled Weight Decay Regularization
• On the difficulty of training Recurrent Neural Networks
• The No-U-Turn Sampler: Adaptively Setting Path Lengths in Hamiltonian Monte Carlo
• Planck 2018 results. I. Overview and the cosmological legacy of Planck
• Constraining the Evolution of the Ionizing Background and the Epoch of Reionization with z ~ 6 Quasars II: A Sample of 19 Quasars
• Activation Functions: Comparison of trends in Practice and Research for Deep Learning
• Equation of State of the Photoionized Intergalactic Medium
• New Constraints on the free-streaming of warm dark matter from intermediate and small scale Lyman-$\alpha$ forest data
• The Lyman-alpha Forest Power Spectrum from the Sloan Digital Sky Survey
• Evidence of patchy hydrogen reionization from an extreme Ly$\alpha$ trough below redshift six
• Probing the Universe with the Lyman-alpha Forest: I. Hydrodynamics of the Low Density IGM
• The thermal history of the intergalactic medium
• Detection of Extended He II Reionization in the Temperature Evolution of the Intergalactic Medium
• Hydrogen reionisation ends by $z=5.3$: Lyman-$\alpha$ optical depth measured by the XQR-30 sample
• Nyx: A Massively Parallel AMR Code for Computational Cosmology
• Large Lyman-alpha opacity fluctuations and low CMB tau in models of late reionization with large islands of neutral hydrogen extending to z<5.5
• Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro
• The Evolution of the Effective Equation of State of the IGM
• The Coyote Universe II: Cosmological Models and Precision Emulation of the Nonlinear Matter Power Spectrum
• The one-dimensional Ly-alpha forest power spectrum from BOSS
• New constraints on Lyman-α opacity with a sample of 62 quasars at z > 5.7
• Constraints on neutrino masses from Lyman-alpha forest power spectrum with BOSS and XQ-100
• Cosmological and astrophysical parameters from the SDSS flux power spectrum and hydrodynamical simulations of the Lyman-alpha forest
• A Measurement of the Temperature-Density Relation in the Intergalactic Medium Using a New Lyman-alpha Absorption Line Fitting Method
• New Constraints on IGM Thermal Evolution from the Lyα Forest Power Spectrum
• Cosmology Constraints from the Weak Lensing Peak Counts and the Power Spectrum in CFHTLenS
• The mean free path of ionizing photons at 5 < z < 6: evidence for rapid evolution near reionization
• A Measurement of Small Scale Structure in the 2.2 < z < 4.2 Lyman-alpha Forest
• Constraints from the Lyman alpha forest power spectrum
• The Opacity of the Intergalactic Medium Measured Along Quasar Sightlines at $z\sim 6$
• Revealing reionization with the thermal history of the intergalactic medium: new constraints from the Lyman-$\alpha$ flux power spectrum
• Minimally Parametric Power Spectrum Reconstruction from the Lyman-alpha Forest
• The Aemulus Project III: Emulation of the Galaxy Correlation Function
• Constraint on neutrino masses from SDSS-III/BOSS Ly$\alpha$ forest and other cosmological probes
• Probing the thermal state of the intergalactic medium at $z>5$ with the transmission spikes in high-resolution Ly$\alpha$ forest spectra
• Measurements of the z ~ 6 Intergalactic Medium Optical Depth and Transmission Spikes Using a New z > 6.3 Quasar Sample
• The Aemulus Project II: Emulating the Halo Mass Function
• 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
• Cosmic Emulation: Fast Predictions for the Galaxy Power Spectrum
• A consistent and robust measurement of the thermal state of the IGM at $2\le z\le 4$ from a large sample of Ly$\alpha$ forest spectra: Evidence for late and rapid HeII reionization
• The DAWES review 10: The impact of deep learning for the analysis of galaxy surveys
• The Temperature-Density Relation in the Intergalactic Medium at Redshift <z>=2.4
• A consistent determination of the temperature of the intergalactic medium at redshift z=2.4
• Models of the Thermal Evolution of the Intergalactic Medium After Reionization
• Cosmological and astrophysical constraints from the Lyman-alpha forest flux probability distribution function
• On the intergalactic temperature-density relation
• Characterizing the Pressure Smoothing Scale of the Intergalactic Medium
• The intergalactic medium thermal history at redshift z=1.7--3.2 from the Lyman alpha forest: a comparison of measurements using wavelets and the flux distribution
• Temperature fluctuations in the intergalactic medium
• A New Measurement of the Temperature Density Relation of the IGM From Voigt Profile Fitting
• The b Distribution of the Lya Forest: Probing Cosmology and the Intergalactic Medium
• Heating of the Intergalactic Medium by Hydrogen Reionization
• Bayesian emulator optimisation for cosmology: application to the Lyman-alpha forest
• Probing the thermal history of the Intergalactic Medium with Lyman-alpha absorption lines
• An Emulator for the Lyman-alpha Forest
• Evaluating machine learning techniques for predicting power spectra from reionization simulations
• A New Method to Directly Measure the Jeans Scale of the Intergalactic Medium Using Close Quasar Pairs
• A New Method to Measure the Post-Reionization Ionizing Background from the Joint Distribution of Lyman-$\alpha$ and Lyman-$\beta$ Forest Transmission
• Measurement of the small-scale structure of the intergalactic medium using close quasar pairs
• The Signatures of Large-scale Temperature and Intensity Fluctuations in the Lyman-alpha Forest
• Inhomogeneous Reionization Models in Cosmological Hydrodynamical Simulations
• An emulator for the Lyman-$\alpha$ forest in beyond-$\Lambda$CDM cosmologies
• General framework for cosmological dark matter bounds using $N$-body simulations
• 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
• Deep Forest: Neural Network reconstruction of the Lyman-alpha forest
• 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$