English

$\texttt{21cmLSTM}$: A Fast Memory-based Emulator of the Global 21 cm Signal with Unprecedented Accuracy

Cosmology and Nongalactic Astrophysics 2024-10-11 v1 Astrophysics of Galaxies Instrumentation and Methods for Astrophysics

Abstract

Neural network (NN) emulators of the global 21 cm signal need emulation error much less than the observational noise in order to be used to perform unbiased Bayesian parameter inference. To this end, we introduce 21cmLSTM\texttt{21cmLSTM} -- a long short-term memory (LSTM) NN emulator of the global 21 cm signal that leverages the intrinsic correlation between frequency channels to achieve exceptional accuracy compared to previous emulators, which are all feedforward, fully connected NNs. LSTM NNs are a type of recurrent NN designed to capture long-term dependencies in sequential data. When trained and tested on the same simulated set of global 21 cm signals as the best previous emulators, 21cmLSTM\texttt{21cmLSTM} has average relative rms error of 0.22% -- equivalently 0.39 mK -- and comparably fast evaluation time. We perform seven-dimensional Bayesian parameter estimation analyses using 21cmLSTM\texttt{21cmLSTM} to fit global 21 cm signal mock data with different adopted observational noise levels, σ21\sigma_{21}. The posterior 1σ1\sigma rms error is 3×\approx3\times less than σ21\sigma_{21} for each fit and consistently decreases for tighter noise levels, showing that 21cmLSTM\texttt{21cmLSTM} can sufficiently exploit even very optimistic measurements of the global 21 cm signal. We made the emulator, code, and data sets publicly available so that 21cmLSTM\texttt{21cmLSTM} can be independently tested and used to retrain and constrain other 21 cm models.

Keywords

Cite

@article{arxiv.2410.07619,
  title  = {$\texttt{21cmLSTM}$: A Fast Memory-based Emulator of the Global 21 cm Signal with Unprecedented Accuracy},
  author = {J. Dorigo Jones and S. M. Bahauddin and D. Rapetti and J. Mirocha and J. O. Burns},
  journal= {arXiv preprint arXiv:2410.07619},
  year   = {2024}
}

Comments

15 pages, 8 figures, 3 tables. Accepted by ApJ