Label-GCN: An Effective Method for Adding Label Propagation to Graph Convolutional Networks
arXiv:2104.02153
Abstract
We show that a modification of the first layer of a Graph Convolutional Network (GCN) can be used to effectively propagate label information across neighbor nodes, for binary and multi-class classification problems. This is done by selectively eliminating self-loops for the label features during the training phase of a GCN. The GCN architecture is otherwise unchanged, without any extra hyper-parameters, and can be used in both a transductive and inductive setting. We show through several experiments that, depending on how many labels are available during the inference phase, this strategy can lead to a substantial improvement in the model performance compared to a standard GCN approach, including with imbalanced datasets.
References in corpus (7)
- Semi-Supervised Classification with Graph Convolutional Networks
- Near linear time algorithm to detect community structures in large-scale networks
- Simplifying Graph Convolutional Networks
- FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling
- Unifying Graph Convolutional Neural Networks and Label Propagation
- Characterizing and Detecting Money Laundering Activities on the Bitcoin Network
- A Graph Convolutional Network Composition Framework for Semi-supervised Classification