Emulation of the Halo Mass Function with Neural Networks
Abstract
The abundance of galaxy clusters as a function of mass --- the halo mass function (HMF) --- depends strongly on cosmological parameters. Large-scale simulations are computationally expensive, motivating emulators that interpolate simulation outputs. We present a neural network emulator that maps cosmological and astrophysical parameters, halo mass, and redshift directly to a cumulative count of halos above the input mass, avoiding the information loss inherent to binning and to the functional or PCA-based data reduction used in prior emulators. Each emulator prediction is accompanied by a self-estimated error, expressed in units of Poisson shot noise, that the network learns to predict alongside the halo count itself --- giving every output a built-in confidence metric. We validate the emulator with leave-one-out tests on three simulation suites --- the N-body MassiveNus suite and the hydrodynamical CAMELS IllustrisTNG and SIMBA suites --- spanning a range of box sizes, redshifts, sub-grid physics, and softwares. Of all the held-out predictions (1.2× 106 in total), 88\% have model bias at or below shot noise and 99.9\% are within 3× shot noise. The only significant high-error tail is in MassiveNus, where the predicted model bias flags 76\% of predictions exceeding 3× shot noise; the remaining number of high error predictions is comparable to the number expected from shot noise alone. This emulator is publicly available and can be trained on other simulation suites making it a flexible HMF tool that can be used for cosmological and astrophysical inference.
Related
- Aletheia : Emulating the halo mass function with evolution mapping
- Aletheia: Emulating the halo mass function with evolution mapping
- A Unified Halo Mass Function Across Dark Matter Models from High-Resolution Multi-Scale Simulations
- From Quantum Field Fluctuations to High-Redshift AGN: Semiclassical Origins of Halo Mass Function Constraints in the JWST Era
- Halo Mass Function Constraints on Early Galaxy and AGN Populations in the JWST Era
BibTeX
@misc{ross2026emulation,
title = {{Emulation of the Halo Mass Function with Neural Networks}},
author = {Olive Ross and Nicholas Battaglia},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.00725},
eprint = {2610.00725},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.00725},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - Emulation of the Halo Mass Function with Neural Networks
AU - Olive Ross
AU - Nicholas Battaglia
PY - 2026
DA - 2026/10/01
PB - arXiv
DO - 10.48550/arxiv.2610.00725
UR - https://arxiv.org/abs/2610.00725
AB - The abundance of galaxy clusters as a function of mass --- the halo mass function (HMF) --- depends strongly on cosmological parameters. Large-scale simulations are computationally expensive, motivating emulators that interpolate simulation outputs. We present a neural network emulator that maps cosmological and astrophysical parameters, halo mass, and redshift directly to a cumulative count of halos above the input mass, avoiding the information loss inherent to binning and to the functional or PCA-based data reduction used in prior emulators. Each emulator prediction is accompanied by a self-estimated error, expressed in units of Poisson shot noise, that the network learns to predict alongside the halo count itself --- giving every output a built-in confidence metric. We validate the emulator with leave-one-out tests on three simulation suites --- the N-body MassiveNus suite and the hydrodynamical CAMELS IllustrisTNG and SIMBA suites --- spanning a range of box sizes, redshifts, sub-grid physics, and softwares. Of all the held-out predictions (1.2× 10^6 in total), 88\% have model bias at or below shot noise and 99.9\% are within 3× shot noise. The only significant high-error tail is in MassiveNus, where the predicted model bias flags 76\% of predictions exceeding 3× shot noise; the remaining number of high error predictions is comparable to the number expected from shot noise alone. This emulator is publicly available and can be trained on other simulation suites making it a flexible HMF tool that can be used for cosmological and astrophysical inference.
ER -