Attributed Heterogeneous Network Embedding for Link Prediction

Network embedding aims to embed the network into a low-dimensional vector space wherein the structural characteristic of the network and the attribute information of nodes are preserved as much as possible. Many existing network embedding works focused on the homogeneous or heterogeneous plain networks. However, networks in the real world are usually not plain since the nodes in the networks have rich attributes, and these attributes play important roles for encoding nodes’ vector representations. Although some works took into account the attribute information, they could not handle the homogeneous and heterogeneous structure information of the network simultaneously. In order to solve this problem, a new network embedding method that considers both the network’s homogeneous and heterogeneous structure information and nodes attribute information simultaneously is proposed in this paper. The proposed method first obtains nodes attribute information, homogeneous and heterogeneous structure information as three views of the network and learns network embeddings of the three views through different technologies respectively. Then, an attention mechanism is utilized to fuse the embedding results learned from the three views to obtain the final vector representations of nodes. We verify the performance of the proposed model through link prediction tasks on four real-world datasets, and extensive experimental results show that the proposed model outperforms the advanced baselines.