DropEdge: Towards Deep Graph Convolutional Networks on Node Classification

By Yu Rong, Wenbing Huang, Tingyang Xu, Junzhou Huang. International Conference on Learning Representations, 2019

1,663 citations recorded by the source corpus.

Over-fitting and 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 retards 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 will be made public upon the publication.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

View Paper: https://www.semanticscholar.org/paper/0a6a9e6d4e3efd7c69357769305b70097281655f

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Dacheng Tao
  2. Philip S. Yu
  3. Jianfeng Gao
  4. Jure Leskovec
  5. Xiangnan He
  6. Percy Liang
  7. Bernard Ghanem
  8. Yoshua Bengio

In Computer Vision and Pattern Recognition

  1. Dacheng Tao
  2. Luc Van Gool
  3. Ming-Hsuan Yang
  4. Xiaogang Wang
  5. Yi Yang
  6. Trevor Darrell
  7. Bernard Ghanem
  8. Jitendra Malik

Citations in the corpus