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

Do We Need Anisotropic Graph Neural Networks?

About

Common wisdom in the graph neural network (GNN) community dictates that anisotropic models -- in which messages sent between nodes are a function of both the source and target node -- are required to achieve state-of-the-art performance. Benchmarks to date have demonstrated that these models perform better than comparable isotropic models -- where messages are a function of the source node only. In this work we provide empirical evidence challenging this narrative: we propose an isotropic GNN, which we call Efficient Graph Convolution (EGC), that consistently outperforms comparable anisotropic models, including the popular GAT or PNA architectures by using spatially-varying adaptive filters. In addition to raising important questions for the GNN community, our work has significant real-world implications for efficiency. EGC achieves higher model accuracy, with lower memory consumption and latency, along with characteristics suited to accelerator implementation, while being a drop-in replacement for existing architectures. As an isotropic model, it requires memory proportional to the number of vertices in the graph ($\mathcal{O}(V)$); in contrast, anisotropic models require memory proportional to the number of edges ($\mathcal{O}(E)$). We demonstrate that EGC outperforms existing approaches across 6 large and diverse benchmark datasets, and conclude by discussing questions that our work raise for the community going forward. Code and pretrained models for our experiments are provided at https://github.com/shyam196/egc.

Shyam A. Tailor, Felix L. Opolka, Pietro Li\`o, Nicholas D. Lane• 2021

Related benchmarks

TaskDatasetResultRank
Node ClassificationOGB-MAG (test)
Accuracy47.21
55
Graph Classificationogbg-code2 standard (test)
F1 Score15.95
20
Node Classificationogbn-arxiv transductive
Accuracy72.21
13
Graph ClassificationCIFAR-10 Superpixels (Unseen Graph)
Accuracy71.03
10
Graph RegressionZINC (Unseen Graph)
MAE0.281
10
Graph ClassificationMolHIV Unseen Graph
ROC AUC0.7818
10
Showing 6 of 6 rows

Other info

Code

Follow for update