ICML 2023

Graph Neural Networks with Learnable and Optimal Polynomial Bases

Yuhe Guo, Zhewei Wei

Main figure for Graph Neural Networks with Learnable and Optimal Polynomial Bases

Abstract

Polynomial filters, a kind of Graph Neural Networks, typically use a predetermined polynomial basis and learn the coefficients from the training data. It has been observed that the effectiveness of the model is highly dependent on the property of the polynomial basis. Consequently, two natural and fundamental questions arise: Can we learn a suitable polynomial basis from the training data? Can we determine the optimal polynomial basis for a given graph and node features? In this paper, we propose two spectral GNN models that provide positive answers to the questions posed above. First, inspired by Favard's Theorem, we propose the FavardGNN model, which learns a polynomial basis from the space of all possible orthonormal bases. Second, we examine the supposedly unsolvable definition of optimal polynomial basis from Wang&Zhang (2022) and propose a simple model, OptBasisGNN, which computes the optimal basis for a given graph structure and graph signal. Extensive experiments are conducted to demonstrate the effectiveness of our proposed models. Our code is available at https://github.com/yuziGuo/FarOptBasis.

Proposed method: FavardGNN

Benchmark Results

Tested on 5 datasets 4 baseline values reported too low

Tables show presented results in global context (best 3 papers + GCN + MLP).

OGB

Heterophilic Graphs

#ModelAccuracy ↑Paper
1
DiGGR
0.4535 ±0.0353
2
TRIGON
0.4402 ±0.0028
3
DoG
0.4320 ±0.0120
4
FavardGNN
0.4305 ±0.0053
This paper
11
OptBasisGNN
0.4239 ±0.0052
This paper
21
BernNet
actual #21 vs paper #22 1 place worse
0.4179 ±0.0101
best reported
actual #21 vs paper #22 1 place worse
paper 0.4171 ±0.0112

Difference within reported uncertainty; not counted as underperformance.

paper 0.4171 ±0.0112 Δ today 0.0008
22
ChebNetII
0.4175 ±0.0107
This paper
34
JacobiConv
0.4117 ±0.0064
This paper
41
APPNP
actual #41 vs paper #54 13 places worse
0.4046 ±0.0064
best reported
actual #41 vs paper #54 13 places worse
paper 0.3976 ±0.0049

Difference within reported uncertainty; not counted as underperformance.

paper 0.3976 ±0.0049 Δ today 0.0070
42
MLP
actual #42 vs paper #43 1 place worse
0.4026 ±0.0108
best reported
actual #42 vs paper #43 1 place worse
paper 0.4018 ±0.0055

Difference within reported uncertainty; not counted as underperformance.

paper 0.4018 ±0.0055 Δ at pub 0.0008
53
GPR-GNN
actual #53 vs paper #53
0.3992 ±0.0067
best reported
actual #53 vs paper #53
paper 0.3991 ±0.0062

Difference within reported uncertainty; not counted as underperformance.

paper 0.3991 ±0.0062 Δ today 0.0001
68
GraphSAGE
0.3899 ±0.0085
88
ChebNet
actual #88 vs paper #114 26 places worse
0.3802 ±0.0023
best reported
actual #88 vs paper #114 26 places worse
paper 0.3742 ±0.0058

Difference within reported uncertainty; not counted as underperformance.

paper 0.3742 ±0.0058 Δ at pub 0.0060
111
ARMA
0.3767 ±0.0054
This paper
135
GCN
actual #135 vs paper #284 149 places worse
0.3690
best reported
actual #135 vs paper #284 149 places worse
paper 0.3326 ±0.0115
paper 0.3326 ±0.0115 Δ at pub 0.0364

LINKX Benchmarks

#ModelAccuracy ↑Paper
1
DC-GNN
0.8669 ±0.0022
2
M2M-GNN
0.8594 ±0.0040
3
GloGNN
0.8574 ±0.0042
6
FavardGNN
0.8492 ±0.0041
This paper
7
ChebNetII
0.8486 ±0.0033
This paper
8
OptBasisGNN
0.8485 ±0.0039
This paper
11
LINKX
0.8471 ±0.0052
This paper
17
MixHop
actual #17 vs paper #20 3 places worse
0.8363 ±0.0054
best reported
actual #17 vs paper #20 3 places worse
paper 0.8347 ±0.0071

Difference within reported uncertainty; not counted as underperformance.

paper 0.8347 ±0.0071 Δ at pub 0.0016
20
GPR-GNN
0.8354 ±0.0032
This paper
23
BernNet
0.8326 ±0.0029
This paper
27
GCNII
0.8292 ±0.0059
This paper
29
GIN
0.8268 ±0.0032
31
GCN
0.8247 ±0.0027
This paper
40
LINK
0.8079 ±0.0049
This paper
51
MLP
actual #51 vs paper #57 6 places worse
0.7534 ±0.0064
best reported
actual #51 vs paper #57 6 places worse
paper 0.7361 ±0.0040
paper 0.7361 ±0.0040 Δ today 0.0173
#ModelAccuracy ↑Paper
1
SAGE
0.8597 ±0.0021
2
Polynormer
0.8595 ±0.0007
3
NID_SAGE
0.8563 ±0.0031
5
OptBasisGNN
0.8283 ±0.0004
This paper
6
ChebNetII
0.8233 ±0.0028
This paper
7
LINKX
0.8204 ±0.0007
This paper
9
BernNet
0.8167 ±0.0017
This paper
10
MixHop
0.8107 ±0.0016
This paper
13
GPR-GNN
0.8074 ±0.0022
This paper
14
LINK
0.8054 ±0.0003
This paper
16
GCNII
0.7894 ±0.0011
This paper
23
GCN
actual #23 vs paper #24 1 place worse
0.7601 ±0.0049
best reported
actual #23 vs paper #24 1 place worse
paper 0.7545 ±0.0017
paper 0.7545 ±0.0017 Δ today 0.0056
24
GraphSAGE
0.7576 ±0.0004
31
MLP
actual #31 vs paper #39 8 places worse
0.7048 ±0.0077
best reported
actual #31 vs paper #39 8 places worse
paper 0.6237 ±0.0002
paper 0.6237 ±0.0002 Δ at pub 0.0811
#ModelAccuracy ↑Paper
1
DGT
0.6609 ±0.0022
2
ICG_u-NN
0.6608 ±0.0074
3
LINKX
0.6606 ±0.0019
This paper
6
MixHop
actual #6 vs paper #6
0.6564 ±0.0027
actual #6 vs paper #6
paper 0.6564 ±0.0027
paper 0.6564 ±0.0027 Δ at pub 0.0044
9
OptBasisGNN
0.6517 ±0.0016
This paper
11
ChebNetII
0.6503 ±0.0027
This paper
13
LINK
0.6485 ±0.0021
This paper
14
GraphSAGE
0.6473 ±0.0011
16
GCN
actual #16 vs paper #28 12 places worse
0.6434 ±0.0012
best reported
actual #16 vs paper #28 12 places worse
paper 0.6218 ±0.0026
best at pub: Deformable Graph Transformer (2022)
paper 0.6218 ±0.0026 Δ at pub 0.0216
18
BernNet
0.6427 ±0.0031
This paper
19
FavardGNN
0.6426 ±0.0012
This paper
23
GCNII
0.6339 ±0.0061
This paper
27
GPR-GNN
0.6259 ±0.0038
This paper
33
MLP
actual #33 vs paper #37 4 places worse
0.6114 ±0.0006
best reported
actual #33 vs paper #37 4 places worse
paper 0.6092 ±0.0007
best at pub: Deformable Graph Transformer (2022)
paper 0.6092 ±0.0007 Δ at pub 0.0022
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported result worse than the best known when published (best reported in green) Best at pub — reported value improved over the best known when published Best today — OK at publication; today's leaderboard best is elsewhere