3 个月前

Rot-Pro:通过投影建模知识图谱嵌入中的传递性

Rot-Pro:通过投影建模知识图谱嵌入中的传递性

摘要

知识图谱嵌入模型旨在学习知识图谱中实体与关系的表示,以预测实体之间的缺失链接(关系)。其性能在很大程度上取决于对不同类型关系模式(如对称性、非对称性、逆关系、复合关系及传递性)的建模与推理能力。尽管现有模型已能够捕捉多种关系模式,但作为一类极为常见的关系模式,传递性仍未得到充分建模与支持。本文首先从理论上证明,传递性关系可通过投影操作进行建模。基于此,我们提出了一种名为 Rot-Pro 的新模型,该模型将投影机制与关系旋转相结合。我们进一步证明,Rot-Pro 模型能够有效推理上述所有关系模式。实验结果表明,所提出的 Rot-Pro 模型能够有效学习传递性关系,并在包含传递性关系的数据集上,于链接预测任务中取得了当前最优的性能表现。

代码仓库

Tigter/ogblwikikg2-RotPro
pytorch
GitHub 中提及
tewiSong/Rot-Pro
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
link-prediction-on-fb15k-237Rot-Pro
Hits@1: 0.246
Hits@10: 0.540
Hits@3: 0.383
MRR: 0.344
link-prediction-on-wn18rrRot-Pro
Hits@1: 0.397
Hits@10: 0.577
Hits@3: 0.482
MRR: 0.457
link-prediction-on-yago3-10Rot-Pro
Hits@1: 0.443
Hits@10: 0.699
Hits@3: 0.596
MRR: 0.542
link-property-prediction-on-ogbl-wikikg2Rot-Pro
Ext. data: No
Number of params: 1000669602
Test MRR: 0.5602 ± 0.0016
Validation MRR: 0.5740 ± 0.0008
link-property-prediction-on-ogbl-wikikg2RotPro
Ext. data: No
Number of params: 1000669602
Test MRR: 0.4277 ± 0.0008
Validation MRR: 0.4174 ± 0.0058

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
Rot-Pro:通过投影建模知识图谱嵌入中的传递性 | 论文 | HyperAI超神经