S$^3$GNN: Efficient Global Mixing and Local Message Passing for Long-Range Graph Learning
Abstract
Lay Summary
Many real-world problems can be described as networks: molecules are networks of atoms, knowledge bases are networks of facts, and physical simulations can be represented as networks of interacting regions. Graph neural networks are widely used to learn from such data, but they often struggle when useful information must travel across many distant parts of the network. As messages are repeatedly passed from one node to another, important long-range information can become compressed or lost. This paper proposes S3GNN, a new graph neural network designed to make long-range communication easier and more efficient. The key idea is to combine local message passing between neighboring nodes with a lightweight global mixing step, allowing distant parts of the same connected network to exchange information more directly. The method is also designed to remain stable as more layers are used, so information does not quickly vanish or become unstable during propagation. Beyond standard graph learning benchmarks, this type of model has potential value for industry-facing applications where long-range interactions matter, such as molecular property prediction, knowledge graph question answering, large-scale network analysis, and mesh-based physical simulation. Experiments show that S3GNN performs strongly across diverse tasks, including long-range graph benchmarks, large-scale node classification, knowledge graph reasoning, brain signal interpolation, and fluid dynamics prediction. It achieves better accuracy while often using fewer parameters and lower computational cost than competing methods.