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 (9)

• ELUCID - Exploring the Local Universe with reConstructed Initial Density field I: Hamiltonian Markov Chain Monte Carlo Method with Particle Mesh Dynamics
• 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
• SIBELIUS-DARK: a galaxy catalogue of the Local Volume from a constrained realisation simulation
• Bayesian forward modelling of cosmic shear data
• 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
• Bayesian power-spectrum inference with foreground and target contamination treatment
• 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