arXiv:2511.19976cs.LGcs.SI2025-11被引 1

通过自监督图聚类提升节点分类性能,让无标签数据也能贡献有效信号。

Rethinking Semi-Supervised Node Classification with Self-Supervised Graph Clustering

  • 将图神经网络与谱聚类目标统一优化,设计软正交消息传递机制
  • 在7个真实数据集上显著超越主流GNN模型,尤其在标签稀缺时优势明显
  • 适合处理标签稀疏的图数据,尤其对社区结构明显的网络效果突出

图神经网络(GNN)为半监督节点分类提供了强大工具。现有研究通过改进消息传递机制或引入数据增强来缓解标注不足问题。然而,真实图中节点常形成紧密社区,蕴含丰富补全标签的信息,但此前方法未充分挖掘。本文提出NCGC框架,将自监督图聚类与半监督分类统一建模。首先,理论统一GNN与谱聚类优化目标,提出软正交GNN(SOGN),采用改进的消息传递生成兼顾分类与聚类的节点表示。在此基础上,构建自监督聚类模块,包含两个非平凡聚类目标和Sinkhorn-Knopp归一化,将预测聚类分配转化为平衡的软伪标签。通过联合监督分类损失与自监督聚类损失的多任务目标,促进两者协同,增强模型能力。大量实验表明,NCGC在七个真实图数据集上,使用多种经典GNN主干网络,均显著优于主流GNN及近期基线方法。

原文摘要 · Abstract (English)

The emergence of graph neural networks (GNNs) has offered a powerful tool for semi-supervised node classification tasks. Subsequent studies have achieved further improvements through refining the message passing schemes in GNN models or exploiting various data augmentation techniques to mitigate limited supervision. In real graphs, nodes often tend to form tightly-knit communities/clusters, which embody abundant signals for compensating label scarcity in semi-supervised node classification but are not explored in prior methods. Inspired by this, this paper presents NCGC that integrates self-supervised graph clustering and semi-supervised classification into a unified framework. Firstly, we theoretically unify the optimization objectives of GNNs and spectral graph clustering, and based on that, develop soft orthogonal GNNs (SOGNs) that leverage a refined message passing paradigm to generate node representations for both classification and clustering. On top of that, NCGC includes a self-supervised graph clustering module that enables the training of SOGNs for learning representations of unlabeled nodes in a self-supervised manner. Particularly, this component comprises two non-trivial clustering objectives and a Sinkhorn-Knopp normalization that transforms predicted cluster assignments into balanced soft pseudo-labels. Through combining the foregoing clustering module with the classification model using a multi-task objective containing the supervised classification loss on labeled data and self-supervised clustering loss on unlabeled data, NCGC promotes synergy between them and achieves enhanced model capacity. Our extensive experiments showcase that the proposed NCGC framework consistently and considerably outperforms popular GNN models and recent baselines for semi-supervised node classification on seven real graphs, when working with various classic GNN backbones.

图神经网络半监督学习自监督聚类

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