Message gain aggregation architecture: a scalable graph neural network for combining large-scale neighborhoods

Xianrui Li,Yan Wang
DOI: https://doi.org/10.1117/12.2680724
2023-05-23
Abstract:Graph Neural Networks (GNNs) extend traditional deep neural networks to graph-structured data and have greatly succeeded in graph representation learning scenarios. However, the current mainstream GNNs based on message passing mechanisms are challenging to scale to large graph data due to their severe scalability and over smoothing issues. In this paper, we propose a novel scheme called the Message Gain Aggregation Architecture (MGAA) to alleviate the over smoothing problem while maintaining a scalable architecture and good expressiveness. Our MGAA network considers the nodes themselves separately from their neighbors and treats the neighbors as gains to be imposed on the nodes. A theoretical complexity analysis demonstrates that our proposed MGAA network maintains linear temporal and spatial complexity. Furthermore, the experimental results show that our model achieves sufficiently competitive performance in a comparison with the current popular state-of-the-art GNNs.
Engineering,Computer Science
What problem does this paper attempt to address?