.png){kind=icon}
.png){kind=icon}
Return
Entropy-aware self-training for graph convolutional networks
.png){kind=icon}
PRE
AI
DOI：10.1016/j.neucom.2021.08.092.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Recently, graph convolutional networks (GCNs) have achieved significant success in many graph-based learning tasks, especially for node classification, due to its excellent ability in representation learning. Nevertheless, it remains challenging for GCN models to obtain satisfying predictions on graphs where only few nodes are with known labels. In this paper, we propose a novel entropy-aware self-training algo-rithm to boost semi-supervised node classification on graphs with little supervised information. Firstly, an entropy-aggregation layer is developed to strengthen the reasoning ability of GCN models. To the best of our knowledge, this is the first work to combine the entropy-based random walk theory with GCN design. Furthermore, we propose an ingenious checking part to add new nodes as supervision after each training round to enhance node prediction. In particular, the checking part is designed based on aggre-gated features, which is demonstrated more effective than previous methods and boosts node classifica-tion significantly. The proposed algorithm is validated on six public benchmarks in comparison with several state-of-the-art baseline algorithms, and the results illustrate its excellent performance. (c) 2021 Elsevier B.V. All rights reserved.
Keywords:
Graph convolutional network (GCN)
Self training
Semi-supervised learning
Node classification