Low-cost neural summaries for cosmological inference: an ensemble of convolutional neural networks trained on approximate simulations

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 $Ω_{\rm 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)