ASTROLABE: A 3D Particle-Level Surrogate for Cosmology Simulation with Physical Flow Matching

ASTROLABE is a flow‑matching generative emulator that operates directly on the 6N phase‑space coordinates of individual particles, avoiding voxel grids and allowing any observable to be measured from inference. It uses a field‑particle hybrid transformer so that compute and memory scale linearly with the number of particles, and incorporates physical flow matching by basing the prior on COLA‑derived coordinates with momentum and mass conservation enforced throughout the flow. In Lambda‑CDM tests, ASTROLABE reproduces realistic fields, accurate power spectra, halo mass functions, velocity profiles, and halo density profiles, and it generalizes to out‑of‑distribution dark‑acoustic oscillation initial conditions while also being fine‑tuned to emulate large‑scale simulations such as TNG50‑3‑dark at a fraction of the cost.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

• llms
• diffusion
• fine-tuning