Comprehensive Analysis of Over-smoothing in Graph Neural Networks from Markov Chains Perspective

Weichen Zhao, Chenguang Wang, Congying Han, Tiande Guo

Main figure for Comprehensive Analysis of Over-smoothing in Graph Neural Networks from Markov Chains Perspective

Abstract

The over-smoothing problem is an obstacle of developing deep graph neural network (GNN). Although many approaches to improve the over-smoothing problem have been proposed, there is still a lack of comprehensive understanding and conclusion of this problem. In this work, we analyze the over-smoothing problem from the Markov chain perspective. We focus on message passing of GNN and first establish a connection between GNNs and Markov chains on the graph. GNNs are divided into two classes of operator-consistent and operator-inconsistent based on whether the corresponding Markov chains are time-homogeneous. Next we attribute the over-smoothing problem to the convergence of an arbitrary initial distribution to a stationary distribution. Based on this, we prove that although the previously proposed methods can alleviate over-smoothing, but these methods cannot avoid the over-smoothing problem. In addition, we give the conclusion of the over-smoothing problem in two types of GNNs in the Markovian sense. On the one hand, operator-consistent GNN cannot avoid over-smoothing at an exponential rate. On the other hand, operator-inconsistent GNN is not always over-smoothing. Further, we investigate the existence of the limiting distribution of the time-inhomogeneous Markov chain, from which we derive a sufficient condition for operator-inconsistent GNN to avoid over-smoothing. Finally, we design experiments to verify our findings. Results show that our proposed sufficient condition can effectively improve over-smoothing problem in operator-inconsistent GNN and enhance the performance of the model.

Proposed method: regularization term

Benchmark Results

Tested on 5 datasets 3 baseline values reported too low

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

Classic

#ModelAccuracy ↑Paper
1
GraphSAGE+CNA
0.9418 ±0.0033
2
GCN+CNA
0.9366 ±0.0048
3
TransformerConv
0.9350 ±0.0058
21
GCN
0.9004
44
GAT
actual #44 vs paper #721 677 places worse
0.8897 ±0.0014
best reported
actual #44 vs paper #721 677 places worse
paper 0.7773 ±0.0054
paper 0.7773 ±0.0054 Δ at pub 0.1117 Δ today 0.1124
700
regularization term
0.7884 ±0.0157
This paper
724
MLP
0.7768
#ModelAccuracy ↑Paper
1
CNA
0.9575 ±0.0058
2
IS-GIB
0.9390 ±0.0187
3
EERM
0.9112 ±0.0145
35
GAT
actual #35 vs paper #733 698 places worse
0.8080 ±0.0026
best reported
actual #35 vs paper #733 698 places worse
paper 0.6643 ±0.0063
paper 0.6643 ±0.0063 Δ at pub 0.1370 Δ today 0.1437
44
GCN
0.8033 ±0.0088
350
MLP
0.7289 ±0.0010
724
regularization term
0.6692 ±0.0072
This paper
#ModelAccuracy ↑Paper
1
LGTL
0.9518 ±0.0011
2
MixHop (LO)
0.9513 ±0.0038
3
GraphSAGE (LO)
0.9511 ±0.0036
18
GCN
0.9296 ±0.0015
20
GAT
actual #20 vs paper #754 734 places worse
0.9233 ±0.0018
best reported
actual #20 vs paper #754 734 places worse
paper 0.7653 ±0.0072
210
MLP
0.8690 ±0.0074
745
regularization term
0.7684 ±0.0063
This paper

OGB

#ModelROC-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
73
regularization term
0.7872 ±0.0083
This paper
79
GEN
0.7858 ±0.0117
This paper
109
GIN
0.7780 ±0.0182
#ModelAccuracy ↑Paper
1
A-NLSF
0.8149 ±0.0067
2
differential encoding
0.8096 ±0.0029
3
MPNN + VN + NoPE
0.8055 ±0.0038
9
regularization term
0.7800 ±0.0037
This paper
12
GEN
0.7712 ±0.0071
This paper
16
GatedGCN
0.7531 ±0.0083
33
GIN
0.7037 ±0.0107
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)