Accelerated, scalable and reproducible AI-driven gravitational wave detection

• Published in: Nature astronomy Vol. 5; no. 10; pp. 1062 - 1068
• Main Authors: Huerta, E. A., Khan, Asad, Huang, Xiaobo, Tian, Minyang, Levental, Maksim, Chard, Ryan, Wei, Wei, Heflin, Maeve, Katz, Daniel S., Kindratenko, Volodymyr, Mu, Dawei, Blaiszik, Ben, Foster, Ian
• Format: Journal Article
• Language: English
• Published: London Nature Publishing Group 01.10.2021
• Subjects: Artificial intelligence; Distributed processing; Gravitational waves
• ISSN: 2397-3366; 2397-3366
• DOI: 10.1038/s41550-021-01405-0

The development of reusable artificial intelligence (AI) models for wider use and rigorous validation by the community promises to unlock new opportunities in multi-messenger astrophysics. Here we develop a workflow that connects the Data and Learning Hub for Science, a repository for publishing AI models, with the Hardware-Accelerated Learning (HAL) cluster, using funcX as a universal distributed computing service. Using this workflow, an ensemble of four openly available AI models can be run on HAL to process an entire month’s worth (August 2017) of advanced Laser Interferometer Gravitational-Wave Observatory data in just seven minutes, identifying all four binary black hole mergers previously identified in this dataset and reporting no misclassifications. This approach combines advances in AI, distributed computing and scientific data infrastructure to open new pathways to conduct reproducible, accelerated, data-driven discovery.By combining a repository for artificial intelligence models and a supercomputing cluster, an entire month’s worth of advanced LIGO data is analysed in just 7 min, finding all binary black hole mergers previously identified in this dataset and reporting no misclassifications.