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.

Enhancing Cosmological Model Selection with Interpretable Machine Learning

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

Distinguishing Coupled Dark Energy Models with Neural Networks

Discriminating among cosmological models by data-driven methods

Cosmological model discrimination with Deep Learning

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

Cosmology with Galaxy Cluster Properties using Machine Learning

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

Bayesian deep learning for cosmic volumes with modified gravity

CoLFI: Cosmological Likelihood-free Inference with Neural Density Estimators

Probing $Λ$CDM through the Weyl potential and machine learning forecasts

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

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

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

A comparative study of cosmological constraints from weak lensing using Convolutional Neural Networks

Constraining Cosmology with Machine Learning and Galaxy Clustering: The CAMELS-SAM Suite

Cosmology with multiple galaxies

Improving Photometric Redshift Estimation for Cosmology with LSST using Bayesian Neural Networks

Estimating the mass of galactic components using machine learning algorithms