The Generalization Gap in Machine Learning EoS Inference from Core-Collapse Supernova Gravitational Waves
Ayan Mitra
astro-ph.HE, nucl-th
Submitted: 2026-07-07
Comments: 12 pages, 11 figures, 4 tables
License: http://creativecommons.org/licenses/by/4.0/
The gist: Core-collapse supernova gravitational waves may carry information about the dense matter equation of state (EoS), which describes the relation between pressure, density, temperature, and composition.
Terminology
Abstract
Core-collapse supernova gravitational waves may carry information about the dense matter equation of state (EoS), which describes the relation between pressure, density, temperature, and composition. This work tests a crucial question for physical inference: can a machine learning model trained on a finite simulation catalogue predict EoS parameters for an EoS family that was absent during training? Under standard random cross-validation, a LightGBM regressor appears highly successful, yielding R 2=(0.70,0.67,0.60) for the nuclear incompressibility, symmetry energy, and slope parameter (K 0,J,L). However, under Leave-One-EoS-Out (LOEO) validation, where all waveforms from a single EoS are withheld, the model fails, yielding mean absolute errors of (44.57,3.19,30.54) MeV and negative pooled R squared scores, performing worse than a baseline mean predictor. This generalisation gap persists across linear models, random forests, neural networks, and gradient-boosted trees. Restricting inputs to physical features (bounce amplitude, bounce width, peak frequency) reduces template leakage, the memorisation of related templates shared across random splits, but does not restore reliable EoS extrapolation. In contrast, a progenitor mass case study shows that classification generalises to unseen rotation speeds, while continuous mass regression compresses predictions towards the catalogue interior. These results demonstrate that while machine learning successfully interpolates within current waveform catalogues, this does not imply robust physical inference for unseen EoS models. Future pipelines should adopt leave-family-out validation, wider simulation coverage, and physics-aware inference frameworks.
Sources
- An Exploration of the Equation of State Dependence of Core-Collapse Supernova Explosion Outcomes and Signatures
- Inferring physical properties of stellar collapse by third-generation gravitational-wave detectors
- Classifying Core-Collapse Supernova Gravitational Waves using Supervised Contrastive Learning
- Toward More Realistic Machine-Learning Inference of the Dense-Matter Equation of State from Supernova Gravitational Waves
- Parameter Estimation Horizon of Core-Collapse Supernovae with Current and Next-Generation Gravitational-Wave Detectors
- Leakage and the Reproducibility Crisis in ML-based Science
- Multivariate Regression Analysis of Gravitational Waves from Rotating Core Collapse
- Classifying the Equation of State from Rotating Core Collapse Gravitational Waves with Deep Learning
- Determine the Core Structure and Nuclear Equation of State of Rotating Core-Collapse Supernovae with Gravitational Waves by Convolutional Neural Networks
- Probing Supernovae through gravitational wave entropy
Related papers
- Numerical Studies of Accretion Flows onto a Neutron Star Engulfed in a Massive Star
- Collisionless Accretion of Finite-Angular-Momentum Plasma onto a Spinning Black Hole
- Impact of Magnetic Field Topology on Electromagnetic and Gravitational Waves from Binary Neutron Star Merger Remnants
- XRISM Resolve Spectroscopy of GX 5-1: Constraints on Iron Spectral Features in a Luminous Neutron-Star Binary
- SN 1006: A Cosmic Laboratory for Investigating Shock Acceleration Physics
- Neutrino Spectral Pinching in 3D Core-Collapse Supernovae: Late-Time Convergence, Failed-Explosion Signatures, and Viewing-Angle Dispersion