Learning cosmology and clustering with cosmic graphs

Abstract

We train deep learning models on thousands of galaxy catalogues from the state-of-the-art hydrodynamic simulations of the CAMELS project to perform regression and inference. We employ Graph Neural Networks (GNNs), architectures designed to work with irregular and sparse data, like the distribution of galaxies in the Universe. We first show that GNNs can learn to compute the power spectrum of galaxy catalogues with a few percent accuracy. We then train GNNs to perform likelihood-free inference at the galaxy-field level. Our models are able to infer the value of ${\Omega }_{\mathrm{m}}$ with a $\sim 12\%-13\%$ accuracy just from the positions of $\sim 1000$ galaxies in a volume of $(25h^{-1}\mathrm{Mpc}{)}^3$ at $z=0$ while accounting for astrophysical uncertainties as modelled in CAMELS. Incorporating information from galaxy properties, such as stellar mass, stellar metallicity, and stellar radius, increases the accuracy to $4\%-8\%$. Our models are built to be translational and rotational invariant, and they can extract information from any scale larger than the minimum distance between two galaxies. However, our models are not completely robust: testing on simulations run with a different subgrid physics than the ones used for training does not yield as accurate results.

arXiv comment21 pages, 8 figures, code publicly available at https://github.com/PabloVD/CosmoGraphNet

References in corpus (57)

• Populating a cluster of galaxies - I. Results at z=0
• E pur si muove: Galiliean-invariant cosmological hydrodynamical simulations on a moving mesh
• Simulating Galaxy Formation with the IllustrisTNG Model
• Simulating galaxy formation with black hole driven thermal and kinetic feedback
• GIZMO: A New Class of Accurate, Mesh-Free Hydrodynamic Simulation Methods
• Simba: Cosmological Simulations with Black Hole Growth and Feedback
• The Arepo public code release
• The Quijote simulations
• The CAMELS project: Cosmology and Astrophysics with MachinE Learning Simulations
• Super-Sample Covariance in Simulations
• Cosmological constraints with deep learning from KiDS-450 weak lensing maps
• Constraining $M_{\nu }$ with the Bispectrum I: Breaking Parameter Degeneracies
• Non-Gaussian information from weak lensing data via deep learning
• Likelihood-free inference with neural compression of DES SV weak lensing map statistics
• Towards an Optimal Estimation of Cosmological Parameters with the Wavelet Scattering Transform
• Fisher for complements: Extracting cosmology and neutrino mass from the counts-in-cells PDF
• Constraining $M_{\nu }$ with the Bispectrum II: The Total Information Content of the Galaxy Bispectrum
• Nearest Neighbor distributions: new statistical measures for cosmological clustering
• Using the Marked Power Spectrum to Detect the Signature of Neutrinos in Large-Scale Structure
• Deep learning from 21-cm tomography of the Cosmic Dawn and Reionization
• Weak lensing cosmology with convolutional neural networks on noisy data
• New Interpretable Statistics for Large Scale Structure Analysis and Generation
• Detecting Neutrino Mass by Combining Matter Clustering, Halos, and Voids
• Cosmic Shear Cosmology Beyond 2-Point Statistics: A Combined Peak Count and Correlation Function Analysis of DES-Y1
• Constraining neutrino mass with tomographic weak lensing one-point probability distribution function and power spectrum
• Constraining the astrophysics and cosmology from 21cm tomography using deep learning with the SKA
• The CAMELS Multifield Dataset: Learning the Universe's Fundamental Parameters with Artificial Intelligence
• Constraining neutrino mass with weak lensing Minkowski Functionals
• Constraining neutrino mass with tomographic weak lensing peak counts
• Weak lensing scattering transform: dark energy and neutrino mass sensitivity
• Primordial non-Gaussianity without tails -- how to measure fNL with the bulk of the density PDF
• Joint analysis of anisotropic power spectrum, bispectrum and trispectrum: application to N-body simulations
• Matter trispectrum: theoretical modelling and comparison to N-body simulations
• Clipping the Cosmos: The Bias and Bispectrum of Large Scale Structure
• Cosmological cross-correlations and nearest neighbor distributions
• Galaxy power-spectrum responses and redshift-space super-sample effect
• Interpreting deep learning models for weak lensing
• Separate Universe Simulations with IllustrisTNG: baryonic effects on power spectrum responses and higher-order statistics
• A Hybrid Deep Learning Approach to Cosmological Constraints From Galaxy Redshift Surveys
• Lossless, Scalable Implicit Likelihood Inference for Cosmological Fields
• Cosmology with one galaxy?
• Removing Astrophysics in 21 cm maps with Neural Networks
• Information Content of Higher-Order Galaxy Correlation Functions
• Finding universal relations in subhalo properties with artificial intelligence
• Breaking baryon-cosmology degeneracy with the electron density power spectrum
• Minkowski Functionals of Convergence Maps and the Lensing Figure of Merit
• Simultaneously constraining cosmology and baryonic physics via deep learning from weak lensing
• Exploring KSZ velocity reconstruction with $N$-body simulations and the halo model
• What Can We Learn by Combining the Skew Spectrum and the Power Spectrum?
• Clipping the Cosmos II: Cosmological information from non-linear scales
• The void halo mass function: a promising probe of neutrino mass
• Weighing neutrinos with the halo environment
• Extracting cosmological parameters from N-body simulations using machine learning techniques
• Probing massive neutrinos with the Minkowski functionals of large-scale structure
• The unequal-time matter power spectrum: impact on weak lensing observables
• Cosmology with the kinetic Sunyaev-Zeldovich effect: Independent of the optical depth and ${\sigma }_8$
• Constraining the Neutrino Mass with the Drifting Coefficient of the Field Cluster Mass Function

Cited by in corpus (5)

• Field Level Neural Network Emulator for Cosmological N-body Simulations
• Large-scale density and velocity field reconstructions with neural networks
• Robust Neural Network-Enhanced Estimation of Local Primordial Non-Gaussianity
• Machine-learning cosmology from void properties
• A graph model for the clustering of dark matter halos