4 个月前

简单高效的异质图神经网络

简单高效的异质图神经网络

摘要

异质图神经网络(HGNNs)具有强大的能力,可以将异质图中的丰富结构和语义信息嵌入节点表示中。现有的HGNNs继承了许多来自同质图上的图神经网络(GNNs)的机制,特别是注意力机制和多层结构。这些机制带来了过高的复杂性,但很少有研究探讨它们在异质图上是否真正有效。本文对这些机制进行了深入而详细的研究,并提出了简单高效的异质图神经网络(SeHGNN)。为了轻松捕获结构信息,SeHGNN预先使用轻量级均值聚合器计算邻居聚合,通过去除过度使用的邻居注意力机制和避免每个训练周期重复进行邻居聚合来降低复杂性。为了更好地利用语义信息,SeHGNN采用了单层结构并使用长元路径扩展感受野,同时引入了一个基于Transformer的语义融合模块来融合不同元路径的特征。因此,SeHGNN表现出简单的网络结构、高预测精度和快速的训练速度的特点。在五个真实世界异质图上的广泛实验表明,SeHGNN在准确性和训练速度方面优于现有最先进方法。

代码仓库

ict-gimlab/sehgnn
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
heterogeneous-node-classification-on-acmSeHGNN
Claim Classification Macro-F1: 93.95
Micro-F1: 93.87
heterogeneous-node-classification-on-dblp-2SeHGNN
Macro-F1: 94.86
Micro-F1: 95.24
heterogeneous-node-classification-on-freebaseSeHGNN
Macro-F1: 50.71
Micro-F1: 63.41
heterogeneous-node-classification-on-imdbSeHGNN
Macro-F1: 66.63
Micro-F1: 68.21
heterogeneous-node-classification-on-oagSeHGNN
MRR: 29.11
NDCG: 46.75
heterogeneous-node-classification-on-oag-l1SeHGNN
MRR: 84.95
NDCG: 86.01

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
简单高效的异质图神经网络 | 论文 | HyperAI超神经