cs.LGNov 3, 2026

ΦΦasmaNet: A Staged Spectral Token-Refinement Architecture for Fast Gravitational-Wave Signal-to-Noise Ratio Prediction

Authors: Ioannis Tsioulis, Anastasios Tefas, Nikolaos Stergioulas

Organizations: Department of Physics, Aristotle University of Thessaloniki, Greece · School of Computing, Mediterranean College (University of Derby), Athens, Greece · Department of Informatics, Aristotle University of Thessaloniki, Greece

Abstract

Quantifying selection effects in gravitational-wave population studies requires millions of costly signal-to-noise ratio (SNR) calculations. We present ΦΦasmaNet, a compact neural architecture for structured tabular regression, with staged spectral refinement as an explicit architectural prior. Learned input tokens are refined through ordered attention-based ΦΦ-Stages, each augmented by a ΦΦ-Refractor subspace-mixing module, then revisited through gated top-down ΦΦ-Feedback. Applied to the 11-dimensional parameters of precessing, higher-mode binary black hole signals (IMRPhenomXPHM), it predicts phase-averaged optimal SNR for a single detector at fixed Advanced LIGO design sensitivity. With approximately 732k parameters, it outperforms MLP, 1D-ResNet and Transformer-encoder baselines under matched training across 10510^5--10610^6 samples. At 10610^6 samples, it achieves 100−MAPE=98.73%100-\mathrm{MAPE}=98.73\% on the held-out test set, 99.29%99.29\% after rank-preserving monotonic (PCHIP) calibration, and 99.39%99.39\% (R2=0.9998R^2=0.9998) in the detection-relevant regime ρ>8ρ>8, with stable accuracy across independent dataset realizations. On a single GPU, it sustains approximately 1.2×1051.2\times10^5 SNR evaluations per second and processes a 10610^6-injection catalogue end-to-end in approximately 8.3 s, about two orders of magnitude faster than conventional waveform-and-inner-product calculations on a 16-core CPU. Exact distance rescaling through ρ∝1/dLρ\propto1/d_L supports repeated catalogue-wide detectability evaluations. The fully documented pipeline is intended as a reproducible benchmark, while results on public tabular datasets indicate competitiveness beyond gravitational-wave applications. Although demonstrated for a fixed single-detector configuration, the construction supports extensions to detector networks and epoch-specific sensitivities. (Abridged)

Figures & tables

Explore similar work

Jul 1, 2026

AI-enabled gravitational-waves searches for binary neutron stars at optimal sensitivity

Gravitational Waves (GWs) represent the newest window of astronomy, furthering our understanding of compact objects like black holes and neutron stars in the Universe. The signal from two merging neutron stars is especially interesting since it brings the prospect of concordant electromagnetic and neutrino emissions. Such multi-messenger observations have a transformational impact on fundamental physics, nuclear matter, astrophysics, and gravity. It was first witnessed in 2017 with the detection of the binary neutron star (BNS) merger GW170817. However, searching for BNS signals in real-time in the LIGO-Virgo-KAGRA (LVK) GW detectors presents a computational challenge, as the data streaming out must be matched against ∼\sim million reference waveforms, which requires up to a thousand CPU cores. We present a different approach using neural networks to learn the presence of a signal in the data. Our algorithm, called Aframe, was deployed in the LVK's fourth observing run and was the first artificial intelligence (AI)-enabled search to detect multiple binary black holes (BBHs) live. In this work, we demonstrate that the approach extends to the lower-mass BNS regime, and is the first AI-enabled search that achieves sensitivity comparable to matched-filter pipelines at lower computational and latency costs. The challenge of the longer-duration BNS signals is addressed by heterodyning the data, following which the network architecture used for BBHs is sufficient to distinguish signal versus background. We also show that this analysis requires a single non-flagship GPU for online deployment. Furthermore, the design and adoption of inference-as-a-service tools allow rapid offline analysis using a distributed pool of GPU resources. Hence, aside from the use case of rapid online data analysis, we also establish the use of Aframe for efficient archival data analysis.
Sep 7, 2026

Survey of Novel Deep Learning Architectures for Denoising Gravitational-wave Signals

Gravitational-wave denoising must handle the full diversity of spinning, precessing binaries, since the recovered waveform underpins parameter estimation, tests of general relativity, and population studies. Matched filtering achieves this at a cost that becomes prohibitive as next-generation detectors push event rates higher; deep learning offers real-time reconstruction, but current methods are developed on narrow parameter spaces, precluding principled comparison and reliable deployment. We present the first controlled comparison of five neural-network architectures for gravitational-wave denoising, trained identically across the full astrophysically-motivated spinning binary-black-hole parameter space. A unifying principle emerges: matching network structure to the spectral anatomy of a coalescence -- inspiral, merger, ringdown -- outperforms brute-force model scaling. Our Multi-Scale Frequency-Aware architecture embodies this via dedicated parallel branches per frequency regime, achieving the best fidelity while using fewer parameters than larger models. It generalizes from simulated training to real LIGO-Virgo-KAGRA data without retraining, recovering merger morphology across confirmed events spanning three observing runs and both detectors, despite training on a single detector's simulated noise. We construct population-level uncertainty bands from denoising residuals, validate their calibration, and stress-test the framework on extended mass ratios and a spin population resembling hierarchical-merger remnants. Applied to real noise with no known signal, the network suppresses its output almost everywhere, with rare exceptions traced to noise transients rather than a general weakness -- indicating the statistic discriminates signal from noise and motivating a future detection study. Released weights give a deployable, reproducible benchmark for future extensions.
May 4, 2026

Testing General Relativity Through Gravitational Wave Classification: A Convolutional Neural Network Framework

We present a machine learning framework for testing general relativity (GR) with gravitational wave signals from binary black hole mergers. Using the source parameters of 173 BBH events from the GWTC catalog as a realistic astrophysical population, we generate simulated GR waveforms and construct beyond GR (BGR) waveforms by applying controlled phase deformations. We introduce a response function formalism that provides a systematic framework for quantifying how any observable responds to modifications of GR. We train convolutional neural networks (CNNs) on two input representations: whitened waveforms and a response function type observable derived from the waveform mismatch, which isolates the effect of phase deviations from the bulk signal. Using response functions as the CNN input improves the classification sensitivity by a factor of approximately 33 compared to whitened waveforms, demonstrating that the choice of observable representation is as important as the classifier architecture. We study the fundamental limits of this classification through Bayes optimal error analysis, averaging methods that reveal coherent patterns hidden in noise, and a comparison between CNN accuracy and a single feature classifier as a proxy for human performance. At all deformation scales, the CNN outperforms the best single feature approach. We extend the framework to physically motivated theories using the parameterized post Einsteinian (ppE) formalism and apply it to massive gravity, where the classifier detects deviations for graviton masses of order mg∼10−23  eV/c2m_g \sim 10^{-23}\;\mathrm{eV}/c^2 with aLIGO design sensitivity.