Do We Need Anisotropic Graph Neural Networks?
Abstract
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 ($ (V)$); in contrast, anisotropic models require memory proportional to the number of edges ($ (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.
Benchmark Results
Tested on 1 dataset
Tables show presented results in global context (best 3 papers + GCN + MLP).
OGB
| # | Model | ROC-AUC ↑ | Paper |
|---|---|---|---|
| 1 | Multi-RF Fusion with Multi-GNN Blending | 0.8476
±0.0002 | |
| 2 | HyperFusion | 0.8475
±0.0003 | |
| 3 | PAS+FPs | 0.8420
±0.0015 | |
| 9 | GatedGCN | 0.8090
±0.0160 | |
| 62 | PNA | 0.7905
±0.0132 | This paper |
| 92 | EGC-M | 0.7818
±0.0153 | This paper |
| 109 | GIN | 0.7780
±0.0182 best reported | best at pub: Parameterized Hypercomplex Graph Neural Networks for Graph Classification (2021) best today: GRPE: Relative Positional Encoding for Graph Transformer (2022) Difference within reported uncertainty; not counted as underperformance. |
| 126 | EGC-S | 0.7744
±0.0108 | This paper |
| 140 | GAT | 0.7717
±0.0137 | This paper |
| 142 | GATv2 | 0.7715
±0.0155 | This paper |
| 151 | MPNN-max | 0.7707
±0.0137 | This paper |
| 160 | GraphSAGE | 0.7690
±0.0136 best reported | best today: SoGCN: Second-Order Graph Convolutional Networks (2021) Difference within reported uncertainty; not counted as underperformance. |
| 184 | GCN | 0.7614
±0.0129 | prev best: Graph convolutions that can finally model local structure (2020) Improvement within reported uncertainty; not highlighted. |
| 235 | MPNN-sum | 0.7519
±0.0357 | This paper |
