Bayesian Inference of Initial Conditions from Non-Linear Cosmic Structures using Field-Level Emulators

Abstract

Analysing next-generation cosmological data requires balancing accurate modeling of non-linear gravitational structure formation and computational demands. We propose a solution by introducing a machine learning-based field-level emulator, within the Hamiltonian Monte Carlo-based Bayesian Origin Reconstruction from Galaxies (BORG) inference algorithm. Built on a V-net neural network architecture, the emulator enhances the predictions by first-order Lagrangian perturbation theory to be accurately aligned with full $N$-body simulations while significantly reducing evaluation time. We test its incorporation in BORG for sampling cosmic initial conditions using mock data based on non-linear large-scale structures from $N$-body simulations and Gaussian noise. The method efficiently and accurately explores the high-dimensional parameter space of initial conditions, fully extracting the cross-correlation information of the data field binned at a resolution of $1.95h^{-1}$ Mpc. Percent-level agreement with the ground truth in the power spectrum and bispectrum is achieved up to the Nyquist frequency $k_{\mathrm{N}}\approx 2.79h{\mathrm{Mpc}}^{-1}$. Posterior resimulations - using the inferred initial conditions for $N$-body simulations - show that the recovery of information in the initial conditions is sufficient to accurately reproduce halo properties. In particular, we show highly accurate $M_{200\mathrm{c}}$ halo mass function and stacked density profiles of haloes in different mass bins $[0.853,16]\times 10^{14}{M}_{\odot }{h}^{-1}$. As all available cross-correlation information is extracted, we acknowledge that limitations in recovering the initial conditions stem from the noise level and data grid resolution. This is promising as it underscores the significance of accurate non-linear modeling, indicating the potential for extracting additional information at smaller scales.

19 pages, 15 figures. Updated to match version accepted by MNRAS (published 2024/11/27)

References in corpus (67)

• The cosmological simulation code GADGET-2
• The Seventh Data Release of the Sloan Digital Sky Survey
• LSST: from Science Drivers to Reference Design and Anticipated Data Products
• The Cosmic Linear Anisotropy Solving System (CLASS) II: Approximation schemes
• SDSS-III: Massive Spectroscopic Surveys of the Distant Universe, the Milky Way Galaxy, and Extra-Solar Planetary Systems
• The Baryon Oscillation Spectroscopic Survey of SDSS-III
• Toward a halo mass function for precision cosmology: the limits of universality
• Deep learning for universal linear embeddings of nonlinear dynamics
• Cosmology and Fundamental Physics with the Euclid Satellite
• Ahf: Amiga's Halo Finder
• Multi-scale initial conditions for cosmological simulations
• Knowledge Distillation and Student-Teacher Learning for Visual Intelligence: A Review and New Outlooks
• Extragalactic Science, Cosmology and Galactic Archaeology with the Subaru Prime Focus Spectrograph (PFS)
• Renormalized Cosmological Perturbation Theory
• Solving Large Scale Structure in Ten Easy Steps with COLA
• The Quijote simulations
• Bayesian physical reconstruction of initial conditions from large scale structure surveys
• The halo mass function through the cosmic ages
• nbodykit: an open-source, massively parallel toolkit for large-scale structure
• The 2M++ galaxy redshift catalogue
• Evolution of the Mass Function of Dark Matter Haloes
• Learning to Predict the Cosmological Structure Formation
• FastPM: a new scheme for fast simulations of dark matter and halos
• ELUCID - Exploring the Local Universe with reConstructed Initial Density field I: Hamiltonian Markov Chain Monte Carlo Method with Particle Mesh Dynamics
• Physical Bayesian modelling of the non-linear matter distribution: new insights into the Nearby Universe
• Fast Hamiltonian sampling for large scale structure inference
• Past and present cosmic structure in the SDSS DR7 main sample
• The Initial Conditions of the Universe from Constrained Simulations
• Towards optimal extraction of cosmological information from nonlinear data
• Accurate initial conditions for cosmological N-body simulations: Minimizing truncation and discreteness errors
• Unmasking the Masked Universe: the 2M++ catalogue through Bayesian eyes
• Cosmological Reconstruction From Galaxy Light: Neural Network Based Light-Matter Connection
• A rigorous EFT-based forward model for large-scale structure
• Bayesian inference of cosmic density fields from non-linear, scale-dependent, and stochastic biased tracers
• Bayesian analysis of the dynamic cosmic web in the SDSS galaxy survey
• Cosmological inference from Bayesian forward modelling of deep galaxy redshift surveys
• How well do cosmological simulations reproduce individual-halo properties?
• How Biased Are Halo Properties in Cosmological Simulations?
• FlowPM: Distributed TensorFlow Implementation of the FastPM Cosmological N-body Solver
• COSMIC BIRTH: Efficient Bayesian Inference of the Evolving Cosmic Web from Galaxy Surveys
• Field Level Neural Network Emulator for Cosmological N-body Simulations
• SIBELIUS-DARK: a galaxy catalogue of the Local Volume from a constrained realisation simulation
• Bayesian forward modelling of cosmic shear data
• Consistency tests of field level inference with the EFT likelihood
• Inferring high redshift large-scale structure dynamics from the Lyman-alpha forest
• Predicting dark matter halo formation in N-body simulations with deep regression networks
• $\mathrm{S}\mathrm{IM}\mathrm{BIG}$: Mock Challenge for a Forward Modeling Approach to Galaxy Clustering
• Impacts of the physical data model on the forward inference of initial conditions from biased tracers
• Cosmology with cosmic web environments I. Real-space power spectra
• BIRTH of the COSMOS Field: Primordial and Evolved Density Reconstructions During Cosmic High Noon
• On the accuracy and precision of correlation functions and field-level inference in cosmology
• NECOLA: Towards a Universal Field-level Cosmological Emulator
• Cosmology inference at the field level from biased tracers in redshift-space
• Taking measurements of the kinematic Sunyaev-Zel'dovich effect forward: including uncertainties from velocity reconstruction with forward modeling
• Joint velocity and density reconstruction of the Universe with nonlinear differentiable forward modeling
• A hierarchical field-level inference approach to reconstruction from sparse Lyman-$\alpha$ forest data
• Field-level inference of galaxy intrinsic alignment from the SDSS-III BOSS survey
• Bayesian cosmic density field inference from redshift space dark matter maps
• Differentiable Cosmological Simulation with Adjoint Method
• Towards Accurate Field-Level Inference of Massive Cosmic Structures
• Bayesian power-spectrum inference with foreground and target contamination treatment
• Reconstructing the Universe with Variational self-Boosted Sampling
• Field-Based Physical Inference From Peculiar Velocity Tracers
• Catalogues of voids as antihalos in the local Universe
• How much cosmological information can be measured?
• The effect of local universe constraints on halo abundance and clustering
• Evaluating the variance of individual halo properties in constrained cosmological simulations

Cited by in corpus (12)

• A Parameter-Masked Mock Data Challenge for Beyond-Two-Point Galaxy Clustering Statistics
• CSST Cosmological Emulator I: Matter Power Spectrum Emulation with one percent accuracy to k = 10 h/Mpc
• A field-level emulator for modified gravity
• COmoving Computer Acceleration (COCA): $N$-body simulations in an emulated frame of reference
• Field-level Reconstruction from Foreground-Contaminated 21-cm Maps
• Learning the Universe: Learning to Optimize Cosmic Initial Conditions with Non-Differentiable Structure Formation Models
• The galaxy-environment connection revealed by constrained simulations
• Fast Sampling of Cosmological Initial Conditions with Gaussian Neural Posterior Estimation
• Diagnosing Systematic Effects Using the Inferred Initial Power Spectrum
• On the Connection between Field-Level Inference and $n$-point Correlation Functions
• Differentiable Fuzzy Cosmic-Web for Field Level Inference
• Predicting large scale cosmological structure evolution with generative adversarial network-based autoencoders