计算机科学
嵌入
二元关系
Dirichlet分布
推论
理论计算机科学
稳健性(进化)
关系(数据库)
欧几里得空间
知识图
子空间拓扑
人工智能
机器学习
数学
数据挖掘
离散数学
数学分析
生物化学
化学
纯数学
基因
边值问题
作者
Feiyang Wang,Zhongbao Zhang,Li Sun,Junda Ye,Yan Yang
标识
DOI:10.1145/3485447.3512028
摘要
Knowledge graph embedding aims to learn representations of entities and relations in low-dimensional space. Recently, extensive studies combine the characteristics of knowledge graphs with different geometric spaces, including Euclidean space, complex space, hyperbolic space and others, which achieves significant progress in representation learning. However, existing methods are subject to at least one of the following limitations: 1) ignoring the uncertainty, 2) incapability of complex relation patterns. To address the above issues simultaneously, we propose a novel model named DiriE, which embeds entities as Dirichlet distributions and relations as multinomial distributions. DiriE employs Bayesian inference to measure the relations between entities and learns binary embeddings of knowledge graphs for modeling complex relation patterns. Additionally, we propose a two-step negative triple generation method that generates negative triples of both entities and relations. We conduct a solid theoretical analysis to demonstrate the effectiveness and robustness of our method, including the expressiveness of complex relation patterns and the ability to model uncertainty. Furthermore, extensive experiments show that our method outperforms state-of-the-art methods in link prediction on benchmark datasets.
科研通智能强力驱动
Strongly Powered by AbleSci AI