Low-cost neural summaries for cosmological inference: an ensemble of convolutional neural networks trained on approximate simulations
astro-ph.CO
Submitted: 2026-10-14
Updated: 2026-10-14
Comments: Submitted to A&A. 15 pages, 17 figures
License: http://creativecommons.org/licenses/by/4.0/
The gist: Machine learning can build near-optimal summary statistics for extracting cosmological information from the large-scale structure of the Universe.
Abstract
Machine learning can build near-optimal summary statistics for extracting cosmological information from the large-scale structure of the Universe. Training them, however, requires large numbers of costly simulations. We aim to build a neural summary trained on fast, approximate low-resolution simulations that remains informative when applied to higher-resolution ones, reducing the cost of the training set. We train an ensemble of convolutional neural networks with different architectures to compress the matter overdensity fields of low-resolution particle-mesh simulations. The outputs of a subset of networks are concatenated into a summary statistic, which we use to constrain Ω m and σ 8 from high-resolution particle-mesh simulations, each about a thousand times more costly than their low-resolution counterparts. Theoretical predictions are provided by a Gaussian process emulator built from the high-resolution simulations. We test three strategies for selecting the smallest informative subset of networks. Constraints are compared with those from a reference power spectrum analysis. We also consider dark matter halo catalogues from independent high-resolution N-body simulations. The neural summary extracts information beyond the binned power spectrum. Using multiple networks is essential: a single network can be less informative than the power spectrum. A greedy search based on posterior volume provides the tightest constraints with the fewest networks. The neural summary also outperforms the power spectrum for dark matter haloes, at different number densities and redshifts, despite being trained on low-resolution matter fields at z=0. This work demonstrates that an ensemble of networks trained on approximate simulations yields a summary statistic that remains informative on higher-fidelity data, including dark matter haloes. (Abridged)
Sources
- The Wide Field Infrared Survey Telescope: 100 Hubbles for the 2020s
- Augmented Correlation Functions for Spectroscopic Galaxy Surveys
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
- Convolutional Vision Transformer for Cosmology Parameter Inference
- Vision Transformers for Cosmological Fields: Application to Weak Lensing Mass Maps
- Adam: A Method for Stochastic Optimization
- Decoupled Weight Decay Regularization
- Hybrid Summary Statistics
- Cross-simulator transfer with foundation model summaries: Towards robust SKA-era reionization inference
- Field-level weak lensing cosmology with 60 simulations using multifidelity simulation-based inference
- Field-level multi-tracers simulation-based inference of cosmological parameters from 3D maps
- Interpretable Neural Marked Statistics for Cosmological Inference
- Dark Quest II: A Wide-Coverage Neural Network Emulator of the Nonlinear Matter Power Spectrum Across Extended Cosmologies
Related papers
- Angular clustering and bias of photometric quasars in the Kilo-Degree Survey Data Release 4
- A Novel kinetic Sunyaev-Zel'dovich Estimator for Electron-Electron Correlations
- Magnetic fields at the dawn of structure formation I. The CARLA J1510+5958 proto-cluster
- Dark Energy Survey Year 6 Results: Weak Lensing and Galaxy Clustering Cosmological Analysis Framework
- Exploring the Impact of Systematic Bias in Type Ia Supernova Cosmology Across Diverse Dark Energy Parametrizations
- Non-Gaussian Galaxy Stochasticity and the Noise-Field Formulation