通过结构平衡提升图神经网络在不平衡节点分类中的表现
GraphSB: Boosting Imbalanced Node Classification on Graphs through Structural Balance
- 提出结构平衡策略,先优化图结构再进行节点合成
- 在多个数据集上使SOTA方法平均提升3.67%准确率
- 可作为通用模块无缝集成,适合处理类别不均衡的图数据
不平衡节点分类是图学习中的关键挑战。现有方法主要分为数据级和算法级:前者通过生成少数类节点缓解数量失衡,后者优化学习过程以突出少数类。然而,两类方法均未关注图结构本身固有的不平衡性,这正是导致多数类主导、少数类被同化的根本原因。我们的理论分析验证了这一关键洞察。为此,我们提出GraphSB框架,将结构平衡作为核心策略,在节点合成前优化图结构。结构平衡包含两阶段:结构增强通过自适应构建相似性边,强化少数类节点的连接性;关系扩散捕捉高阶依赖并放大少数类信号。该方法在节点合成前平衡结构分布,显著提升GNN学习效果。大量实验表明,GraphSB显著优于现有最优方法。更重要的是,结构平衡可作为即插即用模块,使SOTA方法平均提升3.67%准确率。
原文摘要 · 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 category 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 adaptively builds similarity-based edges to strengthen connectivity of minority-class nodes, and Relation Diffusion that captures higher-order dependencies while amplifying signals from minority classes. 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 3.67\%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。