3 个月前

通过分离学习澄清混淆节点

通过分离学习澄清混淆节点

摘要

图神经网络(Graph Neural Networks, GNNs)在面向图结构的任务中取得了显著进展。然而,现实世界中的图数据通常包含一定比例的异质性节点(heterophilous nodes),这打破了传统GNN所依赖的同质性假设(homophily assumption),从而制约了其性能表现。现有大多数研究仍采用在异质性与同质性节点间共享权重的通用模型架构。尽管部分方法引入了高阶消息传递机制或多通道网络结构,但其效果仍不理想。少数研究尝试对不同节点群体进行独立训练,但受限于分离指标设计不当以及计算效率低下,难以取得理想成果。本文首次提出一种新的节点分离度量指标——邻域混淆度(Neighborhood Confusion, NC),以实现更可靠的节点分组。我们观察到,具有不同NC值的节点群体在组内分类准确率及可视化嵌入表示上呈现出明显差异,这为基于NC的图卷积网络设计提供了依据。据此,我们提出了邻域混淆引导的图卷积网络(Neighborhood Confusion-guided Graph Convolutional Network, NCGCN),该模型根据节点的NC值进行分组,并在组内实施权重共享与消息传递机制。在同质性与异质性基准数据集上的大量实验表明,所提出的框架能够有效实现节点分离,并在性能上显著优于当前最先进的方法。相关源代码将公开于:https://github.com/GISec-Team/NCGNN。

代码仓库

GISec-Team/NCGNN
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
node-classification-on-actorNCGCN
Accuracy: 43.16 ± 1.32
node-classification-on-actorNCSAGE
Accuracy: 43.89 ± 1.33
node-classification-on-amz-computersNCGCN
Accuracy: 90.81 ± 0.46
node-classification-on-amz-computersNCSAGE
Accuracy: 90.43 ± 0.72
node-classification-on-amz-photoNCSAGE
Accuracy: 95.93 ± 0.36
node-classification-on-amz-photoNCGCN
Accuracy: 95.45 ± 0.45
node-classification-on-chameleon-60-20-20-1NCGCN
Accuracy: 46.84 ± 4.36
node-classification-on-chameleon-60-20-20-1NCSAGE
Accuracy: 48.07 ± 3.47
node-classification-on-coauthor-csNCSAGE
Accuracy: 96.48 ± 0.25
node-classification-on-coauthor-csNCGCN
Accuracy: 96.64 ± 0.29
node-classification-on-coauthor-physicsNCSAGE
Accuracy: 98.69 ± 0.26
node-classification-on-coauthor-physicsNCGCN
Accuracy: 98.63 ± 0.24
node-classification-on-cora-fullNCSAGE
Accuracy: 72.58 ± 0.65%
node-classification-on-cora-full-supervisedNCGCN
Accuracy: 73.42 ± 0.58%
node-classification-on-penn94NCGCN
Accuracy: 84.74 ± 0.28
node-classification-on-penn94NCSAGE
Accuracy: 81.77 ± 0.71
node-classification-on-pubmedNCSAGE
Accuracy: 91.55 ± 0.38
node-classification-on-pubmedNCGCN
Accuracy: 91.64 ± 0.53
node-classification-on-squirrel-60-20-20-1NCSAGE
Accuracy: 54.42 ± 1.41
node-classification-on-squirrel-60-20-20-1NCGCN
Accuracy: 54.15 ± 1.47

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过分离学习澄清混淆节点 | 论文 | HyperAI超神经