4 个月前

GCNH:一种在异质图上进行表示学习的简单方法

GCNH:一种在异质图上进行表示学习的简单方法

摘要

图神经网络(GNNs)在同质图上表现出色,即边倾向于连接相同类型的节点的图。然而,在异质图上实现一致的GNN性能仍然是一个开放的研究问题。近期的研究提出了一些扩展标准GNN架构的方法,以提高其在异质图上的表现,这些方法通过牺牲模型的简洁性来换取预测准确性。然而,这些模型未能捕捉到一些基本的图属性,如邻域标签分布,而这些属性对于学习过程至关重要。本文中,我们提出了适用于异质和同质场景的简单且有效的GNN架构——GCN用于异质性(GCNH)。GCNH为每个节点及其邻居学习并结合独立表示,并在每一层使用一个学习到的重要性系数来平衡中心节点和邻域的贡献。我们对八个真实世界的图以及一系列具有不同异质程度的合成图进行了广泛的实验,以展示GCNH的设计选择如何显著改善了传统GCN的表现。此外,在八个基准测试中的四个上,GCNH的表现优于复杂度更高的最先进模型,而在其余数据集上则产生了相当的结果。最后,我们讨论并分析了GCNH较低的复杂度,这导致其可训练参数较少且训练时间更快,并展示了GCNH如何缓解过度平滑问题。

代码仓库

smartdata-polito/gcnh
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
node-classification-on-actorGCNH
Accuracy: 36.89 ± 1.50
node-classification-on-chameleonGCNH
Accuracy: 71.56±1.86
node-classification-on-cornellGCNH
Accuracy: 86.49±6.98
node-classification-on-squirrelGCNH-
node-classification-on-texasGCNH
Accuracy: 87.84±3.87
node-classification-on-wisconsinGCNH-

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
GCNH:一种在异质图上进行表示学习的简单方法 | 论文 | HyperAI超神经