arXiv:2502.03139astro-ph.COastro-ph.IM2025-02被引 6

用神经网络快速生成宇宙初始密度场样本,速度提升数万倍。

Fast Sampling of Cosmological Initial Conditions with Gaussian Neural Posterior Estimation

  • 用高斯后验建模初始条件,傅里叶空间对角协方差加速采样。
  • 单张GPU秒级生成数千个后验样本,比现有方法快数个数量级。
  • 适用于高分辨率模拟,适合做序列化贝叶斯推断的科研人员。

宇宙大尺度结构起源于原始物质密度场,但从中推断初始条件极其困难,需在千万维参数空间中结合复杂模拟与统计方法。本文展示如何利用基于模拟的推断(SBI)解决此问题,通过在傅里叶空间建模初始条件的后验分布为对角协方差的高斯分布,实现对数据约束的原始暗物质密度场高效采样。该方法适用于全分辨率暗物质N体模拟,可在单张GPU上秒级生成数千个后验样本,速度较现有方法提升数个数量级,为宇宙学场的序列化SBI铺平道路。此外,我们对协方差随波数的依赖关系进行了解析拟合,将任意点估计器转化为快速采样器。通过摘要统计量和贝叶斯一致性检验验证了样本的有效性。

原文摘要 · Abstract (English)

Knowledge of the primordial matter density field from which the large-scale structure of the Universe emerged over cosmic time is of fundamental importance for cosmology. However, reconstructing these cosmological initial conditions from late-time observations is a notoriously difficult task, which requires advanced cosmological simulators and sophisticated statistical methods to explore a multi-million-dimensional parameter space. We show how simulation-based inference (SBI) can be used to tackle this problem and to obtain data-constrained realisations of the primordial dark matter density field in a simulation-efficient way with general non-differentiable simulators. Our method is applicable to full high-resolution dark matter $N$-body simulations and is based on modelling the posterior distribution of the constrained initial conditions to be Gaussian with a diagonal covariance matrix in Fourier space. As a result, we can generate thousands of posterior samples within seconds on a single GPU, orders of magnitude faster than existing methods, paving the way for sequential SBI for cosmological fields. Furthermore, we perform an analytical fit of the estimated dependence of the covariance on the wavenumber, effectively transforming any point-estimator of initial conditions into a fast sampler. We test the validity of our obtained samples by comparing them to the true values with summary statistics and performing a Bayesian consistency test.

宇宙学贝叶斯推断快速采样神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。