Edge-nodes representation neural machine for link prediction

Abstract

Link prediction is a task predicting whether there is a link between two nodes in a network. Traditional link prediction methods that assume handcrafted features (such as common neighbors) as the link's formation mechanism are not universal. Other popular methods tend to learn the link's representation, but they cannot represent the link fully. In this paper, we propose Edge-Nodes Representation Neural Machine (ENRNM), a novel method which can learn abundant topological features from the network as the link's representation to promote the formation of the link. The ENRNM learns the link's formation mechanism by combining the representation of edge and the representations of nodes on the two sides of the edge as link's full representation. To predict the link's existence, we train a fully connected neural network which can learn meaningful and abundant patterns. We prove that the features of edge and two nodes have the same importance in link's formation. Comprehensive experiments are conducted on eight networks, experiment results demonstrate that the method ENRNM not only exceeds plenty of state-of-the-art link prediction methods but also performs very well on diverse networks with different structures and characteristics.