English

Robust parameter inference for Taiji via time-frequency contrastive learning and normalizing flows

General Relativity and Quantum Cosmology 2026-04-16 v1 Cosmology and Nongalactic Astrophysics High Energy Physics - Phenomenology High Energy Physics - Theory

Abstract

Transient noise artifacts, commonly referred to as glitches, pose a major challenge to parameter inference for space-based gravitational-wave (GW) observations. We develop a glitch-robust amortized inference framework for massive black hole binaries in the Taiji detector configuration by combining conditional normalizing flows, a time-frequency multimodal fusion encoder, and contrastive learning. To enable large-scale training on contaminated data, we further introduce a neural glitch generator that produces high-fidelity synthetic transients at substantially reduced computational cost. Systematic experiments show that, under glitch contamination, the proposed method yields more accurate and better-calibrated posteriors than a conventional Markov Chain Monte Carlo baseline. In ablation studies, the full time-frequency model with contrastive learning performs best overall and remains robust to variations in glitch duration and merger-relative timing. We further show that standard coverage diagnostics alone are insufficient to fully assess posterior fidelity. We therefore complement them with the continuous ranked probability score, which provides a stricter assessment of global distributional agreement in non-ideal GW data. Taken together, these results establish deep-learning-based amortized inference as a promising framework for fast and robust Bayesian parameter estimation in future space-based GW observations.

Keywords

Cite

@article{arxiv.2604.13867,
  title  = {Robust parameter inference for Taiji via time-frequency contrastive learning and normalizing flows},
  author = {Tian-Yang Sun and Bo Liang and Ji-Yu Song and Song-Tao Liu and Shang-Jie Jin and He Wang and Ming-Hui Du and Jing-Fei Zhang and Xin Zhang},
  journal= {arXiv preprint arXiv:2604.13867},
  year   = {2026}
}

Comments

15 pages, 8 figures

R2 v1 2026-07-01T12:10:45.248Z