Abstract:The new generation of galaxy surveys will provide unprecedented data allowing us to test gravity at cosmological scales. A robust cosmological analysis of the large-scale structure demands exploiting the nonlinear information encoded in the cosmic web. Machine Learning techniques provide such tools, however, do not provide a priori assessment of uncertainties. This study aims at extracting cosmological parameters from modified gravity (MG) simulations through deep neural networks endowed with uncertainty estimations. We implement Bayesian neural networks (BNNs) with an enriched approximate posterior distribution considering two cases: one with a single Bayesian last layer (BLL), and another one with Bayesian layers at all levels (FullB). We train both BNNs with real-space density fields and power-spectra from a suite of 2000 dark matter only particle mesh $N$-body simulations including modified gravity models relying on MG-PICOLA covering 256 $h^{-1}$ Mpc side cubical volumes with 128$^3$ particles. BNNs excel in accurately predicting parameters for $\Omega_m$ and $\sigma_8$ and their respective correlation with the MG parameter. We find out that BNNs yield well-calibrated uncertainty estimates overcoming the over- and under-estimation issues in traditional neural networks. We observe that the presence of MG parameter leads to a significant degeneracy with $\sigma_8$ being one of the possible explanations of the poor MG predictions. Ignoring MG, we obtain a deviation of the relative errors in $\Omega_m$ and $\sigma_8$ by at least $30\%$. Moreover, we report consistent results from the density field and power spectra analysis, and comparable results between BLL and FullB experiments which permits us to save computing time by a factor of two. This work contributes in setting the path to extract cosmological parameters from complete small cosmic volumes towards the highly nonlinear regime.

Neural Networks for cosmological model selection and feature importance using Cosmic Microwave Background data

Enhancing Cosmological Model Selection with Interpretable Machine Learning

Cosmological constraints from low redshift 21 cm intensity mapping with machine learning

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

Distinguishing Coupled Dark Energy Models with Neural Networks

Discriminating among cosmological models by data-driven methods

Classifying Modified Gravity and Dark Energy Theories with Bayesian Neural Networks: Massive Neutrinos, Baryonic Feedback, and the Theoretical Error

Not Hydro: Using Neural Networks to estimate galaxy properties on a Dark-Matter-Only simulation

Cosmological model discrimination with Deep Learning

ParamANN: A Neural Network to Estimate Cosmological Parameters for $Λ$CDM Universe Using Hubble Measurements

Recovering the E and B-mode CMB polarization at sub-degree scales with neural networks

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

Classification of radio backgrounds at cosmic dawn and 21-cm signal confirmation using neural networks

Inferring Cosmological Parameters on SDSS via Domain-Generalized Neural Networks and Lightcone Simulations

Machine learning cosmic inflation

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

ParamANN: a neural network to estimate cosmological parameters for ΛCDM universe using hubble measurements

Bayesian deep learning for cosmic volumes with modified gravity

The cosmological analysis of X-ray cluster surveys: VI. Inference based on analytically simulated observable diagrams

Inference of the optical depth to reionization $τ$ from $\textit{Planck}$ CMB maps with convolutional neural networks

Constraining the polarisation flux density and angle of point sources by training a convolutional neural network