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.

Learning from galactic rotation curves: a neural network approach

Reconstructing rotation curves with artificial neural networks

Galaxy Rotation Curve Fitting Using Machine Learning Tools

Machine Learning the Dark Matter Halo Mass of Milky Way-Like Systems

Bayesian deep learning for cosmic volumes with modified gravity

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

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

GalaxyNet: connecting galaxies and dark matter haloes with deep neural networks and reinforcement learning in large volumes

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

Evaluating the galaxy formation histories predicted by a neural network in pure dark matter simulations

Rotational curves decomposition of spiral galaxies using least square non-linear regression

Rotation Curve Fitting Model

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

Estimating the mass of galactic components using machine learning algorithms

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

Learning the galaxy-environment connection with graph neural networks

Robust Simulation-Based Inference in Cosmology with Bayesian Neural Networks

A probabilistic deep learning model to distinguish cusps and cores in dwarf galaxies

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

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