
摘要
知识图谱支持了多种应用,包括问答系统和信息检索。尽管在创建和维护这些图谱上投入了大量努力,但即使是最大的知识图谱(如YAGO、DBPedia或Wikidata)仍然存在不完整性。我们介绍了关系图卷积网络(Relational Graph Convolutional Networks, R-GCNs),并将其应用于两个标准的知识库补全任务:链接预测(恢复缺失的事实,即主语-谓语-宾语三元组)和实体分类(恢复缺失的实体属性)。R-GCNs与最近一类在图数据上运行的神经网络有关,专门针对现实世界知识库中高度多关系的数据进行开发。我们展示了R-GCNs作为独立模型在实体分类任务中的有效性。此外,我们还证明了通过引入编码器模型来在关系图中积累多个推理步骤的证据,可以显著提升链接预测模型(如DistMult)的性能,在FB15k-237数据集上相对于仅使用解码器的基线模型,性能提升了29.8%。
代码仓库
INK-USC/MHGRN
pytorch
GitHub 中提及
shijx12/kqapro_baselines
pytorch
GitHub 中提及
susurrant/flow-imputation
tf
GitHub 中提及
MichSchli/RelationPrediction
tf
GitHub 中提及
toooooodo/rgcn-linkprediction
pytorch
GitHub 中提及
giuseppefutia/semi
pytorch
GitHub 中提及
guillaumejaume/tuto-dl-on-graphs
pytorch
GitHub 中提及
predict-idlab/RR-GCN
pytorch
GitHub 中提及
dglai/wsdm2022-challenge
pytorch
GitHub 中提及
susurrant/spatial-interaction-modeling
tf
GitHub 中提及
thiviyanT/torch-rgcn
pytorch
GitHub 中提及
tkipf/relational-gcn
官方
tf
GitHub 中提及
anilakash/indkgc
pytorch
GitHub 中提及
masakicktashiro/rgcn_pytorch_implementation
pytorch
GitHub 中提及
QustKcz/relational-GCN
tf
GitHub 中提及
berlincho/RGCN-pytorch
pytorch
GitHub 中提及
parkererickson/crunchBaseGraph
pytorch
GitHub 中提及
Microsoft/gated-graph-neural-network-samples
tf
GitHub 中提及
kracr/document-level-relation-extraction
pytorch
GitHub 中提及
INK-USC/RE-Net
pytorch
GitHub 中提及
tkipf/gae
tf
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| heterogeneous-node-classification-on-acm | RGCN | Macro-F1: 91.55 Micro-F1: 91.41 |
| heterogeneous-node-classification-on-dblp-2 | RGCN | Macro-F1: 91.52 Micro-F1: 92.07 |
| heterogeneous-node-classification-on-freebase | RGCN | Macro-F1: 46.78 Micro-F1: 58.33 |
| heterogeneous-node-classification-on-imdb | RGCN | Macro-F1: 58.85 Micro-F1: 62.05 |
| heterogeneous-node-classification-on-oag | RGCN | MRR: 31.51 NDCG: 48.93 |
| heterogeneous-node-classification-on-oag-l1 | RGCN | MRR: 84.92 NDCG: 85.91 |
| node-classification-on-aifb | R-GCN | Accuracy: 95.83 |
| node-classification-on-am | R-GCN | Accuracy: 89.29 |
| node-classification-on-bgs | R-GCN | Accuracy: 83.10 |
| node-classification-on-mutag | R-GCN | Accuracy: 73.23 |