HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

metapath2vec: Scalable Representation Learning for Heterogeneous Networks

{Ananthram Swami Yuxiao Dong Nitesh Vijay Chawla}

Abstract

We study the problem of representation learning in heterogeneous networks. Its unique challenges come from the existence of multiple types of nodes and links, which limit the feasibility of the conventional network embedding techniques. We develop two scalable representation learning models, namely metapath2vec and metapath2vec++. The metapath2vec model formalizes meta-path-based random walks to construct the heterogeneous neighborhood of a node and then leverages a heterogeneous skip-gram model to perform node embeddings. The metapath2vec++ model further enables the simultaneous modeling of structural and semantic correlations in heterogeneous networks. Extensive experiments show that metapath2vec and metapath2vec++ are able to not only outperform state-of-the-art embedding models in various heterogeneous network mining tasks, such as node classification, clustering, and similarity search, but also discern the structural and semantic correlations between diverse network objects.

Benchmarks

BenchmarkMethodologyMetrics
link-prediction-on-movielens-25mmetapath2vec
Hits@10: 0.7956
nDCG@10: 0.5051
link-prediction-on-yelpMetapath2Vec
HR@10: 0.6307
nDCG@10: 0.402

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
metapath2vec: Scalable Representation Learning for Heterogeneous Networks | Papers | HyperAI