Robust parameter inference for Taiji via time-frequency contrastive learning and normalizing flows
General Relativity and Quantum Cosmology2026-04-16v1Cosmology and Nongalactic AstrophysicsHigh Energy Physics - PhenomenologyHigh Energy Physics - Theory
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.
@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}
}