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

被引:10
作者
Lv, Jia [1 ]
Song, Kaikai [1 ]
Ye, Qiang [1 ]
Tian, Guangjian [2 ]
机构
[1] Huawei Technol, Noahs Ark Lab, Xian, Peoples R China
[2] Huawei Technol, Noahs Ark Lab, Hong Kong, Peoples R China
关键词
Graph neural network; Node classification; Data augmentation; Auxiliary learning; CONVOLUTIONAL NETWORKS;
D O I
10.1016/j.patcog.2023.109301
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页数:10
相关论文
共 37 条
  • [1] Bruna J, 2014, P 2 INT C LEARN REPR, DOI DOI 10.48550/ARXIV.1312.6203
  • [2] Defferrard M, 2016, ADV NEUR IN, V29
  • [3] Du YS, 2020, Arxiv, DOI arXiv:1812.02224
  • [4] Feng W., 2020, Advances in neural information processing systems, V33
  • [5] Gao X., 2020, PROC IEEE INT C MULT, P1
  • [6] Hamilton WL, 2017, ADV NEUR IN, V30
  • [7] GraphAIR: Graph representation learning with neighborhood aggregation and interaction
    Hu, Fenyu
    Zhu, Yanqiao
    Wu, Shu
    Huang, Weiran
    Wang, Liang
    Tan, Tieniu
    [J]. PATTERN RECOGNITION, 2021, 112
  • [8] Hu LM, 2019, 2019 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING AND THE 9TH INTERNATIONAL JOINT CONFERENCE ON NATURAL LANGUAGE PROCESSING (EMNLP-IJCNLP 2019), P4821
  • [9] Hui BY, 2020, AAAI CONF ARTIF INTE, V34, P4215
  • [10] Kipf T., 2017, INT C LEARNING REPRE