arXiv:2503.03091cs.CL2025-03KDD被引 8

不依赖描述和负采样,用图结构信息提升知识图谱补全效果

MuCo-KGC: Multi-Context-Aware Knowledge Graph Completion

  • 利用实体和关系的上下文结构信息进行预测
  • 在多个数据集上显著优于现有方法,最高提升20.15%
  • 适合无实体描述且追求高效推理的场景

知识图谱补全(KGC)旨在预测知识图谱中缺失的实体或关系,但图谱常存在数据不完整问题。传统嵌入方法如TransE和ComplEx在尾实体预测上有进展,却难以泛化到测试时未见实体。文本模型虽缓解此问题,但依赖负三元组采样,带来高计算开销、语义不一致和数据不平衡。近期如KG-BERT的方法依赖实体描述,而这类信息在多数图谱中缺失。现有方法忽视了与实体和关系相关的图结构信息。为此,我们提出多上下文感知的知识图谱补全模型MuCo-KGC,通过利用图中关联实体和关系的上下文信息来预测尾实体。该模型无需实体描述和负采样,大幅降低计算复杂度并提升性能。在FB15k-237、WN18RR、CoDEx-S和CoDEx-M等标准数据集上的实验表明,MuCo-KGC在三个数据集上超越当前最优方法。尤其在WN18RR、CoDEx-S和CoDEx-M上,平均正确率(MRR)分别提升1.63%、3.77%和20.15%,验证了其在KGC任务中的有效性。

原文摘要 · Abstract (English)

Knowledge graph completion (KGC) seeks to predict missing entities (e.g., heads or tails) or relationships in knowledge graphs (KGs), which often contain incomplete data. Traditional embedding-based methods, such as TransE and ComplEx, have improved tail entity prediction but struggle to generalize to unseen entities during testing. Textual-based models mitigate this issue by leveraging additional semantic context; however, their reliance on negative triplet sampling introduces high computational overhead, semantic inconsistencies, and data imbalance. Recent approaches, like KG-BERT, show promise but depend heavily on entity descriptions, which are often unavailable in KGs. Critically, existing methods overlook valuable structural information in the KG related to the entities and relationships. To address these challenges, we propose Multi-Context-Aware Knowledge Graph Completion (MuCo-KGC), a novel model that utilizes contextual information from linked entities and relations within the graph to predict tail entities. MuCo-KGC eliminates the need for entity descriptions and negative triplet sampling, significantly reducing computational complexity while enhancing performance. Our experiments on standard datasets, including FB15k-237, WN18RR, CoDEx-S, and CoDEx-M, demonstrate that MuCo-KGC outperforms state-of-the-art methods on three datasets. Notably, MuCo-KGC improves MRR on WN18RR, and CoDEx-S and CoDEx-M datasets by $1.63\%$, and $3.77\%$ and $20.15\%$ respectively, demonstrating its effectiveness for KGC tasks.

知识图谱图神经网络补全结构信息

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。