科研之友
微信
新浪微博
Facebook
分享链接
摘要
知识图谱嵌入（KGE）是预测知识图谱（KGs）中缺失链接的重要工具，它将知识图谱中的实体和关系嵌入到连续低维空间中，并尽可能地保留原数据中隐含的各种信息。近年来，一些知识图谱嵌入方法利用极坐标系对知识图谱中普遍存在的语义层次结构进行建模，提升了链路预测任务的性能。然而，这些方法在建模关系时，使用了简单的标度变换并过度关注于实体的层次差，这在一定程度上限制了模型的拟合力。为了应对上述问题，提出了基于层次保留的知识图谱嵌入方法（RHKE），它在建模知识图谱中的关系时考虑了实体本身的层次。具体来说，提出了混合变换，它包含一个倍率项和一个偏差项，当实体层次较低或较高时，标度变换主要受偏差项或倍率项影响。此外，由于变换后模型丢失了实体原本的层次，RHKE使用层次修正项，它将头尾实体的原本层次用不同比例组合后作为关系的附加信息。在多个公开数据集上的实验结果显示，RHKE在链路预测上的性能优于现有的语义层次模型。
- 单位
