Mean-Field Simulation-Based Inference for Cosmological Initial Conditions

Oleg Savchenko,Florian List,Guillermo Franco Abellán,Noemi Anau Montel,Christoph Weniger
2024-10-21
Abstract:Reconstructing cosmological initial conditions (ICs) from late-time observations is a difficult task, which relies on the use of computationally expensive simulators alongside sophisticated statistical methods to navigate multi-million dimensional parameter spaces. We present a simple method for Bayesian field reconstruction based on modeling the posterior distribution of the initial matter density field to be diagonal Gaussian in Fourier space, with its covariance and the mean estimator being the trainable parts of the algorithm. Training and sampling are extremely fast (training: $\sim 1 \, \mathrm{h}$ on a GPU, sampling: $\lesssim 3 \, \mathrm{s}$ for 1000 samples at resolution $128^3$), and our method supports industry-standard (non-differentiable) $N$-body simulators. We verify the fidelity of the obtained IC samples in terms of summary statistics.
Cosmology and Nongalactic Astrophysics,Instrumentation and Methods for Astrophysics,Machine Learning
What problem does this paper attempt to address?