Abstract:Accurate 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, especially if magnetohydrodynamics are included, 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 catalogue 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 in network weights was realised through incorporating symmetries motivated by first principles into our model architecture. We trained our model using dark-matter-only $N$-body simulations across different resolutions, redshifts, and mass bins. We validated 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 of approximate halo mock generation that became evident from this study.

Psi-GAN: A power-spectrum-informed generative adversarial network for the emulation of large-scale structure maps across cosmologies and redshifts

Emulation of f(R) modified gravity from ΛCDM using conditional GANs

GANSky -- fast curved sky weak lensing simulations using Generative Adversarial Networks

Using conditional GANs for convergence map reconstruction with uncertainties

Emulation of $f(R)$ modified gravity from $Λ$CDM using conditional GANs

Generative modelling for mass-mapping with fast uncertainty quantification

Predicting large scale cosmological structure evolution with GAN-based autoencoders

High precision accelerator for our hybrid model of the redshift space power spectrum

Sesame: A power spectrum emulator pipeline for beyond-Lambda CDM models

Toward using GANs in astrophysical Monte-Carlo simulations

AI-assisted Reconstruction of Cosmic Velocity Field from Redshift-Space Spatial Distribution of Halos

Fast emulation of cosmological density fields based on dimensionality reduction and supervised machine-learning

CHARM: Creating Halos with Auto-Regressive Multi-stage networks

LHC analysis-specific datasets with Generative Adversarial Networks

PineTree : A generative, fast, and differentiable halo model for wide-field galaxy surveys

Posterior Sampling of the Initial Conditions of the Universe from Non-linear Large Scale Structures using Score-Based Generative Models

The Aemulus Project. VI. Emulation of Beyond-standard Galaxy Clustering Statistics to Improve Cosmological Constraints

Sesame: A power spectrum emulator pipeline for beyond-$Λ$CDM models

The Aemulus Project VI: Emulation of beyond-standard galaxy clustering statistics to improve cosmological constraints

Dark energy reconstruction analysis with artificial neural networks: Application on simulated Supernova Ia data from Rubin Observatory

Realistic galaxy images and improved robustness in machine learning tasks from generative modelling