Graph Link Prediction Via Decay Coefficient Based Proportional Aggregation and Hybrid Concatenation

Jiaxin Zhuang,Yahui Chai,Xiaobin Rui,Zhixiao Wang
DOI: https://doi.org/10.1109/ijcnn60899.2024.10649981
2024-01-01
Abstract:As one of the popular topics in social network analysis, link prediction aims to predict the likely but unobserved links between two nodes. It has a wide field of applications such as knowledge graph completion and recommender systems. Currently, graph neural network (GNN) is the most common method with excellent performance, which contains two main steps, i.e., node aggregation and edge concatenation. However, there are some drawbacks with the existing methods. Firstly, traditional node aggregation methods usually iteratively aggregate all or a fixed number of neighbors, which is inflexible and inefficient. Secondly, most of the existing edge concatenation methods only apply a single concatenation approach to obtain edge embeddings, which cannot fully guarantee the embedding quality. To tackle the two problems, this paper proposes a graph embedding approach for link prediction via decay coefficient based Proportional Aggregation and Hybrid Concatenation (PAHC). On one hand, PAHC directly aggregates only proportional neighbors from different orders by correlating the decaying phenomenon of information propagation in social networks with the decaying phenomenon of temperature in Newton’s cooling theorem. On the other hand, PAHC proposes a hybrid concatenation approach to obtain the final edge embedding by mixing weighted summation and weighted direct concatenation. Experiments on five datasets show that through decay proportional aggregation and hybrid concatenation, the proposed PAHC can better predict links in social networks, outperforming state-of-the-art methods.
What problem does this paper attempt to address?