GraphSAGE++: Weighted Multi-scale GNN for Graph Representation Learning

E. Jiawei,Yinglong Zhang,Shangying Yang,Hong Wang,Xuewen Xia,Xing Xu
DOI: https://doi.org/10.1007/s11063-024-11496-1
IF: 2.565
2024-02-11
Neural Processing Letters
Abstract:Graph neural networks (GNNs) have emerged as a powerful tool in graph representation learning. However, they are increasingly challenged by over-smoothing as network depth grows, compromising their ability to capture and represent complex graph structures. Additionally, some popular GNN variants only consider local neighbor information during node updating, ignoring the global structural information and leading to inadequate learning and differentiation of graph structures. To address these challenges, we introduce a novel graph neural network framework, GraphSAGE++. Our model extracts the representation of the target node at each layer and then concatenates all layer weighted representations to obtain the final result. In addition, the strategies combining double aggregations with weighted concatenation are proposed, which significantly enhance the model's discernment and preservation of structural information. Empirical results on various datasets demonstrate that GraphSAGE++ excels in vertex classification, link prediction, and visualization tasks, surpassing existing methods in effectiveness.
computer science, artificial intelligence
What problem does this paper attempt to address?