AI finds merging black holes

An interdisciplinary team develops a neural network to improve the interpretation of gravitational wave data

An interdisciplinary team from the Max Planck Institute for Intelligent Systems and the Max Planck Institute for Gravitational Physics has developed an algorithm that immediately checks its own calculations of merging black holes’ properties and corrects its result if necessary – inexpensively and rapidly. The machine learning method provides very accurate information about the observed gravitational waves and will be ready for use when the global network of gravitational-wave detectors starts its next observing run in May.

When two black holes merge, they emit gravitational waves that race through space and time at the speed of light. When these reach Earth, large detectors in the United States (Ligo), Italy (Virgo) and Japan (Kagra) can detect the signals. By comparing against theoretical predictions, scientists can then determine the black holes' properties: masses, spins, orientation, position in the sky and distance from Earth. A team of researchers from the Department of Empirical Inference at the Max Planck Institute for Intelligent Systems and the Department of Astrophysical and Cosmological Relativity at the Max Planck Institute for Gravitational Physics has now developed a self-checking deep learning system that very accurately extracts information from gravitational-wave data. In the process, the system checks its own predictions about the parameters of merging black holes – a deep neural network with a safety net. A set of 42 detected gravitational waves from merging black holes were successfully analyzed by the algorithm: When cross-checked against computationally expensive standard algorithms, the results were indistinguishable. The study was published on April 26, 2023 in the journal Physical Review Letters.

Dingo: a deep neural network for gravitational-wave analysis

The researchers have developed a deep neural network called Dingo (Deep Inference for Gravitational-wave Observations) to analyze the data. Dingo has been trained to extract – or infer – the gravitational-wave source parameters from the detector data. The network learned to interpret real (observed) gravitational-wave data after training with many millions of simulated signals in different configurations.

Trust, but verify

However, it is not possible to tell at first glance whether the deep neural network is reading the information correctly. Indeed, one disadvantage of common deep learning systems is that their results sound plausible even when they are wrong. That's why the researchers at both institutes have added a control feature to the algorithm. Maximilian Dax, doctoral student at the Max Planck Institute for Intelligent Systems and first author of the publication explains, “We have developed a network with a safety net. First, the algorithm calculates the properties of the black holes from the measured gravitational-wave signal. Based on these calculated parameters, a gravitational wave is modeled, and then compared to the originally observed signal. The deep neural network can thus cross-check its own results and correct them in case of doubt.” The algorithm controls itself, making it much more reliable than previous machine learning methods. Not only that, “We were surprised to discover that the algorithm is often able to identify anomalous events, namely real data inconsistent with our theoretical models. This information can be used to quickly ‘flag’ data for additional investigation,” said Stephen Green from the University of Nottingham and former scientist at the Max Planck Institute for Gravitational Physics.

“We can guarantee the accuracy of our machine learning method – which almost never happens in the field of deep learning. It therefore becomes compelling for the scientific community to use the algorithm to analyze gravitational-wave data,” says Alessandra Buonanno, director of the Department Astrophysical and Cosmological Relativity. Scientists from around the world are studying gravitational waves in large collaborations, such as the Ligo Scientific Collaboration, in which more than 1,500 researchers are organized. 

Bernhard Schölkopf, director at the Max Planck Institute for Intelligent Systems, adds, “Today, Dingo analyzes gravitational-wave data – but such a self-controlling and self-correcting method is also interesting for other scientific applications, where it is crucial to be able to corroborate the correctness of ‘black-box’ neural network methods.”