arXiv:2601.19352cs.LG2026-01

通过结构平衡提升图神经网络在不平衡节点分类中的表现

GraphSB: Boosting Imbalanced Node Classification on Graphs through Structural Balance

  • 提出结构平衡策略,先优化图结构再进行节点合成
  • 在多个数据集上相比最优方法提升平均4.57%准确率
  • 可作为模块嵌入现有方法,适合处理类别严重失衡的图数据

不平衡节点分类是图学习中的关键挑战。现有方法多依赖图神经网络(GNN)学习节点表示,分为数据级与算法级两类:前者通过合成少数类节点缓解数量不平衡,后者优化学习过程以突出少数类。然而,二者均未解决图结构本身固有的不平衡性,这是导致多数类主导、少数类被同化的根本原因。本文理论分析验证了该观点。为此,提出GraphSB框架,将结构平衡作为核心策略,在节点合成前优化图结构。结构平衡包含两阶段:结构增强通过双视角分析挖掘决策边界附近的难样本,并自适应增强少数类连通性;关系扩散在传播增强后的少数类上下文的同时捕捉高阶结构依赖。实验表明,GraphSB显著优于现有方法;更重要的是,结构平衡可无缝集成至主流方法中,平均提升准确率4.57%。

原文摘要 · Abstract (English)

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%.

图神经网络不平衡分类结构优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。