Machine Learning Closure Audits for LSST Photometric Supernova Cosmology
Ayan Mitra
astro-ph.CO, astro-ph.IM
Submitted: 2026-07-07
Comments: 19 pages, 6 figures, 9 tables
License: http://creativecommons.org/licenses/by/4.0/
The gist: Modern and next generation supernova cosmology analyses rely on end to end simulations to train photometric classifiers, characterise selection effects, validate light curve models, and calibrate
Terminology
Abstract
Modern and next generation supernova cosmology analyses rely on end to end simulations to train photometric classifiers, characterise selection effects, validate light curve models, and calibrate distance bias corrections. Standard closure tests based on global Hubble diagram summaries can miss multivariate structure that remains in post correction residuals. We introduce a supervised machine learning closure audit that tests whether measured observables can predict the bias corrected Hubble residuals mu. We apply the audit to LSST Type Ia supernova simulations from two independent analyses: the M23 mock data sets of Mitra et al. (2023), with spectroscopic redshift and photometric redshift samples, and the predominantly photometric LSST like M25 simulation of Mitra et al. (2025). Standard one dimensional Redshift binned diagnostics explain <1% of the residual variance (R squared < 0.01), suggesting apparent closure. In contrast, out of fold LightGBM models recover up to 98.2% of the variance in the simulated residuals (R squared = 0.982), revealing structured residual predictability. Applying the same audit directly to the real Dark Energy Survey 5 Year (DES 5YR) spectroscopic sample yields a held out R squared = 0.725. SHAP feature attribution rankings are highly consistent between independently trained M25 and DES models (Spearman rho = 0.802), with apparent magnitudes, signal to noise ratios, and redshift dominating the shared predictive hierarchy. The resulting scorecard provides a diagnostic protocol for comparing mock ensembles and real observations, identifying residual non closure before cosmological parameters are unblinded.
Sources
- Results of the Photometric LSST Astronomical Time-series Classification Challenge (PLAsTiCC)
- Cosmology Constraints from Type Ia Supernova Simulations of the Nancy Grace Roman Space Telescope Strategy Recommended by the High Latitude Time Domain Survey Definition Committee
- Euclid Definition Study Report
- The LSST Dark Energy Science Collaboration (DESC) Science Requirements Document
- LSST Science Book, Version 2.0
- Revisiting Classifier Two-Sample Tests
- A Fully Photometric Approach to Type Ia Supernova Cosmology in the LSST Era: Host Galaxy Redshifts and Supernova Classification
- Union Through UNITY: Cosmology with 2,000 SNe Using a Unified Bayesian Framework
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