arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.00856cs.AIcs.CLcs.LG
收藏

SEEK:知识图谱的分段嵌入

SEEK: Segmented Embedding of Knowledge Graphs

Sun Yat-sen University(中山大学);Microsoft Research Asia(微软亚洲研究院)

Wentao Xu, Shun Zheng, Liang He, Bin Shao, Jian Yin, Tie-Yan Liu

摘要:

近年来,知识图谱嵌入成为人工智能的一个研究热点,在推荐、问答等下游应用中发挥着越来越重要的作用。然而,现有的知识图谱嵌入方法不能在模型复杂度和模型表达能力之间进行合理的权衡,使得它们还远不能令人满意。为了缓解这个问题,我们提出了一个轻量级的建模框架,可以实现高度竞争力的关系表达,而不增加模型的复杂性。我们的框架侧重于评分函数的设计,并强调了两个关键特征:1)促进充分的特征交互; 2)保持关系的对称性和反对称性。值得注意的是,由于评分函数的一般和优雅的设计,我们的框架可以将许多著名的现有方法作为特例。此外,在公共基准上的大量实验证明了我们的框架的效率和有效性。源代码和数据可在\url{https://github.com/Wentao-Xu/SEEK}上找到。

英文摘要:

In recent years, knowledge graph embedding becomes a pretty hot research topic of artificial intelligence and plays increasingly vital roles in various downstream applications, such as recommendation and question answering. However, existing methods for knowledge graph embedding can not make a proper trade-off between the model complexity and the model expressiveness, which makes them still far from satisfactory. To mitigate this problem, we propose a lightweight modeling framework that can achieve highly competitive relational expressiveness without increasing the model complexity. Our framework focuses on the design of scoring functions and highlights two critical characteristics: 1) facilitating sufficient feature interactions; 2) preserving both symmetry and antisymmetry properties of relations. It is noteworthy that owing to the general and elegant design of scoring functions, our framework can incorporate many famous existing methods as special cases. Moreover, extensive experiments on public benchmarks demonstrate the efficiency and effectiveness of our framework. Source codes and data can be found at \url{https://github.com/Wentao-Xu/SEEK}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑