Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Rethinking Semi-Supervised Imbalanced Node Classification from Bias-Variance Decomposition

About

This paper introduces a new approach to address the issue of class imbalance in graph neural networks (GNNs) for learning on graph-structured data. Our approach integrates imbalanced node classification and Bias-Variance Decomposition, establishing a theoretical framework that closely relates data imbalance to model variance. We also leverage graph augmentation technique to estimate the variance, and design a regularization term to alleviate the impact of imbalance. Exhaustive tests are conducted on multiple benchmarks, including naturally imbalanced datasets and public-split class-imbalanced datasets, demonstrating that our approach outperforms state-of-the-art methods in various imbalanced scenarios. This work provides a novel theoretical perspective for addressing the problem of imbalanced node classification in GNNs.

Liang Yan, Gengchen Wei, Chen Yang, Shengzhong Zhang, Zengfeng Huang• 2023

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora (semi-supervised)--
103
Node ClassificationComputers Random rho=25.50 (test)
Balanced Accuracy85
33
Node ClassificationCS-Random (test)
Balanced Accuracy90.11
33
Node ClassificationCiteseer semi-supervised (test)
Accuracy66.04
26
Node ClassificationPubMed semi-supervised (test)--
11
Node ClassificationComputers Semi-supervised (test)
bAcc86.37
4
Showing 6 of 6 rows

Other info

Code

Follow for update