arXiv:2411.08165v2 宣告类型: 替换 摘要: 知识图谱补全（KGC）任务旨在从不完整的三元组中推断缺失的实体。现有的基于嵌入的方法仅依赖于知识图谱中的三元组，这使其容易受到虚假关系模式和长尾实体的影响。另一方面，基于文本的方法在知识图谱三元组和自然语言之间存在语义差距。除了三元组之外，实体上下文（例如标签、描述、别名）在增强知识图谱方面也发挥着重要作用。为了应对这些局限性，我们提出了一种名为KGR3的上下文增强框架，用于KGC。KGR3由三个模块组成。首先，检索模块从知识图谱中收集支持三元组，从基础嵌入模型中收集可能的候选答案，并为每个相关实体检索上下文。然后，推理模块使用大型语言模型为每个查询三元组生成潜在答案。最后，重新排名模块将上述两个模块的候选答案结合起来，并微调LLM以提供最佳答案。广泛的实验表明，KGR3在广泛使用的数据集上一致性地提高了各种KGC方法的效果。具体而言，KGR3的最佳变体在FB15k-237和WN18RR数据集上的绝对Hits@1提高了12.3%和5.6%。