QuatRE：用于知识图谱嵌入的关系感知四元数模型
计算与语言
2022-03-09 v2 人工智能
机器学习
摘要
我们提出了一种简单而有效的嵌入模型，用于在知识图谱中学习实体与关系的四元数嵌入。我们的模型旨在四元数空间内利用哈密顿积，增强给定关系下头实体与尾实体之间的相关性。该模型通过进一步将每个关系与两个关系感知旋转相关联来实现这一目标，这两个旋转分别用于旋转头实体和尾实体的四元数嵌入。实验结果表明，我们所提出的模型在著名的知识图谱补全基准数据集上取得了最先进的性能。我们的代码可在 https://github.com/daiquocnguyen/QuatRE 获取。
引用
@article{arxiv.2009.12517,
title = {QuatRE: Relation-Aware Quaternions for Knowledge Graph Embeddings},
author = {Dai Quoc Nguyen and Thanh Vu and Tu Dinh Nguyen and Dinh Phung},
journal= {arXiv preprint arXiv:2009.12517},
year = {2022}
}
备注
Accepted to The ACM Web Conference 2022 (WWW '22) (Poster and Demo Track)