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.

Reconstruction of Continuous Cosmological Fields from Discrete Tracers with Graph Neural Networks

Continuous Volumetric Convolution Network with Self-Learning Kernels for Point Clouds

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

Effective cosmic density field reconstruction with convolutional neural network

Learning the galaxy-environment connection with graph neural networks

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

Neural network reconstruction of scalar-tensor cosmology

Velocity recostruction with graph neural networks

Graph Neural Networks for Particle Reconstruction in High Energy Physics detectors

Deep Learning Cosmic Ray Transport from Density Maps of Simulated, Turbulent Gas

Cosmic Velocity Field Reconstruction Using AI

Using Cartesian slice plots of a cosmological simulation as input of a convolutional neural network

Weak-lensing Mass Reconstruction of Galaxy Clusters with a Convolutional Neural Network -- II: Application to Next-Generation Wide-Field Surveys

Neutrino Reconstruction in TRIDENT Based on Graph Neural Network

Using conditional GANs for convergence map reconstruction with uncertainties

Estimating Dark Matter Halo Masses in Simulated Galaxy Clusters with Graph Neural Networks

Baryon acoustic oscillations reconstruction using convolutional neural networks

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

Instance Segmentation GNNs for One-Shot Conformal Tracking at the LHC

Flow reconstruction in time-varying geometries using graph neural networks

Cosmic-CoNN: A Cosmic Ray Detection Deep-Learning Framework, Dataset, and Toolkit