通过关系感知的锚点增强,提升知识图谱补全的准确性
Knowledge Graph Completion with Relation-Aware Anchor Enhancement
- 利用头实体的关系邻域生成锚点,提供目标实体参考
- 使查询嵌入更聚焦,显著提升链接预测性能
- 可无缝集成现有方法,适合知识图谱研究者
基于文本的知识图谱补全方法利用预训练语言模型(PLM)增强原始三元组的语义关联。典型方法将包含实体和关系的文本查询及其候选实体映射为特征向量,并最大化有效三元组的概率。随着大语言模型的发展,这类方法表现日益出色。我们发现:查询中头实体的关系感知邻居可作为有效上下文,提升嵌入的判别性。为此提出关系感知锚点增强方法(RAA-KGC):首先在头实体的关系邻域内生成锚点实体;然后通过拉近查询嵌入与锚点邻域的距离,使其更适用于目标实体匹配。大量实验验证了RAA-KGC的有效性,且可显著提升当前领先方法的性能,无需大幅修改。
原文摘要 · Abstract (English)
Text-based knowledge graph completion methods take advantage of pre-trained language models (PLM) to enhance intrinsic semantic connections of raw triplets with detailed text descriptions. Typical methods in this branch map an input query (textual descriptions associated with an entity and a relation) and its candidate entities into feature vectors, respectively, and then maximize the probability of valid triples. These methods are gaining promising performance and increasing attention for the rapid development of large language models. According to the property of the language models, the more related and specific context information the input query provides, the more discriminative the resultant embedding will be. In this paper, through observation and validation, we find a neglected fact that the relation-aware neighbors of the head entities in queries could act as effective contexts for more precise link prediction. Driven by this finding, we propose a relation-aware anchor enhanced knowledge graph completion method (RAA-KGC). Specifically, in our method, to provide a reference of what might the target entity be like, we first generate anchor entities within the relation-aware neighborhood of the head entity. Then, by pulling the query embedding towards the neighborhoods of the anchors, it is tuned to be more discriminative for target entity matching. The results of our extensive experiments not only validate the efficacy of RAA-KGC but also reveal that by integrating our relation-aware anchor enhancement strategy, the performance of current leading methods can be notably enhanced without substantial modifications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。