用关系特定的双曲锥建模异质层次结构
机器学习
2021-11-02 v2 社会与信息网络
摘要
层次关系在知识图谱(KG)所刻画的人类知识组织中普遍存在且不可或缺。层次关系的关键特性是其在实体上诱导出偏序关系，需对此建模以实现层次推理。然而，当前的 KG 嵌入只能建模单一的全局层次结构(单一全局偏序)，无法建模单个 KG 中存在的多种异质层次结构。本文提出 ConE (Cone Embedding)，一种能够同时建模知识图谱中多种层次与非层次关系的 KG 嵌入模型。ConE 将实体嵌入到双曲锥中，并将关系建模为锥之间的变换。特别地，ConE 利用双曲嵌入空间不同子空间中的锥包含约束来捕捉多种异质层次结构。在标准知识图谱基准上的实验表明，ConE 在层次推理任务以及层次图上的知识图谱补全任务中均取得了 state-of-the-art 性能。具体而言，我们的方法在 WN18RR 上取得了 45.3% 的新 state-of-the-art Hits@1，在 DDB14 上取得了 16.1% (0.231 MRR)。在层次推理任务上，我们的方法在三个数据集上的平均表现优于先前最佳结果 20%。
引用
@article{arxiv.2110.14923,
title = {Modeling Heterogeneous Hierarchies with Relation-specific Hyperbolic Cones},
author = {Yushi Bai and Rex Ying and Hongyu Ren and Jure Leskovec},
journal= {arXiv preprint arXiv:2110.14923},
year = {2021}
}
备注
To be published in NeurIPS 2021, full version including appendix