syren-halofit: A fast, interpretable, high-precision formula for the $\Lambda$CDM nonlinear matter power spectrum

Abstract

Rapid and accurate evaluation of the nonlinear matter power spectrum, $P(k)$, as a function of cosmological parameters and redshift is of fundamental importance in cosmology. Analytic approximations provide an interpretable solution, yet current approximations are neither fast nor accurate relative to numerical emulators. We use symbolic regression to obtain simple analytic approximations to the nonlinear scale, $k_{\sigma }$, the effective spectral index, $n_{\mathrm{eff}}$, and the curvature, $C$, which are required for the halofit model. We then re-optimise the coefficients of halofit to fit a wide range of cosmologies and redshifts. We explore the space of analytic expressions to fit the residuals between $P(k)$ and the optimised predictions of halofit. Our results are designed to match the predictions of EuclidEmulator2, but are validated against $N$-body simulations. Our symbolic expressions for $k_{\sigma }$, $n_{\mathrm{eff}}$ and $C$ have root mean squared fractional errors of 0.8%, 0.2% and 0.3%, respectively, for redshifts below 3 and a wide range of cosmologies. The re-optimised halofit parameters reduce the root mean squared fractional error (compared to EuclidEmulator2) from 3% to below 2% for wavenumbers $k=9\times 10^{-3}-9h\mathrm{Mp}{\mathrm{c}}^{-1}$. We introduce syren-halofit (symbolic-regression-enhanced halofit), an extension to halofit containing a short symbolic correction which improves this error to 1%. Our method is 2350 and 3170 times faster than current halofit and hmcode implementations, respectively, and 2680 and 64 times faster than EuclidEmulator2 (which requires running class) and the BACCO emulator. We obtain comparable accuracy to EuclidEmulator2 and BACCO when tested on $N$-body simulations. Our work greatly increases the speed and accuracy of symbolic approximations to $P(k)$, making them significantly faster than their numerical counterparts without loss of accuracy.

arXiv comment11 pages, 8 figures. Accepted for publication in A&A

References in corpus (47)

• First Year Wilkinson Microwave Anisotropy Probe (WMAP) Observations: Determination of Cosmological Parameters
• Seven-Year Wilkinson Microwave Anisotropy Probe (WMAP) Observations: Cosmological Interpretation
• Wilkinson Microwave Anisotropy Probe (WMAP) Three Year Results: Implications for Cosmology
• The cosmological simulation code GADGET-2
• Five-Year Wilkinson Microwave Anisotropy Probe (WMAP) Observations: Cosmological Interpretation
• Efficient Computation of CMB anisotropies in closed FRW models
• The Cosmic Linear Anisotropy Solving System (CLASS) II: Approximation schemes
• Baryonic Features in the Matter Transfer Function
• Stable clustering, the halo model and nonlinear cosmological power spectra
• Halo Models of Large Scale Structure
• Revising the Halofit Model for the Nonlinear Matter Power Spectrum
• Analytic model for galaxy and dark matter clustering
• Power Spectra for Cold Dark Matter and its Variants
• Nonlinear evolution of cosmological power spectra
• COLOSSUS: A python toolkit for cosmology, large-scale structure, and dark matter halos
• An accurate halo model for fitting non-linear cosmological power spectra and baryonic feedback models
• Deriving the Nonlinear Cosmological Power Spectrum and Bispectrum from Analytic Dark Matter Halo Profiles and Mass Functions
• The Quijote simulations
• Massive Neutrinos and the Non-linear Matter Power Spectrum
• HMcode-2020: Improved modelling of non-linear cosmological power spectra with baryonic feedback
• The Coyote Universe I: Precision Determination of the Nonlinear Matter Power Spectrum
• Accurate halo-model matter power spectra with dark energy, massive neutrinos and modified gravitational forces
• The Coyote Universe II: Cosmological Models and Precision Emulation of the Nonlinear Matter Power Spectrum
• The Coyote Universe Extended: Precision Emulation of the Matter Power Spectrum
• The BACCO Simulation Project: Exploiting the full power of large-scale structure for cosmology
• The Coyote Universe III: Simulation Suite and Precision Emulator for the Nonlinear Matter Power Spectrum
• COSMOPOWER: emulating cosmological power spectra for accelerated Bayesian inference from next-generation surveys
• Euclid preparation: IX. EuclidEmulator2 -- Power spectrum emulation with massive neutrinos and self-consistent dark energy perturbations
• Emulators for the non-linear matter power spectrum beyond $\Lambda$CDM
• Efficient Cosmological Parameter Estimation with Hamiltonian Monte Carlo
• The BACCO simulation project: biased tracers in real space
• Exhaustive Symbolic Regression
• Machine learning constraints on deviations from general relativity from the large scale structure of the Universe
• Preparing for the Cosmic Shear Data Flood: Optimal Data Extraction and Simulation Requirements for Stage IV Dark Energy Experiments
• Kernel-Based Emulator for the 3D Matter Power Spectrum from CLASS
• Extending the Coyote emulator to dark energy models with standard $w_0$-$w_a$ parametrization of the equation of state
• The SZ flux-mass ($Y$-$M$) relation at low halo masses: improvements with symbolic regression and strong constraints on baryonic feedback
• On the functional form of the radial acceleration relation
• A precise symbolic emulator of the linear matter power spectrum
• Cosmic backreaction and the mean redshift drift from symbolic regression
• Machine learning cosmic backreaction and its effects on observations
• Priors for symbolic regression
• Machine learning cosmic inflation
• Optimal Inflationary Potentials
• Using machine learning to compress the matter transfer function $T(k)$
• Machine learning unveils the linear matter power spectrum of modified gravity
• Cosmological parameter constraints using phenomenological symbolic expressions: On the significance of symbolic expression complexity and accuracy

Cited by in corpus (11)

• Scant evidence for thawing quintessence
• CSST Cosmological Emulator I: Matter Power Spectrum Emulation with one percent accuracy to k = 10 h/Mpc
• A precise symbolic emulator of the linear matter power spectrum
• syren-new: Precise formulae for the linear and nonlinear matter power spectra with massive neutrinos and dynamical dark energy
• cp3-bench: A tool for benchmarking symbolic regression algorithms tested with cosmology
• Bayesian Multi-line Intensity Mapping
• Machine Learning-Based Analytical Expressions for Gray-Body Factors and Application to Primordial Black Holes
• The Velocity Field Olympics: Assessing velocity field reconstructions with direct distance tracers
• Interpretable and physics-informed emulator for the linear matter power spectrum from machine learning
• $S_8$ from peculiar velocities: agreement with Planck for Tully--Fisher and supernovae, tension for the fundamental plane
• Genetic algorithm demystified for cosmological parameter estimation