Simulation-Based Inference for Cluster Cosmology with Set-Based Neural Network Architectures
S. Zelmer, E. Bulbul, K. Lehman, S. Krippendorf, E. Artis, S. Grandis, N. Clerc, Z. Ding, L. Fiorino, V. Ghirardini, M. Kluge, N. Malavasi, A. Merloni, T. Mistele, K. Nandra, M. E. Ramos-Ceja, J. S. Sanders, F. Pacaud, A. von der Linden, J. Weller, X. Zhang
astro-ph.CO
Submitted: 2026-06-25
Comments: 18 pages, 11 figures, submitted to A&A
License: http://creativecommons.org/licenses/by/4.0/
The gist: The unprecedented statistical power of galaxy cluster catalogs from the SRG (Spectrum Roentgen Gamma)/eROSITA All-Sky Survey provides a unique opportunity to place stringent constraints on
Terminology
Abstract
The unprecedented statistical power of galaxy cluster catalogs from the SRG (Spectrum Roentgen Gamma)/eROSITA All-Sky Survey provides a unique opportunity to place stringent constraints on cosmological models through measurements of structure growth. Fully exploiting the potential of these large X-ray-selected cluster samples, however, requires robust statistical frameworks that accurately connect observable quantities to the underlying cosmological parameters. We develop and implement a simulation-based inference (SBI) framework for cosmological parameter estimation using a realistic mock-generation pipeline calibrated on eRASS1 simulations. Synthetic galaxy cluster catalogs are propagated through the survey selection function to produce mock eRASS1 observations that reproduce the data's statistical properties. At the core of the method lies a set-based neural network (GNN on sets) that encodes information from individual clusters and is coupled to a masked autoregressive flow for flexible posterior density estimation. This approach enables the use of the full cluster-level information content without compressing the observables into binned summary statistics. Our framework recovers the input cosmologies within the inferred uncertainties, and passes calibration tests, demonstrating robustness in the presence of realistic survey effects. We obtain mock constraints of 11.5% on m and 4.4% on sigma 8 averaged over a suite of simulated cluster catalogs matching the effective sample size of the data set (3,259 clusters). We achieve a precision comparable to that obtained with traditional MCMC analyses based on substantially larger cluster samples. The framework is readily extensible to more complex forward models and additional observables. This work highlights the potential of SBI methods for next-generation large-scale structure analyses with forthcoming X-ray cluster surveys.
Sources
- Cosmology from Galaxy Redshift Surveys with PointNet
- Cosmological constraints from the Planck cluster catalogue with DES shear profiles and Chandra observations
- A Comparative Review of Dimension Reduction Methods in Approximate Bayesian Computation
- Graph Ordering Attention Networks
- Constructing Summary Statistics for Approximate Bayesian Computation: Semi-automatic ABC
- NANCY: Next-generation All-sky Near-infrared Community surveY
- Going Deeper into Permutation-Sensitive Graph Neural Networks
- Learning Summary Statistic for Approximate Bayesian Computation via Deep Neural Network
- Variational Inference with Normalizing Flows
- On permutation-invariant neural networks
- Flexible statistical inference for mechanistic models of neural dynamics
- Learnable Commutative Monoids for Graph Neural Networks
- Fast $\epsilon$-free Inference of Simulation Models with Bayesian Conditional Density Estimation
- Normalizing Flows for Probabilistic Modeling and Inference
- Masked Autoregressive Flow for Density Estimation
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
- The SRG/eROSITA all-sky survey: X-ray scaling relations of galaxy groups and clusters in the western Galactic hemisphere
- Domain Adaptive Graph Neural Networks for Constraining Cosmological Parameters Across Multiple Data Sets
- Validating Bayesian Inference Algorithms with Simulation-Based Calibration
Related papers
- Angular clustering and bias of photometric quasars in the Kilo-Degree Survey Data Release 4
- A Novel kinetic Sunyaev-Zel'dovich Estimator for Electron-Electron Correlations
- Magnetic fields at the dawn of structure formation I. The CARLA J1510+5958 proto-cluster
- Dark Energy Survey Year 6 Results: Weak Lensing and Galaxy Clustering Cosmological Analysis Framework
- Exploring the Impact of Systematic Bias in Type Ia Supernova Cosmology Across Diverse Dark Energy Parametrizations
- Non-Gaussian Galaxy Stochasticity and the Noise-Field Formulation