$\text{PineTree}$: A generative, fast, and differentiable halo model for wide-field galaxy surveys

Abstract

Mock halo catalogues are indispensable data products for developing and validating cosmological inference pipelines. A major challenge in generating mock catalogues is modelling the halo or galaxy bias, which is the mapping from matter density to dark matter halos or observable galaxies. To this end, N-body codes produce state-of-the-art catalogues. However, generating large numbers of these N-body simulations for big volumes, requires significant computational time. We introduce and benchmark a differentiable and physics-informed neural network that can generate mock halo catalogues of comparable quality to those obtained from full N-body codes. The model design is computationally efficient for the training procedure and the production of large mock suites. We present a neural network, relying only on 18 to 34 trainable parameters, that produces halo catalogues from dark matter overdensity fields. The reduction of network weights is realised through incorporating symmetries motivated by first principles into our model architecture. We train our model using dark matter only N-body simulations across different resolutions, redshifts, and mass bins. We validate the final mock catalogues by comparing them to N-body halo catalogues using different N-point correlation functions. Our model produces mock halo catalogues consistent with the reference simulations, showing that this novel network is a promising way to generate mock data for upcoming wide-field surveys due to its computational efficiency. Moreover, we find that the network can be trained on approximate overdensity fields to reduce the computational cost further. We also present how the trained network parameters can be interpreted to give insights into the physics of structure formation. Finally, we discuss the current limitations of our model as well as more general requirements and pitfalls for approximate halo mock generation.

17 pages, 17 figures, 3 tables. Abstract abridged for arXiv, layout corrections, added reference section 2, accepted for publication in A&A

References in corpus (23)

• Euclid. I. Overview of the Euclid mission
• Galaxy Groups in the SDSS DR4: II. halo occupation statistics
• Gravity and Large-Scale Non-local Bias
• The Clustering of the SDSS Main Galaxy Sample II: Mock galaxy catalogues and a measurement of the growth of structure from Redshift Space Distortions at $z=0.15$
• The Scale Dependence of Halo and Galaxy Bias: Effects in Real Space
• Non-local Lagrangian bias
• Towards optimal parallel PM N-body codes: PMFAST
• nIFTy Cosmology: Galaxy/halo mock catalogue comparison project on clustering statistics
• Constraining the halo bispectrum in real and redshift space from perturbation theory and non-linear stochastic bias
• Semi-analytic forecasts for JWST -- VI. Simulated lightcones and galaxy clustering predictions
• Cosmological constraints from the BOSS DR12 void size function
• Field Level Neural Network Emulator for Cosmological N-body Simulations
• Robust Field-level Likelihood-free Inference with Galaxies
• ADDGALS: Simulated Sky Catalogs for Wide Field Galaxy Surveys
• The Cosmic Graph: Optimal Information Extraction from Large-Scale Structure using Catalogues
• Precise Cosmological Constraints from BOSS Galaxy Clustering with a Simulation-Based Emulator of the Wavelet Scattering Transform
• Galaxy Clustering Analysis with SimBIG and the Wavelet Scattering Transform
• Semi-analytic forecasts for Roman -- the beginning of a new era of deep-wide galaxy surveys
• Impacts of the physical data model on the forward inference of initial conditions from biased tracers
• Evolution of the luminosity-to-halo mass relation of LRGs from a combined SDSS-DR10+RCS2 analysis
• Fast and realistic large-scale structure from machine-learning-augmented random field simulations
• Fast generation of mock galaxy catalogues with COLA
• Field-level multiprobe analysis of the CMB, integrated Sachs-Wolfe effect, and the galaxy density maps

Cited by in corpus (6)

• Euclid preparation. Simulating thousands of Euclid spectroscopic skies
• COmoving Computer Acceleration (COCA): $N$-body simulations in an emulated frame of reference
• Secondary halo bias through cosmic time II: Reconstructing halo properties using clustering information
• Learning the Universe: Learning to Optimize Cosmic Initial Conditions with Non-Differentiable Structure Formation Models
• Hybrid summary statistics: neural weak lensing inference beyond the power spectrum
• Diagnosing Systematic Effects Using the Inferred Initial Power Spectrum