Network embedding based on high-degree penalty and adaptive negative sampling

Gang-Feng Ma,Xu-Hua Yang,Wei Ye,Xin-Li Xu,Lei Ye
DOI: https://doi.org/10.1007/s10618-023-00973-1
IF: 5.406
2023-09-03
Data Mining and Knowledge Discovery
Abstract:Network embedding can effectively dig out potentially useful information and discover the relationships and rules which exist in the data, that has attracted increasing attention in many real-world applications. The goal of network embedding is to map high-dimensional and sparse networks into low-dimensional and dense vector representations. In this paper, we propose a network embedding method based on high-degree penalty and adaptive negative sampling (NEPS). First, we analyze the problem of imbalanced node training in random walk and propose an indicator base on high-degree penalty, which can control the random walk and avoid over-sampling high-degree neighbor node. Then, we propose a two-stage adaptive negative sampling strategy, which can dynamically obtain negative samples suitable for the current training according to the training stage to improve training effect. By comparing with seven well-known network embedding algorithms on eight real-world data sets, experiments show that the NEPS has good performance in node classification, network reconstruction and link prediction. The code is available at: https://github.com/Andrewsama/NEPS-master.
computer science, information systems, artificial intelligence
What problem does this paper attempt to address?