Graph Neural Network with curriculum learning for imbalanced node classification

• Published in: Neurocomputing (Amsterdam) Vol. 574; p. 127229
• Main Authors: Li, Xiaohe, Fan, Zide, Huang, Feilong, Hu, Xuming, Deng, Yawen, Wang, Lei, Zhao, Xinyu
• Format: Journal Article
• Language: English
• Published: Elsevier B.V 14.03.2024
• Subjects: Curriculum learning; Graph neural network; Imbalanced classification; Curriculum learning; Graph neural network; Imbalanced classification
• ISSN: 0925-2312; 1872-8286
• DOI: 10.1016/j.neucom.2023.127229

Graph Neural Network (GNN) stands as an emerging methodology for graph-based learning tasks, particularly for node classification. This study elucidates the susceptibility of GNN to discrepancies arising from imbalanced node labels. Conventional solutions for imbalanced classification, such as resampling, falter in node classification task, primarily due to their negligence of graph structure. Worse still, they often exacerbate the model’s inclination towards overfitting or underfitting, especially in the absence of adequate priori knowledge. To circumvent these limitations, we introduce a novel Graph Neural Network framework with Curriculum Learning (GNN-CL). This framework integrates two pivotal components. Initially, leveraging the principles of smoothness and homophily, we endeavor to procure dependable interpolation nodes and edges via adaptive graph oversampling. For another, we combine the Graph Classification Loss with the Metric Learning Loss, thereby refining the spatial proximity of nodes linked to the minority class in the feature space. Drawing inspiration from curriculum learning, the parameters of these components are dynamically modulated during the training phase to accentuate generalization and discrimination capabilities. Comprehensive evaluations on several widely used graph datasets affirm the superiority of our proposed model, which consistently outperforms the existing state-of-the-art methods.