Graph neural network speeds multiple scattering simulations with adaptive sampling

ScaGNN: a Graph Neural Network for Multiple Scattering Simulations

Machine Learning

Summary

Simulating how waves bounce around many objects is usually slow because of the complex math involved. The authors developed ScaGNN, which uses a special kind of neural network that treats the problem like a network of points and connections. This network learns to pick out the most important interactions to calculate, making the simulation faster without losing much accuracy. Their method works well on different test cases, even when the number or shape of the objects changes.

What this means in practice

  • For acoustic engineers: Perform faster simulations of sound wave scattering around multiple objects to improve noise control and design.
  • For electromagnetic simulation teams: Simulate electromagnetic wave scattering more efficiently in environments with many obstacles using learned graph models.

Authors

Rémi Marsal, Stéphanie Chaillat, Alexandre Chapoutot

Abstract

The boundary element method (BEM) provides an efficient numerical framework for solving multiple scattering problems in unbounded homogeneous domains. By restricting the discretization to the domain boundaries, it substantially reduces computational complexity. The procedure first consists in determining the solution trace on the boundaries of the domain by solving a boundary integral equation. Then, the volumetric solution can be recovered at low computational cost using a boundary integral representation. As the first step of the BEM represents the main computational bottleneck, we present ScaGNN, a learning-based approach designed to approximate the solution trace. It relies on a graph neural network architecture that incorporates a dynamic adaptive edge sampling mechanism for selecting the most relevant interactions to model. Guided by intermediate predictions of expected error and edge length, this mechanism selects, at various stages of the forward pass, the most relevant distant interactions to model. The proposed method is tailored to achieve linear complexity with the number of nodes in the input graph. To train and evaluate our network, we present a benchmark consisting of several datasets with different types of multiple scattering problems. Our experiments show that our approach surpasses existing state-of-the-art learning-based methods on the considered tasks and investigate the generalization capabilities to settings with an increased number of obstacles and out-of-distribution obstacle shapes. github.com/LARIAD/ScaGNN