Abstract:We propose a lightweight deep convolutional neural network (lCNN) to estimate cosmological parameters from simulated three-dimensional dark matter (DM) halo distributions and associated statistics. The training dataset comprises 2000 realizations of a cubic box with a side length of 1000 $h^{-1}{\rm Mpc}$, and interpolated over a cubic grid of $300^3$ voxels, with each simulation produced using $512^3$ DM particles and $512^3$ neutrinos. Under the flat $\Lambda$CDM model, simulations vary standard six cosmological parameters including $\Omega_m$, $\Omega_b$, $h$, $n_s$, $\sigma_8$, $w$, along with the neutrino mass sum, $M_\nu$. We find that: 1) within the framework of lCNN, extracting large-scale structure information is more efficient from the halo density field compared to relying on the statistical quantities including the power spectrum, the two-point correlation function, and the coefficients from wavelet scattering transform; 2) combining the halo density field with its Fourier transformed counterpart enhances predictions, while augmenting the training dataset with measured statistics further improves performance; 3) achieving high accuracy in inferring $\Omega_m$, $h$, and $\sigma_8$ by the neural network model, while being inefficient in predicting $\Omega_b$, { $n_s$}, $M_\nu$ and $w$; 4) { compared to the simple fully connected network trained with three statistical quantities, our CNN yields statistically reduced errors, showing improvements of approximately 23\% for $\Omega_m$, 11\% for $h$, 8\% for $n_s$, and 21\% for $\sigma_8$. Additionally, in comparison with the likelihood-based analysis on $P(k)$ data, our CNN provides much tighter constraints on parameters, especially on $\Omega_m$ and $\sigma_8$.} Our study emphasizes this lCNN-based novel approach in extracting large-scale structure information and estimating cosmological parameters.

Deep learning from 21-cm tomography of the Cosmic Dawn and Reionization

Inferring Astrophysics and Dark Matter Properties from 21cm Tomography using Deep Learning

An Observationally Driven Multifield Approach for Probing the Circum-Galactic Medium with Convolutional Neural Networks

Reionisation time fields reconstruction from 21 cm signal maps

AI-assisted Reconstruction of Cosmic Velocity Field from Redshift-Space Spatial Distribution of Halos

Emulation of the Cosmic Dawn 21-cm Power Spectrum and Classification of Excess Radio Models Using an Artificial Neural Network

How Accurately Can 21 Cm Tomography Constrain Cosmology?

Deep learning approach for identification of HII regions during reionization in 21-cm observations -- III. image recovery

Constraining the X-ray heating and reionization using 21-cm power spectra with Marginal Neural Ratio Estimation

A Generative Modeling Approach to Reconstructing 21-cm Tomographic Data

Inferring astrophysical parameters using the 2D cylindrical power spectrum from reionization

Deep learning for cosmological parameter inference from a dark matter halo density field

Simulation-based Inference of Reionization Parameters from 3D Tomographic 21 cm Light-cone Images -- II: Application of Solid Harmonic Wavelet Scattering Transform

Inferring astrophysical parameters using the 2D cylindrical power spectrum from reionisation

The Universe is worth $64^3$ pixels: Convolution Neural Network and Vision Transformers for Cosmology

Cosmological model discrimination with Deep Learning

Inferring the density, spin-temperature and neutral-fraction fields of HI from its 21-cm brightness temperature field using machine learning

Machine Learning Constraints on Dark Matter Annihilation during the Epoch of Reionization: A CNN Analysis of the 21-cm Signal

Machine-learning recovery of foreground wedge-removed 21-cm light cones for high-$z$ galaxy mapping

Machine-learning recovery of foreground wedge-removed 21-cm light cones for high- z galaxy mapping

Deep learning approach for identification of H ii regions during reionization in 21-cm observations – II. Foreground contamination