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

Self-attention Dual Embedding for Graphs with Heterophily

About

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.

Yurui Lai, Taiyan Zhang, Rui Fan• 2023

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy87.93
885
Node ClassificationCiteseer
Accuracy77.45
804
Node ClassificationPubmed
Accuracy90.07
742
Node ClassificationChameleon
Accuracy75.57
549
Node ClassificationSquirrel
Accuracy68.2
500
Node ClassificationCornell
Accuracy86.21
426
Node ClassificationWisconsin
Accuracy88.63
410
Node ClassificationTexas
Accuracy0.8649
410
Node ClassificationFilm
Accuracy37.91
127
Showing 9 of 9 rows

Other info

Follow for update