4 个月前

知识图谱的高效并行翻译嵌入

知识图谱的高效并行翻译嵌入

摘要

知识图谱嵌入旨在将知识图谱中的实体和关系嵌入到低维向量空间中。翻译嵌入方法将关系视为从头实体到尾实体的翻译,这些方法在知识图谱嵌入方法中取得了最先进的结果。然而,这些方法的一个主要限制是训练过程耗时较长,对于大型知识图谱可能需要数天甚至数周的时间,从而导致实际应用中的巨大困难。本文提出了一种高效的并行框架——ParTrans-X,该框架通过利用知识图谱的独特结构,使得翻译嵌入方法能够在无锁的情况下实现并行化。实验在两个数据集上对三种典型的翻译嵌入方法(即TransE [3]、TransH [17]以及一个更高效的变体TransE-AdaGrad [10])进行了验证,结果显示ParTrans-X可以将训练过程加速超过一个数量级。

代码仓库

基准测试

基准方法指标
link-prediction-on-fb15k-1ParTransH
Hits@10: 0.468
MR: 60
link-prediction-on-fb15k-filteredParTransH
Hits@10: 65.7
MR: 60
link-prediction-on-wn18ParTransH
Hits@10: 0.668
MR: 215
link-prediction-on-wn18-filteredParTransH
Hits@10: 76.6
MR: 203

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
知识图谱的高效并行翻译嵌入 | 论文 | HyperAI超神经