4 个月前

使用图卷积网络建模关系数据

使用图卷积网络建模关系数据

摘要

知识图谱支持了多种应用,包括问答系统和信息检索。尽管在创建和维护这些图谱上投入了大量努力,但即使是最大的知识图谱(如YAGO、DBPedia或Wikidata)仍然存在不完整性。我们介绍了关系图卷积网络(Relational Graph Convolutional Networks, R-GCNs),并将其应用于两个标准的知识库补全任务:链接预测(恢复缺失的事实,即主语-谓语-宾语三元组)和实体分类(恢复缺失的实体属性)。R-GCNs与最近一类在图数据上运行的神经网络有关,专门针对现实世界知识库中高度多关系的数据进行开发。我们展示了R-GCNs作为独立模型在实体分类任务中的有效性。此外,我们还证明了通过引入编码器模型来在关系图中积累多个推理步骤的证据,可以显著提升链接预测模型(如DistMult)的性能,在FB15k-237数据集上相对于仅使用解码器的基线模型,性能提升了29.8%。

基准测试

基准方法指标
heterogeneous-node-classification-on-acmRGCN
Macro-F1: 91.55
Micro-F1: 91.41
heterogeneous-node-classification-on-dblp-2RGCN
Macro-F1: 91.52
Micro-F1: 92.07
heterogeneous-node-classification-on-freebaseRGCN
Macro-F1: 46.78
Micro-F1: 58.33
heterogeneous-node-classification-on-imdbRGCN
Macro-F1: 58.85
Micro-F1: 62.05
heterogeneous-node-classification-on-oagRGCN
MRR: 31.51
NDCG: 48.93
heterogeneous-node-classification-on-oag-l1RGCN
MRR: 84.92
NDCG: 85.91
node-classification-on-aifbR-GCN
Accuracy: 95.83
node-classification-on-amR-GCN
Accuracy: 89.29
node-classification-on-bgsR-GCN
Accuracy: 83.10
node-classification-on-mutagR-GCN
Accuracy: 73.23

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用图卷积网络建模关系数据 | 论文 | HyperAI超神经