Rethinking Semi-Supervised Imbalanced Node Classification from Bias-Variance Decomposition
Divin Yan, Gengchen Wei, Chen Yang, Shengzhong Zhang, Zengfeng Huang
Abstract
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f7f5c95e-0071-4cb4-a422-9b672990303bCited by top-tier papers5
- DIVE: Subgraph Disagreement for Graph Out-of-Distribution GeneralizationXin Sun, Liang Wang, Qiang Liu, Shu Wu et al.KDD 2024 · 6 citations
- Geometric Imbalance in Semi-Supervised Node ClassificationLiang Yan, Shengzhong Zhang, Bisheng Li, Menglin Yang et al.NeurIPS 2025 · 2 citations
- NodeImport: Imbalanced Node Classification with Node Importance AssessmentNan Chen, Zemin Liu, Bryan Hooi, Bingsheng He et al.KDD 2025 · 1 citation
- Rel-MOSS: Towards Imbalanced Relational Deep Learning on Relational DatabasesJun Yin, Peng Huo, Bangguo Zhu, Hao Yan et al.ICML 2026 · 1 citation
- ContraDiff: Planning Towards High Return States via Contrastive LearningYixiang Shan, Zhengbang Zhu, Ting Long, Qifan Liang et al.ICLR 2025
Builds on26
- FixMatch: Simplifying Semi-Supervised Learning with Consistency and ConfidenceKihyuk Sohn, David Berthelot, Nicholas Carlini, Zizhao Zhang et al.NeurIPS 2020 · 5,129 citations
- Graph Contrastive Learning with AugmentationsYuning You, Tianlong Chen, Yongduo Sui, Ting Chen et al.NeurIPS 2020 · 3,042 citations
- Unsupervised Data Augmentation for Consistency TrainingQizhe Xie, Zihang Dai, Eduard H. Hovy, Thang Luong et al.NeurIPS 2020 · 2,774 citations
- Decoupling Representation and Classifier for Long-Tailed RecognitionBingyi Kang, Saining Xie, Marcus Rohrbach, Zhicheng Yan et al.ICLR 2020 · 1,496 citations
- Graph Contrastive Learning with Adaptive AugmentationYanqiao Zhu, Yichen Xu, Feng Yu, Qiang Liu et al.WWW 2021 · 1,415 citations
Related papers
- When Imbalance Meets Imbalance: Structure-driven Learning for Imbalanced Graph ClassificationWei Xu, Pengkun Wang, Zhe Zhao, Binwu Wang et al.WWW 2024 · 19 citations
- GraphENS: Neighbor-Aware Ego Network Synthesis for Class-Imbalanced Node ClassificationJoonhyung Park, Jaeyun Song, Eunho YangICLR 2022 · 145 citations
- BIM: Improving Graph Neural Networks with Balanced Influence MaximizationWentao Zhang, Xinyi Gao, Ling Yang, Meng Cao et al.ICDE 2024 · 4 citations
- Class-Imbalanced Graph Learning without Class RebalancingZhining Liu, Ruizhong Qiu, Zhichen Zeng, Hyunsik Yoo et al.ICML 2024 · 35 citations
- IceBerg: Debiased Self-Training for Class-Imbalanced Node ClassificationZhixun Li, Dingshuo Chen, Tong Zhao, Daixin Wang et al.WWW 2025 · 7 citations
