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

Yanfeng Sun,Yujia Wang,Shaofan Wang
DOI: https://doi.org/10.1007/s10489-024-05590-z
IF: 5.3
2024-06-16
Applied Intelligence
Abstract: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 N eighbor i mbalanced- a ware G raph N eural N etwork s , 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.
computer science, artificial intelligence
What problem does this paper attempt to address?