3 个月前

KERMIT:基于逆变换增强关系建模的知识图谱补全

KERMIT:基于逆变换增强关系建模的知识图谱补全

摘要

知识图谱补全(Knowledge Graph Completion, KGC)的核心任务是利用已有信息填补知识图谱中缺失的三元组。基于文本的方法依赖于三元组的文本描述进行预测,但在实际应用中常因描述信息不足而导致预测精度下降,这一问题根植于数据集本身,难以仅通过模型设计加以解决。为应对上述挑战并保障数据的一致性,本文提出以下三方面改进:首先,利用大规模语言模型(Large Language Models, LLMs)生成语义连贯的三元组描述,有效弥合查询与答案之间的语义鸿沟;其次,引入逆关系构建对称图结构,从而扩充训练样本,增强模型的泛化能力;此外,充分挖掘知识图谱中固有的标签信息,对现有对比学习框架进行改进,使其转变为完全监督的学习范式。实验结果表明,所提方法在WN18RR和FB15k-237两个标准数据集上均取得显著性能提升。根据标准评估指标,该方法在WN18RR数据集上的Hit@1指标提升4.2%,在FB15k-237数据集上的Hit@3指标提升3.4%,充分验证了其优越性。

代码仓库

lirt1231/kermit
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
link-prediction-on-fb15k-237KERMIT
Hits@1: 0.266
Hits@10: 0.547
Hits@3: 0.396
MRR: 0.359
link-prediction-on-wn18rrKERMIT
Hits@1: 0.629
Hits@10: 0.832
Hits@3: 0.738
MRR: 0.700

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
KERMIT:基于逆变换增强关系建模的知识图谱补全 | 论文 | HyperAI超神经