asmaNet: A Staged Spectral Token-Refinement Architecture for Fast Gravitational-Wave Signal-to-Noise Ratio Prediction
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 -- samples. At samples, it achieves on the held-out test set, after rank-preserving monotonic (PCHIP) calibration, and () in the detection-relevant regime , with stable accuracy across independent dataset realizations. On a single GPU, it sustains approximately SNR evaluations per second and processes a -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 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
| Parameter | Symbol | Distribution | Range |
|---|---|---|---|
| Component masses | uniform ( ) | ||
| (derived) | — | ||
| Spin magnitudes | uniform | ||
| Spin polar angles | uniform | ||
| Spin azimuthal angles | uniform | ||
| Inclination | uniform |
| Layer | Operation | Output shape | Notes |
|---|---|---|---|
| Input | – | standardized parameter vector | |
| 1 | Linear | GELU Dropout ( ) | |
| 2 | Linear | GELU Dropout ( ) | |
| 3 | Linear | GELU Dropout ( ) | |
| 4 | Linear | GELU Dropout ( ) | |
| Output | Linear | predicts |
| Component | Operation | Output shape | Notes |
|---|---|---|---|
| Input | reshape | 11D feature vector | |
| Stem | Conv , , stride 1 BN ReLU | padding 1 | |
| Stage 1 | BasicBlock1D , | stride 1 | |
| Stage 2 | BasicBlock1D , | stride-2 in first block | |
| Stage 3 | BasicBlock1D , | stride-2 in first block | |
| Stage 4 | BasicBlock1D , | stride-2 in first block |
| Component | Operation | Output shape | Notes |
|---|---|---|---|
| Input | – | parameter vector | |
| Reshape | treat each feature as a token | ||
| Tokenization | Linear (per token) | scalar token embedding | |
| Position enc. | Learnable embeddings | added to tokens | |
| Encoder | Transformer layers | pre-LN; MHA: ; FFN: ; GELU; dropout | |
| Pooling | Mean over LayerNorm | global mean pooling + LN |
| Component | Operation | Output shape | Notes |
|---|---|---|---|
| Input | – | standardized parameter vector | |
| Tokenizer | Linear , reshape | learned token set | |
| -Stage ( ) | Enc. layer -Refractor | MHA , FFN , GELU; -Refractor ( ), softmax-gated residual; dropout | |
| -Feedback ( ) | Gated update recompute | learned gates parameterless top gate | |
| Readout | Per-token LN mean over | global mean pooling | |
| Reg. head | Linear | GELU; outputs |
| Architecture | 100k | 500k | 1M | |||
|---|---|---|---|---|---|---|
| Pre | Post | Pre | Post | Pre | Post | |
| asmaNet | 98.32 | 98.37 | 98.92 | 99.16 | 98.73 | 99.29 |
| Transformer | ||||||
| ResNet | ||||||
| MLP | ||||||
| Setting | Architecture | Pre | Post |
|---|---|---|---|
| 100k (seed-change check) | |||
| Train seed 2024 | asmaNet (Refractor OFF) | ||
| Train seed 2024 | Transformer | ||
| Train seed 2024 | ResNet | ||
| Train seed 2024 | MLP | ||
| 1M (dataset-realization check) | |||
| Configuration | 100k | 500k | 1M | |||
|---|---|---|---|---|---|---|
| Pre | Post | Pre | Post | Pre | Post | |
| -Refractor ON | ||||||
| -Refractor OFF | ||||||
| Architecture | Params | Relative | Key sizing |
|---|---|---|---|
| MLP | 201 k | width , depth | |
| Transformer | 540 k | , , , | |
| asmaNet | 732 k | , , , , | |
| ResNet (1D) | 3.98 M | width , 4 stages 2 blocks |
| asmaNet variant | Accuracy | Cross-entropy |
|---|---|---|
| -Refractor ON | ||
| -Refractor OFF | – |
| asmaNet variant | LogLoss | Accuracy | ROC-AUC |
|---|---|---|---|
| Default (Refractor ON ) |
Gravitational-Wave Astronomy
Gravitational-wave astronomy uses ripples in spacetime to detect cosmic events, measure compact objects, and test gravity.
See all 21 papers