STAR NRE: Solving supernova selection effects with set-based truncated auto-regressive neural ratio estimation
arXiv:2409.03837 · doi:10.1088/1475-7516/2025/07/031
Abstract
Accounting for selection effects in supernova type Ia (SN Ia) cosmology is crucial for unbiased cosmological parameter inference -- even more so for the next generation of large, mostly photometric-only surveys. The conventional "bias correction" procedure has a built-in systematic bias towards the fiducial model used to derive it and fails to account for the additional Eddington bias that arises in the presence of significant redshift uncertainty. On the other hand, likelihood-based analyses within a Bayesian hierarchical model, e.g. using MCMC, scale poorly with the data set size and require explicit assumptions for the selection function that may be inaccurate or contrived. To address these limitations, we introduce STAR NRE, a simulation-based approach that makes use of a conditioned deep set neural network and combines efficient high-dimensional global inference with subsampling-based truncation in order to scale to very large survey sizes while training on sets with varying cardinality. Applying it to a simplified SN Ia model consisting of standardised brightnesses and redshifts with Gaussian uncertainties and a selection procedure based on the expected LSST sensitivity, we demonstrate precise and unbiased inference of cosmological parameters and the redshift evolution of the volumetric SN Ia rate from ~100 000 mock SNae Ia. Our inference procedure can incorporate arbitrarily complex selection criteria, including transient classification, in the forward simulator and be applied to complex data like light curves. We outline these and other steps aimed at integrating STAR NRE into an end-to-end simulation-based pipeline for the analysis of future photometric-only SN Ia data.
published in JCAP; 25 pages, 6 figures (+ appendices)
References in corpus (59)
- Adam: A Method for Stochastic Optimization
- SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
- emcee: The MCMC Hammer
- dynesty: A Dynamic Nested Sampling Package for Estimating Bayesian Posteriors and Evidences
- Improved cosmological constraints from a joint analysis of the SDSS-II and SNLS supernova samples
- SALT2: using distant supernovae to improve the use of Type Ia supernovae as distance indicators
- SNANA: A Public Software Package for Supernova Analysis
- SALT: a Spectral Adaptive Light curve Template for Type Ia Supernovae
- Distance measures in cosmology
- Correcting Type Ia Supernova Distances for Selection Biases and Contamination in Photometrically Identified Samples
- Fast likelihood-free cosmology with neural density estimators and active learning
- SuperNNova: an open-source framework for Bayesian, Neural Network based supernova classification
- Simulations of the WFIRST Supernova Survey and Forecasts of Cosmological Constraints
- Type Ia Supernova Rate Measurements to Redshift 2.5 from CANDELS : Searching for Prompt Explosions in the Early Universe
- Type Ia Supernova Light Curve Inference: Hierarchical Models in the Optical and Near Infrared
- Type-Ia Supernova Rates to Redshift 2.4 from CLASH: the Cluster Lensing And Supernova survey with Hubble
- SALT3: An Improved Type Ia Supernova Model for Measuring Cosmic Distances
- Models and Simulations for the Photometric LSST Astronomical Time Series Classification Challenge (PLAsTiCC)
- Type Ia Supernova Light Curve Inference: Hierarchical Bayesian Analysis in the Near Infrared
- A Measurement of the Rate of type-Ia Supernovae at Redshift 0.1 from the First Season of the SDSS-II Supernova Survey
- Nuisance hardened data compression for fast likelihood-free inference
- Avocado: Photometric Classification of Astronomical Transients with Gaussian Process Augmentation
- Likelihood-Free Cosmological Inference with Type Ia Supernovae: Approximate Bayesian Computation for a Complete Treatment of Uncertainty
- First Cosmology Results using Type Ia Supernovae from the Dark Energy Survey: The Effect of Host Galaxy Properties on Supernova Luminosity
- UNITY: Confronting Supernova Cosmology's Statistical and Systematic Uncertainties in a Unified Bayesian Framework
- Improved constraints on cosmological parameters from SNIa data
- Improved Treatment of Host-Galaxy Correlations in Cosmological Analyses With Type Ia Supernovae
- BAHAMAS: new SNIa analysis reveals inconsistencies with standard cosmology
- Expectation propagation as a way of life: A framework for Bayesian inference on partitioned data
- Union Through UNITY: Cosmology with 2,000 SNe Using a Unified Bayesian Framework
- Fast and Credible Likelihood-Free Cosmology with Truncated Marginal Neural Ratio Estimation
- Hierarchical Bayesian inference of galaxy redshift distributions from photometric surveys
- From Images to Dark Matter: End-To-End Inference of Substructure From Hundreds of Strong Gravitational Lenses
- ASTROMER: A transformer-based embedding for the representation of light curves
- SICRET: Supernova Ia Cosmology with truncated marginal neural Ratio EsTimation
- Dark energy by natural evolution: Constraining dark energy using Approximate Bayesian Computation
- Type Ia Supernovae: Colors, Rates, and Progenitors
- Steve: A hierarchical Bayesian model for Supernova Cosmology
- zBEAMS: A unified solution for supernova cosmology with redshift uncertainties
- Likelihood-free Inference with Mixture Density Network
- Application of Bayesian graphs to SN Ia data analysis and compression
- CoLFI: Cosmological Likelihood-free Inference with Neural Density Estimators
- Scalable inference with Autoregressive Neural Ratio Estimation
- Debiasing Standard Siren Inference of the Hubble Constant with Marginal Neural Ratio Estimation
- Test of Artificial Neural Networks in Likelihood-free Cosmological Constraints: A Comparison of IMNN and DAE
- SIDE-real: Supernova Ia Dust Extinction with truncated marginal neural ratio estimation applied to real data
- Probing the Consistency of Cosmological Contours for Supernova Cosmology
- Amortized Bayesian Inference for Supernovae in the Era of the Vera Rubin Observatory Using Normalizing Flows
- Detection is truncation: studying source populations with truncated marginal neural ratio estimation
- Accounting for Selection Effects in Supernova Cosmology with Simulation-Based Inference and Hierarchical Bayesian Modelling
- HNPE: Leveraging Global Parameters for Neural Posterior Estimation
- Hierarchical Neural Simulation-Based Inference Over Event Ensembles
- SimSIMS: Simulation-based Supernova Ia Model Selection with thousands of latent variables
- Fishnets: Information-Optimal, Scalable Aggregation for Sets and Graphs
- Improved Weak Lensing Photometric Redshift Calibration via StratLearn and Hierarchical Modeling
- Compositional Score Modeling for Simulation-based Inference
- Positional Encodings for Light Curve Transformers: Playing with Positions and Attention
- Time Delay Cosmography with a Neural Ratio Estimator
- GPU-Accelerated Hierarchical Bayesian Inference with Application to Modeling Cosmic Populations: CUDAHM