HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Bridging the Gap between Community and Node Representations: Graph Embedding via Community Detection

Artem Lutov Dingqi Yang Philippe Cudré-Mauroux

Bridging the Gap between Community and Node Representations: Graph Embedding via Community Detection

Abstract

Graph embedding has become a key component of many data mining and analysis systems. Current graph embedding approaches either sample a large number of node pairs from a graph to learn node embeddings via stochastic optimization or factorize a high-order proximity/adjacency matrix of the graph via computationally expensive matrix factorization techniques. These approaches typically require significant resources for the learning process and rely on multiple parameters, which limits their applicability in practice. Moreover, most of the existing graph embedding techniques operate effectively in one specific metric space only (e.g., the one produced with cosine similarity), do not preserve higher-order structural features of the input graph and cannot automatically determine a meaningful number of embedding dimensions. Typically, the produced embeddings are not easily interpretable, which complicates further analyses and limits their applicability. To address these issues, we propose DAOR, a highly efficient and parameter-free graph embedding technique producing metric space-robust, compact and interpretable embeddings without any manual tuning. Compared to a dozen state-of-the-art graph embedding algorithms, DAOR yields competitive results on both node classification (which benefits form high-order proximity) and link prediction (which relies on low-order proximity mostly). Unlike existing techniques, however, DAOR does not require any parameter tuning and improves the embeddings generation speed by several orders of magnitude. Our approach has hence the ambition to greatly simplify and speed up data analysis tasks involving graph representation learning.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
node-classification-on-blogcatalogDAOR
Macro F1: 17.25
Micro F1: 33.05
node-classification-on-dblpDAOR
Macro F1: 87.64
Micro F1: 87.86
node-classification-on-wikiDAOR
Macro F1: 15.97
Micro F1: 53.24

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
Bridging the Gap between Community and Node Representations: Graph Embedding via Community Detection | Papers | HyperAI