$\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.

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

References in corpus (117)

• The cosmological simulation code GADGET-2
• Large scale bias and the peak background split
• Populating a cluster of galaxies - I. Results at z=0
• The Cosmic Linear Anisotropy Solving System (CLASS) II: Approximation schemes
• Statistical Properties of X-ray Clusters: Analytic and Numerical Comparisons
• Cosmological Hydrodynamics with Adaptive Mesh Refinement: a new high resolution code called RAMSES
• Toward a halo mass function for precision cosmology: the limits of universality
• Ellipsoidal collapse and an improved model for the number and spatial distribution of dark matter haloes
• An analytic model for the spatial clustering of dark matter haloes
• The Rockstar Phase-Space Temporal Halo Finder and the Velocity Offsets of Cluster Cores
• Constraints on the relationship between stellar mass and halo mass at low and high redshift
• Analytic model for galaxy and dark matter clustering
• A Comprehensive Analysis of Uncertainties Affecting the Stellar Mass - Halo Mass Relation for 0<z<4
• Halo occupation numbers and galaxy bias
• An excursion set model of hierarchical clustering: Ellipsoidal collapse and the moving barrier
• Theoretical Models of the Halo Occupation Distribution: Separating Central and Satellite Galaxies
• Ahf: Amiga's Halo Finder
• Large-Scale Galaxy Bias
• Galaxy Clustering in the Completed SDSS Redshift Survey: The Dependence on Color and Luminosity
• Extragalactic Science, Cosmology and Galactic Archaeology with the Subaru Prime Focus Spectrograph (PFS)
• Constraining Galaxy Formation and Cosmology with the Conditional Luminosity Function of Galaxies
• Linking halo mass to galaxy luminosity
• Euclid. I. Overview of the Euclid mission
• The WiggleZ Dark Energy Survey: Survey Design and First Data Release
• Biasing and Hierarchical Statistics in Large-scale Structure
• HOP: A New Group-Finding Algorithm for N-body Simulations
• Spatial correlation function and pairwise velocity dispersion of galaxies: CDM models versus the Las Campanas Survey
• Simulating cosmic structure formation with the GADGET-4 code
• Fundamental Physics with the Square Kilometre Array
• The Connection between Galaxies and Dark Matter Structures in the Local Universe
• Stochastic Nonlinear Galaxy Biasing
• The WiggleZ Dark Energy Survey: Improved Distance Measurements to z = 1 with Reconstruction of the Baryonic Acoustic Feature
• Solving Large Scale Structure in Ten Easy Steps with COLA
• The origin and implications of dark matter anisotropic cosmic infall on ~L* haloes
• The Quijote simulations
• Galaxy Groups in the SDSS DR4: II. halo occupation statistics
• Galaxy Groups in the SDSS DR4: III. the luminosity and stellar mass functions
• The clustering of galaxies in the SDSS-III Baryon Oscillation Spectroscopic Survey: mock galaxy catalogues for the BOSS Final Data Release
• Bayesian physical reconstruction of initial conditions from large scale structure surveys
• The clustering of galaxies in the SDSS-III Baryon Oscillation Spectroscopic Survey: a large sample of mock galaxy catalogues
• 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 Uchuu Simulations: Data Release 1 and Dark Matter Halo Concentrations
• The Clustering of Galaxies in the SDSS-III Baryon Oscillation Spectroscopic Survey: Including covariance matrix errors
• The Scale Dependence of Halo and Galaxy Bias: Effects in Real Space
• PINOCCHIO: pinpointing orbit-crossing collapsed hierarchical objects in a linear density field
• FastPM: a new scheme for fast simulations of dark matter and halos
• EZmocks: extending the Zel'dovich approximation to generate mock galaxy catalogues with accurate clustering statistics
• PTHalos: A fast method for generating mock galaxy distributions
• Mock galaxy catalogs using the quick particle mesh method
• Halo Stochasticity from Exclusion and non-linear Clustering
• Large scale bias and stochasticity of halos and dark matter
• Modelling Baryon Acoustic Oscillations with Perturbation Theory and Stochastic Halo Biasing
• Minimizing the stochasticity of halos in large-scale structure surveys
• Subhalo abundance matching and assembly bias in the EAGLE simulation
• Physical Bayesian modelling of the non-linear matter distribution: new insights into the Nearby Universe
• Cosmological constraints with deep learning from KiDS-450 weak lensing maps
• Origin and evolution of halo bias in linear and non-linear regimes
• Biasing and the distribution of dark matter haloes
• Combining perturbation theories with halo models
• The Concentration Dependence of the Galaxy-Halo Connection: Modeling Assembly Bias with Abundance Matching
• Modeling Nonlinear Evolution of Baryon Acoustic Oscillations: Convergence Regime of N-body Simulations and Analytic Models
• Reconstructing Nonlinear Stochastic Bias from Velocity Space Distortions
• Past and present cosmic structure in the SDSS DR7 main sample
• Towards optimal extraction of cosmological information from nonlinear data
• The Abacus Cosmos: A Suite of Cosmological N-body Simulations
• Non-local Lagrangian bias
• ICE-COLA: Towards fast and accurate synthetic galaxy catalogues optimizing a quasi $N$-body method
• Predicting the number, spatial distribution and merging history of dark matter haloes
• Stochasticity of Bias and Nonlocality of Galaxy Formation: Linear Scales
• Towards optimal parallel PM N-body codes: PMFAST
• On the Distribution of Haloes, Galaxies and Mass
• Unmasking the Masked Universe: the 2M++ catalogue through Bayesian eyes
• Comparing approximate methods for mock catalogues and covariance matrices II: Power spectrum multipoles
• Observational evidence for stochastic biasing
• Precision cosmology with voids in the final BOSS data
• nIFTy Cosmology: Galaxy/halo mock catalogue comparison project on clustering statistics
• Fast and accurate mock catalogue generation for low-mass galaxies
• Cosmological Reconstruction From Galaxy Light: Neural Network Based Light-Matter Connection
• A rigorous EFT-based forward model for large-scale structure
• Constraining the halo bispectrum in real and redshift space from perturbation theory and non-linear stochastic bias
• The clustering of galaxies in the SDSS-III Baryon Oscillation Spectroscopic Survey: mock galaxy catalogues for the low-redshift sample
• A halo bias function measured deeply into voids without stochasticity
• HALOGEN: A tool for fast generation of mock halo catalogues
• Semi-analytic forecasts for JWST -- VI. Simulated lightcones and galaxy clustering predictions
• PINOCCHIO and the hierarchical build-up of dark matter haloes
• Cosmological constraints from the BOSS DR12 void size function
• Reconstructing the distribution of haloes and mock galaxies below the resolution limit in cosmological simulations
• Scale Dependence of Halo Bispectrum from Non-Gaussian Initial Conditions in Cosmological N-body Simulations
• CARPool: fast, accurate computation of large-scale structure statistics by pairing costly and cheap cosmological simulations
• A volumetric deep Convolutional Neural Network for simulation of mock dark matter halo catalogues
• Field Level Neural Network Emulator for Cosmological N-body Simulations
• Stochastic Biasing and Galaxy-Mass Density Relation in the Weakly Non-linear Regime
• 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
• ASOHF: a new adaptive spherical overdensity halo finder
• BAM: Bias Assignment Method to generate mock catalogs
• 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
• Non-Linear Matter Power Spectrum Covariance Matrix Errors and Cosmological Parameter Uncertainties
• Impacts of the physical data model on the forward inference of initial conditions from biased tracers
• Semi-analytic forecasts for Roman -- the beginning of a new era of deep-wide galaxy surveys
• Cosmic velocity--gravity relation in redshift space
• Revealing the cosmic web dependent halo bias
• Non-linear matter power spectrum from Time Renormalisation Group: efficient computation and comparison with one-loop
• Neural physical engines for inferring the halo mass distribution function
• Evolution of the luminosity-to-halo mass relation of LRGs from a combined SDSS-DR10+RCS2 analysis
• The dependence of subhalo abundance matching on galaxy photometry and selection criteria
• Fast and realistic large-scale structure from machine-learning-augmented random field simulations
• The cosmic web connection to the dark matter halo distribution through gravity
• Accurate halo-galaxy mocks from automatic bias estimation and particle mesh gravity solvers
• Perfectly parallel cosmological simulations using spatial comoving Lagrangian acceleration
• A galaxy-halo model of large-scale structure
• Fast generation of mock galaxy catalogues with COLA
• An Ising model for galaxy bias
• 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