GraphSB: Boosting Imbalanced Node Classification on Graphs through Structural Balance

TL;DR

GraphSB introduces a novel structural balance framework that improves imbalanced node classification on graphs by optimizing graph structure before node synthesis, leading to significant performance gains.

Contribution

The paper proposes GraphSB, a new framework that addresses imbalanced graph structures through a two-stage optimization, enhancing minority class connectivity and structural dependencies.

Findings

GraphSB outperforms state-of-the-art methods in experiments.

Structural Balance improves accuracy by an average of 4.57%.

The framework is easily integrated into existing methods.

Abstract

Imbalanced node classification is a critical challenge in graph learning, where most existing methods typically utilize Graph Neural Networks (GNNs) to learn node representations. These methods can be broadly categorized into the data-level and the algorithm-level. The former aims to synthesize minority-class nodes to mitigate quantity imbalance, while the latter tries to optimize the learning process to highlight minority classes. However, neither of them addresses the inherently imbalanced graph structure, which is a fundamental factor that incurs majority-class dominance and minority-class assimilation in GNNs. Our theoretical analysis further supports this critical insight. Therefore, we propose GraphSB (Graph Structural Balance), a novel framework that incorporates Structural Balance as a key strategy to address the underlying imbalanced graph structure before node synthesis. Structural Balance performs a two-stage structure optimization: Structure Enhancement that mines hard samples near decision boundaries through dual-view analysis and enhances connectivity for minority classes through adaptive augmentation, and Relation Diffusion that propagates the enhanced minority context while simultaneously capturing higher-order structural dependencies. Thus, GraphSB balances structural distribution before node synthesis, enabling more effective learning in GNNs. Extensive experiments demonstrate that GraphSB significantly outperforms the state-of-the-art methods. More importantly, the proposed Structural Balance can be seamlessly integrated into state-of-the-art methods as a simple plug-and-play module, increasing their accuracy by an average of 4.57%.