Cross-simulator transfer with foundation model summaries: Towards robust SKA-era reionization inference
Not on the map. Why?
Abstract (arXiv)
Simulation-based inference (SBI) for parameter estimation is vulnerable to model misspecification: neural summaries and density estimators trained on a specific forward model typically fail when applied to data drawn from another model, or from real observations, and no training simulator can capture the full observational pipeline of a real measurement exactly. We show that a self-supervised Vision Transformer (ViT), pretrained label-free on a fast approximate simulator, produces transferable data summaries that generalize across simulators. Without retraining, it can be reused as a frozen encoder to infer astrophysical parameters from a completely different simulator that resolves the radiative transfer explicitly, on which it has never seen either data or parameters. As a concrete use case in 21cm cosmology, SKATR, a ViT pretrained with a Joint Embedding Predictive Architecture (JEPA), serves as a foundation model for reionization inference from upcoming SKA measurements: SKATR is pretrained once on 67k low-cost, noiseless semi-numerical 21cmFAST lightcones, then frozen and applied to hydrodynamical Loreli II lightcones, where a lightweight conditional flow matching head infers five astrophysical parameters; the encoder is never shown Loreli data, its parameters, or any noise. In our comparison, SKATR yields the most precise and best-calibrated posteriors across all five parameters, matching the accuracy of the fully-supervised in-domain baseline while requiring 2.6x fewer radiative-transfer simulations. Under realistic SKA AA* noise, only SKATR remains simultaneously accurate, informative, and calibrated, outperforming even a supervised baseline retrained from scratch on noisy data. Self-supervised pretraining on computationally efficient semi-numerical simulations is therefore a viable route to calibrated, simulator- and noise-agnostic reionization inference for the SKA-era.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 0 of its 5 references that are in the map. First posted on arXiv 26 Aug 2026.
Builds on
0 of 5 references.
No references here yet.
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Flow Matching for Posterior Inference with Simulator Feedback
- 2025Simulation-Efficient Cosmological Inference with Multi-Fidelity SBI
- 2024Diffusion-HMC: Parameter Inference with Diffusion-model-driven Hamiltonian Monte Carlo
- 2026What AstroPT knows about galaxies, and what that can teach us about LLMs
- 2026A Missing Latent, Not a Missing Simulator: Radius-Augmented Inference for Real JWST Retrieval
- 2024Using Galaxy Evolution as Source of Physics-Based Ground Truth for Generative Models
- 2024Bayesian Deconvolution of Astronomical Images with Diffusion Models: Quantifying Prior-Driven Features in Reconstructions
- 2026Strong Gravitational Lensing Posterior Sampling in Pixel-Space Using Diffusion Models and Recurrent Inference Machines