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.

Nonlinearity-free prediction of the growth-rate using convolutional neural networks

Non-Linearity-Free prediction of the growth-rate $fσ_8$ using Convolutional Neural Networks

Theoretical Accuracy in Cosmological Growth Estimation

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

Quantifying Non-parametric Structure of High-redshift Galaxies with Deep Learning

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

The effects of non-linearity on the growth rate constraint from velocity correlation functions

Beware of Fake $ν$s: The Effect of Massive Neutrinos on the Nonlinear Evolution of Cosmic Structure

Distinguishing Coupled Dark Energy Models with Neural Networks

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

Effective cosmic density field reconstruction with convolutional neural network

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

Bayesian deep learning for cosmic volumes with modified gravity

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

Non-parametric Cosmology with Cosmic Shear

Probabilistic Galaxy Field Generation with Diffusion Models

Constraining the growth of perturbations with lensing of supernovae

Using peculiar velocity surveys to constrain the growth rate of structure with the wide-angle effect

Massive $ν$s through the CNN lens: interpreting the field-level neutrino mass information in weak lensing