Transformers with Physics-Informed Encodings and Simulation-Based Inference for Robust Detection of Eccentric Binary Black Holes in Pulsar Timing Array Data
cs.LG, astro-ph.IM, gr-qc
Submitted: 2026-07-04
Updated: 2026-09-28
Comments: 24 pages, 7 figures, 4 tables
Code: https://github.com/subhajitphy/PIPE-GWs-PTA
License: http://creativecommons.org/licenses/by/4.0/
The gist: Pulsar timing arrays (PTAs) provide a unique window into nanohertz gravitational waves (GWs), but extracting astrophysical parameters from noisy, long-baseline timing residuals remains
Terminology
Abstract
Pulsar timing arrays (PTAs) provide a unique window into nanohertz gravitational waves (GWs), but extracting astrophysical parameters from noisy, long-baseline timing residuals remains computationally challenging with traditional Bayesian techniques due to the high dimensionality of the parameter space, complex and correlated noise models, and the cost of repeated likelihood evaluations. We introduce a Transformer with a physics-informed positional-encoding framework for the efficient inference of eccentric binary black holes in relativistic orbits from PTA data. Our approach embeds analytical GW phase evolution directly into the model through structured positional encodings, enabling the network to learn physically meaningful representations from raw PTA timing residuals. We then use generative models, including discrete and continuous conditional normalizing flows, to infer posterior distributions within a simulation-based inference framework. Across a range of signal-to-noise ratios, the proposed method achieves improved accuracy, sharper posteriors, and faster inference compared to physics-agnostic baselines. While presented for deterministic white-noise signals, the modular framework readily generalizes to realistic PTA analyses incorporating red noise and additional components. This work highlights the potential of physics-aware deep learning models as scalable alternatives to conventional inference pipelines for next-generation PTA datasets.
Sources
- Observation of Gravitational Waves from a Binary Black Hole Merger
- The second data release from the European Pulsar Timing Array III. Search for gravitational wave signals
- The International Pulsar Timing Array: Second data release
- Comparing recent PTA results on the nanohertz stochastic gravitational wave background
- A Practical Theorem on Gravitational Wave Backgrounds
- The NANOGrav 15-year Data Set: Bayesian Limits on Gravitational Waves from Individual Supermassive Black Hole Binaries
- The international pulsar timing array project: using pulsars as a gravitational wave detector
- Deep Learning for Real-time Gravitational Wave Detection and Parameter Estimation: Results with Advanced LIGO Data
- Matching matched filtering with deep networks in gravitational-wave astronomy
- Deep Residual Networks for Gravitational Wave Detection
- Pre-trained Audio Transformer as a Foundational AI Tool for Gravitational Waves
- Bayesian parameter estimation using conditional variational autoencoders for gravitational-wave astronomy
- Machine Learning for Nanohertz Gravitational Wave Detection and Parameter Estimation with Pulsar Timing Array
- Attention Is All You Need
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
- Transformers in Time Series: A Survey
- The Universe is worth $64^3$ pixels: Convolution Neural Network and Vision Transformers for Cosmology
- Transformer models for astrophysical time series and the GRB prompt-afterglow relation
- End-to-End Population Inference from Gravitational-Wave Strain using Transformers
- Vision Transformer for Transient Noise Classification
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks