Link prediction analysis based on Node2Vec embedding technique

被引:0
作者
Devi, Salam Jayachitra [1 ]
Singh, Buddha [1 ]
机构
[1] Jawaharlal Nehru Univ, Sch Comp & Syst Sci, New Delhi, India
关键词
embedding; link prediction; random walk; Node2Vec; natural language processing; PERFORMANCE; ALGORITHM;
D O I
10.1504/IJCAT.2023.134091
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
The paper focuses on analysing link prediction using the Node2Vec embedding technique, which is based on the Random Walk algorithm. In addition to this, several machine learning models have been employed to assess the effectiveness of the embedding technique. Node2Vec employs various embedding operators, including Hadamard, Concatenation, Average, Weighted L1, and Weighted L2. The comparative analysis of this embedding technique is done on real world network data sets using various machine learning models with state of the art link prediction algorithms. Performance assessment of Node2Vec's embedding technique is based on the AUC metric. According to the simulation results, it has been determined that the concatenation operator with the bagging classifier yields mean AUC value of 0.939, outperforming the other operators, which produce AUC values below 0.91. Furthermore, the study has also revealed that the embedding technique provides superior results when applied to networks with a low ratio of nodes to edges.
引用
收藏
页码:79 / 89
页数:12
相关论文
共 46 条
[1]  
Ackland R, 2005, BLOGTALK DOWN C SYDN
[2]  
Aggarwal CC, 2011, SOCIAL NETWORK DATA ANALYTICS, P1, DOI 10.1007/978-1-4419-8462-3
[3]   AN INTRODUCTION TO KERNEL AND NEAREST-NEIGHBOR NONPARAMETRIC REGRESSION [J].
ALTMAN, NS .
AMERICAN STATISTICIAN, 1992, 46 (03) :175-185
[4]  
Batagelj Vladimir, 2006, Pajek datasets
[5]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[6]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[7]   Link Prediction with Mutual Attention for Text-Attributed Networks [J].
Brochier, Robin ;
Guille, Adrien ;
Velcin, Julien .
COMPANION OF THE WORLD WIDE WEB CONFERENCE (WWW 2019 ), 2019, :283-284
[8]   A Comprehensive Survey of Graph Embedding: Problems, Techniques, and Applications [J].
Cai, HongYun ;
Zheng, Vincent W. ;
Chang, Kevin Chen-Chuan .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2018, 30 (09) :1616-1637
[9]   Network embedding for link prediction: The pitfall and improvement [J].
Cao, Ren-Meng ;
Liu, Si-Yuan ;
Xu, Xiao-Ke .
CHAOS, 2019, 29 (10)
[10]  
Cao S., 2015, P 24 ACM INT C INF K, P891