4 个月前

TuckER:用于知识图谱补全的张量分解方法

TuckER:用于知识图谱补全的张量分解方法

摘要

知识图谱是对现实世界事实的结构化表示。然而,它们通常只包含所有可能事实的一小部分。链接预测是一项基于现有事实推断缺失事实的任务。我们提出了一种名为TuckER的模型,该模型相对简单但功能强大,基于知识图谱三元组的二进制张量表示的Tucker分解。TuckER在标准链接预测数据集上的表现优于先前的最先进模型,成为更复杂模型的强大基线。我们证明了TuckER是一种完全表达性的模型,推导了其嵌入维度的充分界限,并展示了几个先前引入的线性模型可以被视为TuckER的特例。

代码仓库

ibalazevic/TuckER
官方
pytorch
GitHub 中提及
adaruna3/explainable-kge
pytorch
GitHub 中提及
allenai/kb
pytorch
GitHub 中提及
Sujit-O/pykg2vec
tf
GitHub 中提及
luffycodes/neptune
pytorch
GitHub 中提及

基准测试

基准方法指标
link-prediction-on-fb15kTuckER
Hits@1: 0.741
Hits@10: 0.892
Hits@3: 0.833
MRR: 0.795
link-prediction-on-fb15k-237TuckER
Hits@1: 0.266
Hits@10: 0.544
Hits@3: 0.394
MRR: 0.358
link-prediction-on-wn18TuckER
Hits@1: 0.949
Hits@10: 0.958
Hits@3: 0.955
MRR: 0.953
link-prediction-on-wn18rrTuckER
Hits@1: 0.443
Hits@10: 0.526
Hits@3: 0.482
MRR: 0.470

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
TuckER:用于知识图谱补全的张量分解方法 | 论文 | HyperAI超神经