Link prediction algorithm based on network representation learning and random walk

Si LIU,Hai LIU,Qimai CHEN,Chaobo HE
DOI: https://doi.org/10.11772/j.issn.1001-9081.2017.08.2234
2017-01-01
Journal of Computer Applications
Abstract:The transition process of existing link prediction indexes based on random walk exists strong randomness in the unweighted network and does not consider the effect of the similarity of the network structure between different neighbor nodes on transition probability.In order to solve the problems,a new link prediction algorithm based on network representation learning and random walk was proposed.Firstly,the latent structure features of network node were learnt by DeepWalk which is a network representation learning algorithm based on deep learning,and the network nodes were encoded into lowdimensional vector space.Secondly,the similarity between neighbor nodes in vector space was incorporated into the transition process of Random Walk with Restart (RWR) and Local Random Walk (LRW) respectively and the transition probability of each random walk was redefined.Finally,a large number of experiments on five real datasets were carried out.The experimental results show that the AUC (Area Under the receiver operating characteristic Curve) values of the proposed algorithms are improved up to 3.34% compared with eight representative link prediction benchmarks based on network structure.
What problem does this paper attempt to address?