Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Meta-Path Guided Embedding for Similarity Search in Large-Scale Heterogeneous Information Networks

About

Most real-world data can be modeled as heterogeneous information networks (HINs) consisting of vertices of multiple types and their relationships. Search for similar vertices of the same type in large HINs, such as bibliographic networks and business-review networks, is a fundamental problem with broad applications. Although similarity search in HINs has been studied previously, most existing approaches neither explore rich semantic information embedded in the network structures nor take user's preference as a guidance. In this paper, we re-examine similarity search in HINs and propose a novel embedding-based framework. It models vertices as low-dimensional vectors to explore network structure-embedded similarity. To accommodate user preferences at defining similarity semantics, our proposed framework, ESim, accepts user-defined meta-paths as guidance to learn vertex vectors in a user-preferred embedding space. Moreover, an efficient and parallel sampling-based optimization algorithm has been developed to learn embeddings in large-scale HINs. Extensive experiments on real-world large-scale HINs demonstrate a significant improvement on the effectiveness of ESim over several state-of-the-art algorithms as well as its scalability.

Jingbo Shang, Meng Qu, Jialu Liu, Lance M. Kaplan, Jiawei Han, Jian Peng• 2016

Related benchmarks

TaskDatasetResultRank
Node ClassificationDBLP (test)
Macro-F192.53
70
Node ClassificationIMDB (test)
Macro F1 Score32.1
70
Node ClusteringACM
ARI34.32
57
Node ClusteringDBLP
NMI0.6632
39
Node ClusteringIMDB
NMI0.55
24
Node ClassificationDBLP (80% train)
Macro F192.27
20
Node ClassificationDBLP (20% train)
Macro F190.68
20
Node ClassificationIMDb (20% train)
Macro F1 Score48.37
20
Node ClusteringDBLP (test)
NMI68.33
17
Link PredictionLast.FM (test)
AUC82
10
Showing 10 of 12 rows

Other info

Follow for update