Mitigating simulator dependence in AI parameter inference for the Epoch of Reionization: the importance of simulation diversity
Year: 2026
Type: article
Abstract: ABSTRACT The 21 cm signal of neutral hydrogen contains a wealth of information about the poorly constrained era of cosmological history, the Epoch of Reionization (EoR). Recently, AI models trained on EoR simulations have gained significant attention as a powerful and flexible option for inferring parameters from 21 cm observations. However, previous works show that AI models trained on data from one simulator fail to generalize to data from another, raising doubts about AI models’ ability to accurately infer parameters from observation. We develop a new strategy for training AI models on cosmological simulations based on the principle that increasing the diversity of the training data set improves model robustness by averaging out spurious and contradictory information. We train AI models on data from different combinations of four simulators, then compare the models’ performance when predicting on data from held-out simulators acting as proxies for our Universe. We find that models trained on data from multiple simulators perform better on data from a held-out simulator than models trained on data from a single simulator, indicating that increasing the diversity of the training data set improves a model’s ability to generalize. This result suggests that future EoR parameter inference methods can mitigate simulator-specific bias by incorporating multiple simulation approaches into their analyses.
Authors: Jasper Solt, Jonathan C Pober, Stephen H Bach
Institution: Brown University
Language: English
Subfield: Astronomy and Astrophysics
Field: Physics and Astronomy
Domain: Physical Sciences
Open Access status: gold
Estimated APC paid: $3,046
OpenAlex ID: W7161747800