mlsanityData to 14 Oct 2026

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

arXiv 2608.26354 ↗First posted 26 Aug 2026astro-ph.CO astro-ph.IM cs.LGcited by 0

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

  1. 2024Flow Matching for Posterior Inference with Simulator Feedback
  2. 2025Simulation-Efficient Cosmological Inference with Multi-Fidelity SBI
  3. 2024Diffusion-HMC: Parameter Inference with Diffusion-model-driven Hamiltonian Monte Carlo
  4. 2026What AstroPT knows about galaxies, and what that can teach us about LLMs
  5. 2026A Missing Latent, Not a Missing Simulator: Radius-Augmented Inference for Real JWST Retrieval
  6. 2024Using Galaxy Evolution as Source of Physics-Based Ground Truth for Generative Models
  7. 2024Bayesian Deconvolution of Astronomical Images with Diffusion Models: Quantifying Prior-Driven Features in Reconstructions
  8. 2026Strong Gravitational Lensing Posterior Sampling in Pixel-Space Using Diffusion Models and Recurrent Inference Machines