Graph representation learning with encoding edges

Qi Li, Zehong Cao, J. Zhong, Qing Li

Research output: Contribution to journalArticlepeer-review

20 Scopus citations

Abstract

Network embedding aims at learning the low dimensional representation of nodes. These representations can be widely used for network mining tasks, such as link prediction, anomaly detection, and classification. Recently, a great deal of meaningful research work has been carried out on this emerging network analysis paradigm. The real-world network contains different size clusters because of the edges with different relationship types. These clusters also reflect some features of nodes, which can contribute to the optimization of the feature representation of nodes. However, existing network embedding methods do not distinguish these relationship types. In this paper, we propose an unsupervised network representation learning model that can encode edge relationship information. Firstly, an objective function is defined, which can learn the edge vectors by implicit clustering. Then, a biased random walk is designed to generate a series of node sequences, which are put into Skip-Gram to learn the low dimensional node representations. Extensive experiments are conducted on several network datasets. Compared with the state-of-art baselines, the proposed method is able to achieve favorable and stable results in multi-label classification and link prediction tasks.

Original languageEnglish
Pages (from-to)29-39
Number of pages11
JournalNeurocomputing
Volume361
DOIs
StatePublished - 7 Oct 2019
Externally publishedYes

Keywords

  • Edge representation
  • Feature learning
  • Network embedding
  • Network mining

Fingerprint

Dive into the research topics of 'Graph representation learning with encoding edges'. Together they form a unique fingerprint.

Cite this