HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Self-attention Dual Embedding for Graphs with Heterophily

Yurui Lai Taiyan Zhang Rui Fan

Self-attention Dual Embedding for Graphs with Heterophily

Abstract

Graph Neural Networks (GNNs) have been highly successful for the node classification task. GNNs typically assume graphs are homophilic, i.e. neighboring nodes are likely to belong to the same class. However, a number of real-world graphs are heterophilic, and this leads to much lower classification accuracy using standard GNNs. In this work, we design a novel GNN which is effective for both heterophilic and homophilic graphs. Our work is based on three main observations. First, we show that node features and graph topology provide different amounts of informativeness in different graphs, and therefore they should be encoded independently and prioritized in an adaptive manner. Second, we show that allowing negative attention weights when propagating graph topology information improves accuracy. Finally, we show that asymmetric attention weights between nodes are helpful. We design a GNN which makes use of these observations through a novel self-attention mechanism. We evaluate our algorithm on real-world graphs containing thousands to millions of nodes and show that we achieve state-of-the-art results compared to existing GNNs. We also analyze the effectiveness of the main components of our design on different graphs.

Benchmarks

BenchmarkMethodologyMetrics
node-classification-on-actorSADE-GCN
Accuracy: 37.91 ± 0.97
node-classification-on-chameleonSADE-GCN
Accuracy: 75.57±1.57
node-classification-on-cornellSADE-GCN
Accuracy: 86.21±5.59
node-classification-on-squirrelSADE-GCN
Accuracy: 68.20±1.57
node-classification-on-texasSADE-GCN
Accuracy: 86.49±5.12
node-classification-on-wisconsinSADE-GCN
Accuracy: 88.63±4.54

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
Self-attention Dual Embedding for Graphs with Heterophily | Papers | HyperAI