Differentiable Predictions for Large Scale Structure with SHAMNet

Abstract

In simulation-based models of the galaxy-halo connection, theoretical predictions for galaxy clustering and lensing are typically made based on Monte Carlo realizations of a mock universe. In this paper, we use Subhalo Abundance Matching (SHAM) as a toy model to introduce an alternative to stochastic predictions based on mock population, demonstrating how to make simulation-based predictions for clustering and lensing that are both exact and differentiable with respect to the parameters of the model. Conventional implementations of SHAM are based on iterative algorithms such as Richardson-Lucy deconvolution; here we use the JAX library for automatic differentiation to train SHAMNet, a neural network that accurately approximates the stellar-to-halo mass relation (SMHM) defined by abundance matching. In our approach to making differentiable predictions for large scale structure, we map parameterized PDFs onto each simulated halo, and calculate gradients of summary statistics of the galaxy distribution by using autodiff to propagate the gradients of the SMHM through the statistical estimators used to measure one- and two-point functions. Our techniques are quite general, and we conclude with an overview of how they can be applied in tandem with more complex, higher-dimensional models, creating the capability to make differentiable predictions for the multi-wavelength universe of galaxies.

11 pages, 6 appendices, version accepted for publication by the Open Journal of Astrophysics

References in corpus (16)

• The Effective Field Theory of Cosmological Large Scale Structures
• The star formation histories of galaxies in the Sloan Digital Sky Survey
• KiDS-450 + 2dFLenS: Cosmological parameter constraints from weak gravitational lensing tomography and overlapping redshift-space galaxy clustering
• emcee v3: A Python ensemble sampling toolkit for affine-invariant MCMC
• Dark Matter Substructure in Numerical Simulations: A Tale of Discreteness Noise, Runaway Instabilities, and Artificial Disruption
• The Excursion Set Theory of Halo Mass Functions, Halo Clustering, and Halo Growth
• A 2.5% measurement of the growth rate from small-scale redshift space clustering of SDSS-III CMASS galaxies
• Corrfunc --- A Suite of Blazing Fast Correlation Functions on the CPU
• The Mira-Titan Universe II: Matter Power Spectrum Emulation
• Void Statistics in Large Galaxy Redshift Surveys: Does Halo Occupation of Field Galaxies Depend on Environment?
• AbacusHOD: A highly efficient extended multi-tracer HOD framework and its application to BOSS and eBOSS data
• Cosmic Calibration
• The cosmology dependence of galaxy clustering and lensing from a hybrid $N$-body-perturbation theory model
• Predicting Galaxy Star Formation Rates via the Co-evolution of Galaxies and Halos
• Cosmological cross-correlations and nearest neighbor distributions
• Cosmological inference from the emulator based halo model II: Joint analysis of galaxy-galaxy weak lensing and galaxy clustering from HSC-Y1 and SDSS

Cited by in corpus (7)

• Stringent ${\sigma }_8$ constraints from small-scale galaxy clustering using a hybrid MCMC+emulator framework
• Diffstar: A Fully Parametric Physical Model for Galaxy Assembly History
• DSPS: Differentiable Stellar Population Synthesis
• OpenUniverse2024: A shared, simulated view of the sky for the next generation of cosmological surveys
• Interpretable Uncertainty Quantification in AI for HEP
• pop-cosmos: Insights from generative modeling of a deep, infrared-selected galaxy population
• Analytic auto-differentiable $\Lambda$CDM cosmography