A Multi-information Embedding Link Prediction Approach with Collective Attention Flow Network

Link prediction and community detection reveal the basic mechanism and evolution law of the network from different perspectives, and the relationship between community members can provide valuable information for link prediction. Most link prediction methods are based on local structural features and lack the application of community topology information. To deal with this challenge, we propose a link prediction method NCELP with collective attention flow network. Firstly, we apply the Louvain algorithm to give each node a community label as an explicit feature. Secondly, using local structural features and community topology information, learned node and community embedding, which serve as implicit features. Finally, combined with the graph structure features, the link prediction problem is transformed into a binary classification problem and realized by the existence probability of the edge. We validated NCELP using behavior data from China Internet Network Information Center with more than 30,000 online users and three public datasets. Experimental results verify that NCELP not only outperforms the state-of-the-art methods on real-world datasets but also improves its AUC value by at least 9.83% and its AP value by at least 3.39%.

[1]  Xuelong Li,et al.  Community Detection in Graph: An Embedding Method , 2022, IEEE Transactions on Network Science and Engineering.

[2]  Francisco Valente Link Prediction of Artificial Intelligence Concepts using Low Computational Power , 2021, 2021 IEEE International Conference on Big Data (Big Data).

[3]  Junbin Gao,et al.  How Neural Processes Improve Graph Link Prediction , 2021, ICASSP 2022 - 2022 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[4]  Yan Chen,et al.  Dig users' intentions via attention flow network for personalized recommendation , 2021, Inf. Sci..

[5]  Mykola Pechenizkiy,et al.  NodeSim: node similarity based network embedding for diverse link prediction , 2021, EPJ Data Science.

[6]  George Karypis,et al.  Graph InfoClust: Leveraging cluster-level node information for unsupervised graph representation learning , 2020, ArXiv.

[7]  Sibo Wang,et al.  Inductive Link Prediction for Nodes Having Only Attribute Information , 2020, IJCAI.

[8]  Dimitris N. Metaxas,et al.  Rethinking Kernel Methods for Node Representation Learning on Graphs , 2019, NeurIPS.

[9]  Jie Cao,et al.  Link prediction in temporal networks: Integrating survival analysis and game theory , 2019, Inf. Sci..

[10]  Lexing Xie,et al.  Estimating Attention Flow in Online Video Networks , 2019, Proc. ACM Hum. Comput. Interact..

[11]  Ning Feng,et al.  Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting , 2019, AAAI.

[12]  Yixin Chen,et al.  An End-to-End Deep Learning Architecture for Graph Classification , 2018, AAAI.

[13]  Yixin Chen,et al.  Link Prediction Based on Graph Neural Networks , 2018, NeurIPS.

[14]  Niloy Ganguly,et al.  Discriminative Link Prediction using Local, Community, and Global Signals , 2016, IEEE Transactions on Knowledge and Data Engineering.

[15]  John E. Hopcroft,et al.  Using community information to improve the precision of link prediction methods , 2012, WWW.

[16]  Jean-Loup Guillaume,et al.  Fast unfolding of communities in large networks , 2008, 0803.0476.