中文

卷积复数知识图谱嵌入

机器学习 2021-06-10 v3 计算与语言 机器学习

摘要

本文研究知识图谱连续向量表示学习以预测缺失链接的问题。我们提出一种称为ConEx的新方法,其通过利用二维卷积与复值嵌入向量的Hermitian内积的组合来推断缺失链接。我们在WN18RR、FB15K-237、KINSHIP和UMLS基准数据集上将ConEx与最先进方法进行比较。实验结果表明,在全部数据集的链接预测任务上,ConEx性能优于RotatE、QuatE和TuckER等最先进方法,同时所需参数至少减少8倍。我们通过提供开源实现(包含训练、评估脚本及预训练模型,见https://github.com/conex-kge/ConEx)来确保结果可复现。

关键词

引用

@article{arxiv.2008.03130,
  title  = {Convolutional Complex Knowledge Graph Embeddings},
  author = {Caglar Demir and Axel-Cyrille Ngonga Ngomo},
  journal= {arXiv preprint arXiv:2008.03130},
  year   = {2021}
}