4 个月前

可扩展的异质图神经网络

可扩展的异质图神经网络

摘要

图神经网络(GNNs)是一类用于学习图结构数据的流行参数模型。近期的研究表明,GNNs 主要利用图进行特征平滑,并且通过仅对图平滑后的节点特征进行操作,而不是使用难以扩展到大规模图的端到端学习特征层次结构,在基准任务中展示了具有竞争力的结果。在本研究中,我们探讨这些结果是否可以推广到异质图,这类图编码了不同实体之间的多种关系类型。为此,我们提出了基于关系子图的邻居平均方法(NARS),该方法在“元图”(metagraph)的随机采样子图上训练一个分类器,使用邻居平均特征。我们描述了一些优化措施,以确保这些节点特征集在训练和推理过程中能够高效地计算。NARS 在多个基准数据集上达到了新的最先进水平,其性能优于更为昂贵的基于 GNN 的方法。

代码仓库

facebookresearch/NARS
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
heterogeneous-node-classification-on-acmNARS
Macro-F1: 93.36
Micro-F1: 93.31
heterogeneous-node-classification-on-dblp-2NARS
Macro-F1: 94.18
Micro-F1: 94.61
heterogeneous-node-classification-on-freebaseNARS
Macro-F1: 49.98
Micro-F1: 63.26
heterogeneous-node-classification-on-imdbNARS
Macro-F1: 63.51
Micro-F1: 66.18
heterogeneous-node-classification-on-oagNARS
MRR: 34.38
NDCG: 52.28
heterogeneous-node-classification-on-oag-l1NARS
MRR: 85.15
NDCG: 86.06

用 AI 构建 AI

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

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

Hyper Newsletters

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