arXiv AI

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.

arXiv AI
Sep 10

Field-level weak lensing cosmology with $60$ simulations using multifidelity simulation-based inference

arXiv:2606.23346v2 Announce Type: replace-cross Abstract: We perform a realistic KiDS-Legacy mock analysis with field-level neural compression and simulation-based inference using just 60 $N$-body si...

By Alex A. Saoulis, Kiyam Lin, Niall Jeffrey, Maximilian von Wietersheim-Kramsta, Davide Piras, Alessio Spurio Mancini, Ana M. G. Ferreira, Benjamin Joachimi
arXiv Machine Learning
Aug 28

Cross-simulator transfer with foundation model summaries: Towards robust SKA-era reionization inference

The paper demonstrates that a self‑supervised Vision Transformer (ViT) pretrained on a fast, low‑cost semi‑numerical simulator can produce data summaries that transfer across different simulators without retraining. In 21cm cosmology, the ViT—named SKATR—pretrained on 67,000 21cmFAST lightcones is applied unchanged to hydrodynamical Loreli II lightcones, enabling accurate inference of five astrophysical parameters with fewer radiative‑transfer simulations than a fully‑supervised baseline. SKATR remains accurate, informative, and calibrated even under realistic SKA antenna array noise, outperforming supervised models retrained on noisy data.

By Yannic Pietschke, Caroline Heneka, Ayodele Ore, Romain Meriot
arXiv Machine Learning
Aug 6

Galaxy Phase-Space and Field-Level Cosmology: The Strength of Semi-Analytic Models

arXiv:2512. 10222v2 Announce Type: replace-cross Abstract: Semi-analytic models are a widely used approach to simulate galaxy properties within a cosmological framework, relying on simplified yet physically motivated prescriptions.

By Natal\'i S. M. de Santi, Francisco Villaescusa-Navarro, Pablo Araya-Araya, Gabriella De Lucia, Fabio Fontanot, Lucia A. Perez, Manuel Arn\'es-Curto, Violeta Gonzalez-Perez, \'Angel Chandro-G\'omez, Rachel S. Somerville, Tiago Castro
arXiv Computer Vision
Sep 7

Mapping Dark-Matter Clusters via Physics-Guided Diffusion Models

The paper presents a fully automated method for reconstructing the surface mass density of galaxy clusters using photometry and gravitational lensing data. It introduces DarkClusters-15k, a benchmark dataset of 15,000 simulated clusters with paired mass and photometry maps across multiple redshifts and simulation frameworks. By training a diffusion prior on this dataset, the authors generate posterior samples constrained by weak- and strong-lensing observables, achieving accurate, physics‑guided reconstructions with well‑calibrated uncertainties in minutes.

By Diego Royo, Brandon Zhao, Adolfo Mu\~noz, Diego Gutierrez, Katherine L. Bouman
arXiv Machine Learning
Sep 4

Generative Nested Sampling of Atomistic Thermodynamic Landscapes

The paper introduces NS‑Flows, a flow‑based nested sampling method that replaces Markov‑chain updates with a conditional normalizing flow trained on live sets. By applying this technique to a Lennard‑Jones particle system, the authors achieve over two orders of magnitude fewer energy evaluations and a roughly one‑third reduction in wall‑clock time compared to traditional nested sampling. The study also shows that the flow’s generation efficiency varies non‑monotonically along the annealing trajectory, providing a diagnostic of the system’s internal mode complexity and identifying liquid‑like ensembles as the most challenging for current flow architectures.

By Alessandro Coretti, Nico Unglert, Sebastian Falkner, Georg K. H. Madsen, Christoph Dellago
arXiv Machine Learning
Jun 8

ScatterPrism: convergence for generative simulation and inverse problems in particle and nuclear physics

arXiv:2604. 01313v2 Announce Type: replace Abstract: High-fidelity simulations and complex inverse problems, such as detector modeling and unfolding, are computationally intensive bottlenecks across subatomic physics, yet essential for accurate physical interpretation.

By Zeyu Xia, Tyler Kim, Trevor Reed, Judy Fox, Geoffrey Fox, Adam Szczepaniak