astro-ph.COSep 8, 2026

Inductive Biases in Field-Level Cosmological Inference from Galaxy Catalogs

Authors: James O. BaldwinShy GenelFrancisco Villaescusa-Navarro

Abstract

We perform field-level likelihood-free inference of the matter density parameter ΩmΩ_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Ω_m from velocities alone with mean relative errors of approximately 18%18\% in-distribution and 25%\sim25\% OOD, with KANs and MLPs achieving comparable performance. In contrast, the same set-based approach does not yield useful σ8σ_8 predictions in either in-distribution or cross-suite tests. Adding positions does not improve Deep Sets, while GNNs infer ΩmΩ_m with mean relative errors of about 10%10\% in-distribution and 1010--17%17\% OOD. These results indicate that peculiar velocities provide the dominant source of ΩmΩ_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.

Explore similar work

Jun 22, 2026astro-ph.CO

Field-level weak lensing cosmology with 60 simulations using multifidelity simulation-based inference

We perform a realistic KiDS-Legacy mock analysis with field-level neural compression and simulation-based inference using just 60 NN-body simulations. The weak lensing shear field encodes substantially more cosmological information than standard two-point summary statistics such as the power spectrum. Field-level inference can fully exploit this information, but physical realism at the field-level requires very high-fidelity simulations. This poses a major challenge for simulation-based inference (SBI): accurate empirical density modelling and deep-learning-based neural compression require tens of thousands of training samples, but achieving physical realism at the field level makes each simulation extremely costly. We demonstrate that multifidelity SBI can alleviate this tension by substantially reducing the number of high-fidelity simulations needed for accurate cosmological inference. We pre-train neural inference models on realistic KiDS-Legacy-like shear mocks using fast log-normal \texttt{GLASS} simulations and fine-tune them on a small set of high-fidelity NN-body simulations. We show that 6060 high-fidelity simulations are sufficient to obtain informative and well-calibrated cosmological posteriors, enabling at least an order-of-magnitude reduction in simulation cost for accurate field-level inference in a realistic setting.
Alex A. Saoulis, Kiyam Lin, Niall Jeffrey +5
Jun 9, 2026astro-ph.CO

Interpretable Neural Marked Statistics for Cosmological Inference

Recovering cosmological information beyond the power spectrum is a central goal for upcoming cosmological surveys, since late-time non-Gaussian signal in the matter density cannot be accessed through two-point statistics alone. Marked statistics fold part of this information back into the two-point level by reweighting the field with non-linear functions. We propose a neural marking scheme to generalize this process through a set of interpretable, physically motivated transformations that directly allow to interpret the gain in cosmological information at the morphological level. We employ a contrastive learning objective to align learnable marked summaries with the underlying cosmological parameters. At kmax=0.2hMpc1k_{\max}=0.2\,h\mathrm{Mpc}^{-1}, our neural mark tightens the marginalized constraint on σ8σ_8 by 2.9×2.9\times and on ΩmΩ_m by 1.8×1.8\times compared to classical marks, breaking the Ωmσ8Ω_m-σ_8 degeneracy at the Fisher information level. It further reduces the parameter MSE across our cosmological parameter prior by 1.45×1.45\times over the best classical mark. The learned latent geometry aligns with the ΩmΩ_m and σ8σ_8 directions in parameter space, indicating that the contrastive objective recovers the dominant axes of cosmological information. Our approach opens the door to more powerful, interpretable summary statistics for cosmological inference.
Federico Semenzato, Benjamin D. Wandelt, Michele Liguori +1
May 20, 2026astro-ph.CO

Velocityformer: Broken-Symmetry-Matched Equivariant Graph Transformers for Cosmological Velocity Reconstruction

Precise measurement of the kinematic Sunyaev-Zel'dovich (kSZ) effect - a probe of the large-scale distribution of baryonic matter, a key observable for cosmological inference - requires accurate reconstruction of galaxy velocities from spectroscopic surveys. The signal-to-noise ratio (SNR) of kSZ measurements scales directly with the correlation coefficient rr between reconstructed and true velocities. We introduce Velocityformer, an equivariant graph transformer architecture designed to match the specific symmetry of the observational data. While the underlying physics is equivariant with respect to translations and rotations, observational effects break this symmetry due to the preferred line-of-sight direction. Matching the model's inductive bias to the data's broken symmetry consistently improves performance across all model sizes and training volumes, with Velocityformer improving rr by 35% over the standard linear theory baseline and outperforming ML baselines at every data volume. By matching the model's inductive bias to the data and conditioning on the physics-based long-wavelength solution, Velocityformer is highly data-efficient, training to high accuracy on as few as 4 low-fidelity simulations, and generalises zero-shot across input geometry, cosmological parameters, and galaxy sample. On high-fidelity simulated galaxy catalogues, this yields a 30% improvement in rr over the physical baseline, directly translating to the same SNR gain on observational data.
Tilman Tröster, David Mirkovic, Veronika Oehl +1