Inductive Biases in Field-level Cosmological Inference from Galaxy Catalogs
4 affiliations
- 1City University of New York Graduate Center
- 2Flatiron Institute
- 3Columbia University
- 4Princeton University
Abstract
We perform field-level likelihood-free inference of the matter density parameter Ω m from simulated galaxy catalogs using machine learning models with differing inductive biases. Using hydrodynamic simulations from CAMELS, we examine how observable choice and architecture govern cosmological information extraction. We consider galaxy positions and line-of-sight peculiar velocities, separately and jointly, and compare permutation-invariant deep sets, implemented with either multilayer perceptrons (MLPs) or Kolmogorov–Arnold networks (KANs), to graph neural networks (GNNs), which explicitly encode spatial relations. We test in-distribution and out-of-distribution (OOD) performance across simulations with different subgrid galaxy-formation prescriptions. Deep sets infer Ω m from velocities alone with mean relative errors of approximately 18% in-distribution and ∼25% OOD, with KANs and MLPs achieving comparable performance. In contrast, the same set-based approach does not yield useful σ ₈ predictions in either in-distribution or cross-suite tests. Adding positions does not improve deep sets, while GNNs infer Ω m with mean relative errors of about 10% in-distribution and 10%–17% OOD. These results indicate that peculiar velocities provide the dominant source of Ω m information for set-based models in this setting, while spatial information is most effectively used by architectures that explicitly encode galaxy–galaxy relations. Because the velocity inputs are exact simulated peculiar velocities, applications to survey data will require validation under realistic velocity-measurement noise, selection effects, and survey geometry.
Versions
Related
- Lens Modeling and Cosmological Inference from an Impure Sample of Galaxy-Galaxy Strong Lenses
- Lens Modeling and Cosmological Inference from an Impure Sample of Galaxy-Galaxy Strong Lenses
- Stress-Testing Small-Scale Cosmological Inference With Empirical Galaxy--Halo Connection Models
- The NIKA2 Cosmological Legacy Survey in COSMOS: Final 1.2,mm and 2,mm source catalogs and redshift distribution of dusty star-forming galaxies
- Cosmological implications of two mock TianQin standard siren catalogs generated under different background cosmological models
BibTeX
@article{baldwin2026inductive,
title = {{Inductive Biases in Field-level Cosmological Inference from Galaxy Catalogs}},
author = {James O. Baldwin and Shy Genel and Francisco Villaescusa-Navarro},
journal = {The Astrophysical Journal},
year = {2026},
month = oct,
doi = {10.3847/1538-4357/aea400},
url = {https://iopscience.iop.org/article/10.3847/1538-4357/aea400},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - Inductive Biases in Field-level Cosmological Inference from Galaxy Catalogs
AU - James O. Baldwin
AU - Shy Genel
AU - Francisco Villaescusa-Navarro
PY - 2026
DA - 2026/10/16
JO - The Astrophysical Journal
DO - 10.3847/1538-4357/aea400
UR - https://iopscience.iop.org/article/10.3847/1538-4357/aea400
AB - We perform field-level likelihood-free inference of the matter density parameter Ω m from simulated galaxy catalogs using machine learning models with differing inductive biases. Using hydrodynamic simulations from CAMELS, we examine how observable choice and architecture govern cosmological information extraction. We consider galaxy positions and line-of-sight peculiar velocities, separately and jointly, and compare permutation-invariant deep sets, implemented with either multilayer perceptrons (MLPs) or Kolmogorov–Arnold networks (KANs), to graph neural networks (GNNs), which explicitly encode spatial relations. We test in-distribution and out-of-distribution (OOD) performance across simulations with different subgrid galaxy-formation prescriptions. Deep sets infer Ω m from velocities alone with mean relative errors of approximately 18% in-distribution and ∼25% OOD, with KANs and MLPs achieving comparable performance. In contrast, the same set-based approach does not yield useful σ ₈ predictions in either in-distribution or cross-suite tests. Adding positions does not improve deep sets, while GNNs infer Ω m with mean relative errors of about 10% in-distribution and 10%–17% OOD. These results indicate that peculiar velocities provide the dominant source of Ω m information for set-based models in this setting, while spatial information is most effectively used by architectures that explicitly encode galaxy–galaxy relations. Because the velocity inputs are exact simulated peculiar velocities, applications to survey data will require validation under realistic velocity-measurement noise, selection effects, and survey geometry.
SN - 0004-637X
SN - 1538-4357
ER -