HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Dimensionality Reduction Meets Message Passing for Graph Node Embeddings

Krzysztof Sadowski Michał Szarmach Eddie Mattia

Dimensionality Reduction Meets Message Passing for Graph Node Embeddings

Abstract

Graph Neural Networks (GNNs) have become a popular approach for various applications, ranging from social network analysis to modeling chemical properties of molecules. While GNNs often show remarkable performance on public datasets, they can struggle to learn long-range dependencies in the data due to over-smoothing and over-squashing tendencies. To alleviate this challenge, we propose PCAPass, a method which combines Principal Component Analysis (PCA) and message passing for generating node embeddings in an unsupervised manner and leverages gradient boosted decision trees for classification tasks. We show empirically that this approach provides competitive performance compared to popular GNNs on node classification benchmarks, while gathering information from longer distance neighborhoods. Our research demonstrates that applying dimensionality reduction with message passing and skip connections is a promising mechanism for aggregating long-range dependencies in graph structured data.

Code Repositories

ksadowski13/PCAPass
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
node-classification-on-redditPCAPass + XGBoost
Accuracy: 96.26 ± 0.02%

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