3 个月前

用于建模多关系数据的嵌入翻译

用于建模多关系数据的嵌入翻译

摘要

我们研究多关系数据中实体与关系在低维向量空间中的嵌入问题。目标是提出一种易于训练、参数量较少且能够扩展至超大规模数据库的通用模型。为此,我们提出了TransE方法,该方法通过将关系建模为作用于实体低维嵌入上的平移操作来实现。尽管该假设形式简单,但实验证明其具有强大的表达能力:在两个知识库上的链接预测任务中,TransE显著优于现有最先进方法。此外,该方法还可成功应用于包含100万实体、2.5万个关系以及超过1700万条训练样本的大规模数据集。

基准测试

基准方法指标
link-prediction-on-fb122TransE
HITS@3: 58.9
Hits@10: 70.2
Hits@5: 64.2
MRR: 48.0
link-prediction-on-fb15kTransE
Hits@10: 0.471
MR: 125
link-prediction-on-fb15k-237TransE
Hits@1: 0.1987
Hits@10: .4709
MRR: 0.2904
link-prediction-on-umlsTransE
Hits@10: 0.989
MR: 1.84
link-prediction-on-wn18TransE
Hits@10: 0.754
MR: 263
link-prediction-on-wn18rrTransE
Hits@1: 0.4226
Hits@10: 0.5555
MRR: 0.4659

用 AI 构建 AI

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

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

Hyper Newsletters

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