3 个月前

基于差异性引用的半监督图节点分类

基于差异性引用的半监督图节点分类

摘要

在深度学习应用于图数据之后,图神经网络(Graph Neural Networks, GNNs)近年来已成为图上节点分类任务的主流方法。大多数GNN采用深度学习的端到端范式:将节点特征作为模型输入,并利用已标注节点的标签进行监督学习,以实现节点的分类。然而,尽管这些方法能够充分挖掘节点特征及其相互关联,却将标签视为独立信息,忽略了标签本身的结构特性。为充分利用标签的结构信息,本文提出了一种名为3ference的新方法,其核心思想是“基于差异的参考推理”。具体而言,3ference通过结合目标节点自身的特征,以及其相关节点的特征与标签,来推断该节点的类别标签。得益于对相关节点标签信息的引入,3ference能够捕捉节点间标签的转移模式,这一能力在后续的分析与可视化中得到了验证。在合成图及七个真实世界图数据集上的实验表明,相较于传统GNN方法,3ference在参数更少、已标注节点更少、且面对多样化标签分布的情况下,仍能实现更准确的节点分类,充分体现了标签关联性知识的有效性。

基准测试

基准方法指标
node-classification-on-amazon-computers-13ference
Accuracy: 90.74%
node-classification-on-amazon-photo-13ference
Accuracy: 95.05%
node-classification-on-citeseer3ference
Accuracy: 76.33
node-classification-on-coauthor-cs3ference
Accuracy: 95.99%
node-classification-on-coauthor-physics3ference
Accuracy: 97.22%
node-classification-on-cora3ference
Accuracy: 87.78%
node-classification-on-pubmed3ference
Accuracy: 88.90

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于差异性引用的半监督图节点分类 | 论文 | HyperAI超神经