Nia-GNNs: neighbor-imbalanced aware graph neural networks for imbalanced node classification

被引:0
作者
Sun, Yanfeng [1 ]
Wang, Yujia [1 ]
Wang, Shaofan [1 ]
机构
[1] Beijing Univ Technol, Fac Informat Technol, Beijing 100124, Peoples R China
关键词
Graph neural networks; Class imbalance learning; Node classification; Oversampling;
D O I
10.1007/s10489-024-05590-z
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
It has been proven that Graph Neural Networks focus more on the majority class instances and ignore minority class instances when the class distribution is imbalanced. To address the class imbalance problems on graphs, most of the existing approaches rely on the availability of minority nodes in the training set, which may be scarce in extremely imbalanced situations and lead to overfitting. To tackle this issue, this paper proposes a novel oversampling-based Neighbor imbalanced-aware Graph Neural Networks, abbreviated as Nia-GNNs. Specifically, we propose a novel interpolation method that selects interpolated minority nodes from the entire dataset according to their predicted labels and similarity. Meanwhile, a class-wise interpolation ratio is applied to prevent the generation of out-of-domain nodes. Additionally, the generated minority nodes are inserted into the neighbor of minority nodes according to their neighbor distribution to balance the graph both neighborly and globally. Numerous experiments on different imbalanced datasets demonstrate the superiority of our method in classifying imbalanced nodes.
引用
收藏
页码:7941 / 7957
页数:17
相关论文
共 39 条
[1]   SMOTE: Synthetic minority over-sampling technique [J].
Chawla, Nitesh V. ;
Bowyer, Kevin W. ;
Hall, Lawrence O. ;
Kegelmeyer, W. Philip .
2002, American Association for Artificial Intelligence (16)
[2]   Financial time series forecasting with multi-modality graph neural network [J].
Cheng, Dawei ;
Yang, Fangzhou ;
Xiang, Sheng ;
Liu, Jin .
PATTERN RECOGNITION, 2022, 121
[3]   Automatic detection of relevant information, predictions and forecasts in financial news through topic modelling with Latent Dirichlet Allocation [J].
Garcia-Mendez, Silvia ;
de Arriba-Perez, Francisco ;
Barros-Vila, Ana ;
Gonzalez-Castano, Francisco J. J. ;
Costa-Montenegro, Enrique .
APPLIED INTELLIGENCE, 2023, 53 (16) :19610-19628
[4]  
Hamilton WL, 2017, ADV NEUR IN, V30
[5]   Disentangling Label Distribution for Long-tailed Visual Recognition [J].
Hong, Youngkyu ;
Han, Seungju ;
Choi, Kwanghee ;
Seo, Seokjun ;
Kim, Beomsu ;
Chang, Buru .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :6622-6632
[6]  
Japkowicz N., 2002, Intelligent Data Analysis, V6, P429
[7]  
Kang B., 2020, Decoupling Representation and Classifier for Long-Tailed Recognition, P1
[8]   SOPHIE velocimetry of Kepler transit candidates XVII. The physical properties of giant exoplanets within 400 days of period [J].
Santerne, A. ;
Moutou, C. ;
Tsantaki, M. ;
Bouchy, F. ;
Hebrard, G. ;
Adibekyan, V. ;
Almenara, J. -M. ;
Amard, L. ;
Barros, S. C. C. ;
Boisse, I. ;
Bonomo, A. S. ;
Bruno, G. ;
Courcol, B. ;
Deleuil, M. ;
Demangeon, O. ;
Diaz, R. F. ;
Guillot, T. ;
Havel, M. ;
Montagnier, G. ;
Rajpurohit, A. S. ;
Rey, J. ;
Santos, N. C. .
ASTRONOMY & ASTROPHYSICS, 2016, 587
[9]   DLIR: a deep learning-based initialization recommendation algorithm for trust-aware recommendation [J].
Liu, Taiheng ;
He, Zhaoshui .
APPLIED INTELLIGENCE, 2022, 52 (10) :11103-11114
[10]   Pick and Choose: A GNN-based Imbalanced Learning Approach for Fraud Detection [J].
Liu, Yang ;
Ao, Xiang ;
Qin, Zidi ;
Chi, Jianfeng ;
Feng, Jinghua ;
Yang, Hao ;
He, Qing .
PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE 2021 (WWW 2021), 2021, :3168-3177