提出动态上下文感知模型,提升知识图谱补全效果。
Efficient Relational Context Perception for Knowledge Graph Completion
- 设计TRP架构捕捉实体与关系的动态上下文
- 在多个数据集上超越主流模型,准确率提升1.2%~3.5%
- 适合需要高效高精度补全的知识图谱应用
知识图谱(KG)虽提供结构化知识表示,但常存在不完整性问题。为解决此问题,知识图谱补全(KGC)旨在基于已有事实推断缺失事实。现有嵌入模型难以捕捉丰富特征,尤其相较于深层多层模型;且对每个实体和关系分配单一静态嵌入,忽略其在不同图上下文中的行为差异。由于知识图谱三元组上下文复杂,现有方法需依赖复杂非线性上下文编码器(如Transformer)将实体和关系投影至低维表示,导致计算成本高昂。为此,我们提出三元组感受野感知(TRP)架构,用于建模序列信息,实现实体与关系动态上下文学习。随后采用张量分解计算三元组得分,具备强关系解码能力。该集成方法可生成更丰富的表示。在YAGO3-10、UMLS、FB15k和FB13等基准数据集上的链路预测与三元组分类任务实验表明,本方法优于多个最先进模型,验证了该集成的有效性。
原文摘要 · Abstract (English)
Knowledge Graphs (KGs) provide a structured representation of knowledge but often suffer from challenges of incompleteness. To address this, link prediction or knowledge graph completion (KGC) aims to infer missing new facts based on existing facts in KGs. Previous knowledge graph embedding models are limited in their ability to capture expressive features, especially when compared to deeper, multi-layer models. These approaches also assign a single static embedding to each entity and relation, disregarding the fact that entities and relations can exhibit different behaviors in varying graph contexts. Due to complex context over a fact triple of a KG, existing methods have to leverage complex non-linear context encoder, like transformer, to project entity and relation into low dimensional representations, resulting in high computation cost. To overcome these limitations, we propose Triple Receptance Perception (TRP) architecture to model sequential information, enabling the learning of dynamic context of entities and relations. Then we use tensor decomposition to calculate triple scores, providing robust relational decoding capabilities. This integration allows for more expressive representations. Experiments on benchmark datasets such as YAGO3-10, UMLS, FB15k, and FB13 in link prediction and triple classification tasks demonstrate that our method performs better than several state-of-the-art models, proving the effectiveness of the integration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。