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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Jones, J. Dorigo, Bahauddin, S. M., Rapetti, D., Mirocha, J., Burns, J. O.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914969267732480
author Jones, J. Dorigo
Bahauddin, S. M.
Rapetti, D.
Mirocha, J.
Burns, J. O.
author_facet Jones, J. Dorigo
Bahauddin, S. M.
Rapetti, D.
Mirocha, J.
Burns, J. O.
contents 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 $\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, $\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 $\texttt{21cmLSTM}$ to fit global 21 cm signal mock data with different adopted observational noise levels, $σ_{21}$. The posterior $1σ$ rms error is $\approx3\times$ less than $σ_{21}$ for each fit and consistently decreases for tighter noise levels, showing that $\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 $\texttt{21cmLSTM}$ can be independently tested and used to retrain and constrain other 21 cm models.
format Preprint
id arxiv_https___arxiv_org_abs_2410_07619
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle $\texttt{21cmLSTM}$: A Fast Memory-based Emulator of the Global 21 cm Signal with Unprecedented Accuracy
Jones, J. Dorigo
Bahauddin, S. M.
Rapetti, D.
Mirocha, J.
Burns, J. O.
Cosmology and Nongalactic Astrophysics
Astrophysics of Galaxies
Instrumentation and Methods for Astrophysics
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 $\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, $\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 $\texttt{21cmLSTM}$ to fit global 21 cm signal mock data with different adopted observational noise levels, $σ_{21}$. The posterior $1σ$ rms error is $\approx3\times$ less than $σ_{21}$ for each fit and consistently decreases for tighter noise levels, showing that $\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 $\texttt{21cmLSTM}$ can be independently tested and used to retrain and constrain other 21 cm models.
title $\texttt{21cmLSTM}$: A Fast Memory-based Emulator of the Global 21 cm Signal with Unprecedented Accuracy
topic Cosmology and Nongalactic Astrophysics
Astrophysics of Galaxies
Instrumentation and Methods for Astrophysics
url https://arxiv.org/abs/2410.07619