English

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

Machine Learning 2025-02-26 v4

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.

Keywords

Cite

@article{arxiv.2310.18765,
  title  = {Rethinking Semi-Supervised Imbalanced Node Classification from Bias-Variance Decomposition},
  author = {Liang Yan and Gengchen Wei and Chen Yang and Shengzhong Zhang and Zengfeng Huang},
  journal= {arXiv preprint arXiv:2310.18765},
  year   = {2025}
}

Comments

Accepted by NeurIPS 2023

R2 v1 2026-06-28T13:04:44.105Z