Semi-supervised node classification via fine-grained graph auxiliary augmentation learning

doi:10.1016/j.patcog.2023.109301

2023-05-01

PRE

Jia Lv *

Qiang Ye

Guangjian Tian

DOI：10.1016/j.patcog.2023.109301

原文链接

原文求助

摘要

En 中文

Node classification has become an important research topic in recent years. Since there are always a few training samples, researchers improve the performance by properly leveraging the predictions of unlabeled nodes during training. However, suffering from the model degradation resulted from the accumulative error of pseudo-labels, there is limited improvement. In this paper we present fine-grained Graph Auxiliary aUgmentation (GAU). It trains the primary task together with an automatically created auxiliary task which is a fine-grained node classification task. And an auxiliary augmentation strategy is designed to enlarge the labeled set for the auxiliary task by utilizing the pseudo-labels of the primary task. Comprehensive experiments show that GAU alleviates the sensitivity of the model to the pseudo-label quality, so more unlabeled nodes can participate in the training. From the perspective of co-training, the fine-grained auxiliary task which is trained by much more unlabeled nodes helps to learn better node representations from a different view, thereby boosting the final performance. Extensive experiments verify the superior performance of the GAU on different GNN architectures when compared with other state-of-the-art approaches. (c) 2023 Elsevier Ltd. All rights reserved.

Keyword:

Graph neural network

Node classification

Data augmentation

Auxiliary learning