arXiv AI

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

arXiv AI
Oct 28

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.

By Ibrahim Elsharkawy, David Curtin, Yonatan Kahn, Neal Weiner
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
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 Machine Learning
Oct 19

Learning without relearning physical symmetry: an SO(3)-equivariant convolutional network (CNN) on the sphere

The paper introduces a novel SO(3)-equivariant convolutional neural network designed for data on the sphere, specifically using the Hierarchical Equal Area isoLatitude Pixelization (HEALPix). By employing steerable filters, parallel transport, and spin-weighted features, the architecture guarantees exact equivariance and avoids the need for the network to learn spherical symmetry from data. Benchmarks show the model achieves percent-level SO(3)-invariance error and outperforms existing methods in dense gradient regression, cosmic microwave background foreground removal, and denoising tasks.

By Joshua Williamson, Niall Jeffrey, Lorne Whiteway
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
arXiv Machine Learning
Sep 22

Neural Residual Modeling for Scientific Data Compression under Guaranteed Error Bounds

The paper introduces a new compression pipeline for scientific simulation data that combines Residual Vector Quantization (RVQ) with a U‑Net post‑processing network to correct pixel‑space residuals, followed by a Guaranteed Autoencoder (GAE) that enforces block‑wise error bounds. The U‑Net is trained to predict spatially structured residuals, addressing limitations of latent‑space only approaches. Experiments on S3D, JHTDB, and E3SM datasets show improved NRMSE and compression ratios compared to RVQ alone while maintaining strict error guarantees.

By Surya Majumder, Liangji Zhu, Sanjay Ranka, Anand Rangarajan