Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DropEdge: Towards Deep Graph Convolutional Networks on Node Classification

About

\emph{Over-fitting} and \emph{over-smoothing} are two main obstacles of developing deep Graph Convolutional Networks (GCNs) for node classification. In particular, over-fitting weakens the generalization ability on small dataset, while over-smoothing impedes model training by isolating output representations from the input features with the increase in network depth. This paper proposes DropEdge, a novel and flexible technique to alleviate both issues. At its core, DropEdge randomly removes a certain number of edges from the input graph at each training epoch, acting like a data augmenter and also a message passing reducer. Furthermore, we theoretically demonstrate that DropEdge either reduces the convergence speed of over-smoothing or relieves the information loss caused by it. More importantly, our DropEdge is a general skill that can be equipped with many other backbone models (e.g. GCN, ResGCN, GraphSAGE, and JKNet) for enhanced performance. Extensive experiments on several benchmarks verify that DropEdge consistently improves the performance on a variety of both shallow and deep GCNs. The effect of DropEdge on preventing over-smoothing is empirically visualized and validated as well. Codes are released on~\url{https://github.com/DropEdge/DropEdge}.

Yu Rong, Wenbing Huang, Tingyang Xu, Junzhou Huang• 2019

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy75.29
1252
Node ClassificationCora
Accuracy87.46
1215
Graph ClassificationMUTAG
Accuracy86.11
1103
Node ClassificationCiteseer
Accuracy78.92
1037
Node ClassificationCora (test)
Mean Accuracy83.5
951
Node ClassificationCiteseer (test)
Accuracy0.727
945
Node ClassificationChameleon
Accuracy62.08
867
Node ClassificationPubmed
Accuracy89.45
865
Node ClassificationWisconsin
Accuracy50.59
864
Node ClassificationCornell
Accuracy61.62
851
Showing 10 of 174 rows
...

Other info

Code

Follow for update