BiasedWalk: Biased Sampling for Representation Learning on Graphs

Duong Nguyen (1) , Fragkiskos Malliaros (2, 3)

1 UC San Diego - University of California [San Diego]
2 CVN - Centre de vision numérique
3 GALEN-POST - Organ Modeling through Extraction, Representation and Understanding of Medical Image Content

Network embedding algorithms are able to learn latent feature representations of nodes, transforming networks into lower dimensional vector representations. Typical key applications, which have effectively been addressed using network embeddings, include link prediction, multilabel classification and community detection. In this paper, we propose Biased-Walk, a scalable, unsupervised feature learning algorithm that is based on biased random walks to sample context information about each node in the network. Our random-walk based sampling can behave as Breath-First-Search (BFS) and Depth-First-Search (DFS) samplings with the goal to capture homophily and role equivalence between the nodes in the network. We have performed a detailed experimental evaluation comparing the performance of the proposed algorithm against various baseline methods, on several datasets and learning tasks. The experiment results show that the proposed method outperforms the baseline ones in most of the tasks and datasets.

Mots clés: Graph mining; random walks; network representation learning; unsupervised feature learning; node sampling

Domaines: Informatique [cs]; Réseaux sociaux et d'information [cs.SI]; Apprentissage [cs.LG]

Duong Nguyen, Fragkiskos Malliaros. BiasedWalk: Biased Sampling for Representation Learning on Graphs. International Workshop on High Performance Big Graph Data Management, Analysis, and Mining (BigGraphs), Dec 2018, Seattle, WA, United States. ⟨10.1109/bigdata.2018.8621872⟩. ⟨hal-01958902⟩