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.

Predicting the Thermal Sunyaev-Zel'dovich Field using Modular and Equivariant Set-Based Neural Networks

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

Attention-Based Neural Network Emulators for Multi-Probe Data Vectors Part II: Assessing Tension Metrics

Optimizing NILC Extractions of the Thermal Sunyaev-Zeldovich Effect with Deep Learning

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

Learning to Predict the Cosmological Structure Formation.

Neural network reconstruction of scalar-tensor cosmology

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

Optimizing NILC Extractions of the Thermal Sunyaev–Zel'Dovich Effect with Deep Learning

Neural network prediction of model parameters for strong lensing samples from Hyper Suprime-Cam Survey

Redshift Prediction with Images for Cosmology Using a Bayesian Convolutional Neural Network with Conformal Predictions

Bayesian deep learning for cosmic volumes with modified gravity

Improved Modeling of the Kinematic Sunyaev-Zel'dovich Projected-Fields signal and its Cosmological Dependence

Photometric Redshifts for Cosmology: Improving Accuracy and Uncertainty Estimates Using Bayesian Neural Networks

Field-level simulation-based inference of galaxy clustering with convolutional neural networks

Cosmological model discrimination with Deep Learning

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

A Tale of Two Fields: Neural Network-Enhanced non-Gaussianity Search with Halos

Simple lessons from complex learning: what a neural network model learns about cosmic structure formation