arXiv · 2502.00716
UPL: Uncertainty-aware Pseudo-labeling for Imbalance Transductive Node Classification
Abstract
Graph-structured datasets often suffer from class imbalance, which complicates node classification tasks. In this work, we address this issue by first providing an upper bound on population risk for imbalanced transductive node classification. We then propose a simple and novel algorithm, Uncertainty-aware Pseudo-labeling (UPL). Our approach leverages pseudo-labels assigned to unlabeled nodes to mitigate the adverse effects of imbalance on classification accuracy. Furthermore, the UPL algorithm enhances the accuracy of pseudo-labeling by reducing training noise of pseudo-labels through a novel uncertainty-aware approach. We comprehensively evaluate the UPL algorithm across various benchmark datasets, demonstrating its superior performance compared to existing state-of-the-art methods.
Explore related subjects
Keep this discovery
Mohammad T. Teimuri, Zahra Dehghanian, Gholamali Aminian, Hamid R. Rabiee. 2025-02-02. UPL: Uncertainty-aware Pseudo-labeling for Imbalance Transductive Node Classification. https://arxiv.org/abs/2502.00716
Cite the original work for its findings. Save a collection to share your selection of sources.