Network Embedding with Enhanced Feature Representations for Link Prediction.
Weisheng Li,Wentao Chen,Zhihong Pan,Yuanfei Deng,Yong Tang,Chaobo He
DOI: https://doi.org/10.1109/cscwd57460.2023.10152845
2023-01-01
Abstract:Link prediction is a hot research topic in graph data analytics, which aims to identify the likelihood of a future connection between two nodes in a network. Among the approaches to link prediction, network embedding has gained tremendous popularity in tackling link prediction tasks. Basically, there are several network embedding methods, such as matrix factorization, random walk, and deep learning. However, most of the existing methods for network embedding solely rely on structure information or fail to effectively combine network structure and feature data. To this end, this paper proposes a novel method that learns network embedding with enhanced feature representations for link prediction (EFRLP). Specifically, structure graphs, bipartite graphs, and co-feature graphs are generated from the attributed network, weighted bipartite graphs are then built over bipartite graphs and correlation weights between features. We further adopt a biased random walk to obtain more representative node sequences. With the help of enriched feature representations, not only can the higher-order semantic relationships be effectively captured but also network sparsity can be alleviated. Then, node sequences are selected into the Skip-Gram model for learning low-dimensional embeddings. Finally, the probability of the existence of edges between nodes is predicted by deep fusion. The effectiveness of the EFRLP method is demonstrated by extensive comparison experiments with several baselines on two real-world datasets.