arXiv cs.AI· Ibrahim Elsharkawy, David Curtin, Yonatan Kahn, Neal Weiner·· 13 小时前AI 评分29
Astrolabe：基于物理流匹配的宇宙学模拟 3D 粒子级代理模型
ASTROLABE: A 3D Particle-Level Surrogate for Cosmology Simulation with Physical Flow Matching
AI 导读
Astrolabe 是直接作用于 N 个粒子 6N 相空间坐标的流匹配生成式代理模型，输出不含体素网格，因此没有体素分辨率上限，原则上可从推理结果测量任意观测量。
正文 · 原文
Abstract:The microphysics of dark matter (DM) is imprinted on cosmological structure across an enormous dynamic range. Nonlinear cosmological observables are most reliably modeled with N-body and hydrodynamic simulations. However, systematic N-body scans over DM theory space are prohibitively expensive. Machine learning emulators provide a possible solution, but are currently limited by either low resolution or only the ability to emulate specific observables. We introduce Astrolabe, a flow-matching-based generative emulator designed to be free of both constraints. Astrolabe acts directly on the 6N phase-space coordinates of N individual particles. With no voxel grid in the output, there is no voxel-resolution ceiling, and in principle any observable can be measured from inference. Astrolabe is computationally tractable via a field-particle hybrid transformer in which every particle cross-attends to multiscale field tokens, so compute and memory are linear in N. Astrolabe incorporates physical flow matching by taking the prior distribution for the flow to be approximate COmoving Lagrangian Acceleration (COLA)-based coordinates rather than unphysical noise, with momentum and mass conservation enforced at all points along flow time. In the Lambda-CDM cases we test, Astrolabe produces realistic fields with accurate power spectra, halo mass functions, velocity profiles, and halo density profiles. We also observe out-of-distribution generalization, with accurate emulation of dark-acoustic oscillation initial conditions. Astrolabe is also capable of being fine-tuned to simulate very large N and we demonstrate this by emulating a TNG50-3-dark simulation at a fraction of the cost. This method represents a first step toward a general and powerful cosmology simulation surrogate, which may eventually permit the inclusion of baryonic physics and a systematic exploration of richer dark sector interactions.
| Comments: | 24 pages + 7 appendix (31 total), 20 figures, 11 tables |
| Subjects: | Cosmology and Nongalactic Astrophysics (astro-ph.CO); Artificial Intelligence (cs.AI); High Energy Physics - Phenomenology (hep-ph) |
| Cite as: | arXiv:2610.27068 [astro-ph.CO] |
| (or arXiv:2610.27068v1 [astro-ph.CO] for this version) | |
| https://doi.org/10.48550/arXiv.2610.27068 arXiv-issued DOI via DataCite |
Submission history
From: Ibrahim Elsharkawy [view email]
[v1]
Mon, 19 Oct 2026 18:00:02 UTC (16,620 KB)
来源：arXiv cs.AI · arxiv.org