Research News

Osaka Metropolitan University researchers introduced a technique to enhance the reliability of gravitational wave parameter estimation results produced by machine learning. The team developed two independent machine learning models based on the Vision Transformer to estimate effective spin and chirp mass from spectrograms of gravitational wave signals from binary black hole mergers.

To enhance the reliability of these models, the researchers utilized attention maps to visualize the areas the models focus on when making predictions. This approach enabled the team to demonstrate that both models perform parameter estimation based on physically meaningful information. Furthermore, by leveraging these attention maps, the team demonstrated a method to quantify the impact of glitches on parameter estimation. They showed that as the models focus more on glitches, the parameter estimation results become more strongly biased. This suggests that attention maps could potentially be used to distinguish between cases where the results produced by the machine learning model are reliable and cases where they are not.

Paper information

Journal: Physical Review D
Title: Enhancing the reliability of machine learning for gravitational wave parameter estimation with attention-based models
DOI: 10.1103/blfk-7k9f
Authors: Hibiki Iwanaga, Mahoro Matsuyama, and Yousuke Itoh
Published:  9 October 2025
URL: https://doi.org/10.1103/blfk-7k9f

Contact

Yousuke Itoh
Graduate School of Science
Email: yousuke.itoh[at]omu.ac.jp

*Please change [at] to @.