4 个月前

Query2Box:使用盒状嵌入在向量空间中对知识图谱进行推理

Query2Box:使用盒状嵌入在向量空间中对知识图谱进行推理

摘要

在大规模不完整知识图谱(KGs)上回答复杂的逻辑查询是一项基本而具有挑战性的任务。最近,一种有前景的方法是将知识图谱实体和查询嵌入到向量空间中,使得能够回答查询的实体在向量空间中靠近查询。然而,先前的研究将查询建模为向量空间中的单个点,这存在一个问题,因为一个复杂的查询可能代表其答案实体的一个潜在较大的集合,但如何将这样一个集合表示为单个点尚不清楚。此外,先前的工作只能处理使用合取($\wedge$)和存在量化($\exists$)的查询。处理包含逻辑析取($\vee$)的查询仍然是一个未解决的问题。本文提出了一种基于嵌入的框架——query2box,用于在大规模且不完整的知识图谱中处理任意包含 $\wedge$、$\vee$ 和 $\exists$ 操作符的逻辑查询。我们的主要见解是,可以将查询嵌入为盒子(即超矩形),其中盒子内部的一组点对应于查询的答案实体集。我们展示了合取可以自然地表示为盒子的交集,并证明了一个负面结果:处理析取需要维度与知识图谱实体数量成比例的嵌入。然而,通过将查询转换为析取范式(Disjunctive Normal Form),我们表明 query2box 能够以可扩展的方式处理任意包含 $\wedge$、$\vee$ 和 $\exists$ 的逻辑查询。我们在三个大型知识图谱上验证了 query2box 的有效性,并显示 query2box 在这些数据集上的表现比现有最佳方法提高了高达 25% 的相对性能。

代码仓库

snap-stanford/KGReasoning
pytorch
GitHub 中提及
pminervini/kgreasoning
pytorch
GitHub 中提及
Blidge/KGReasoning
pytorch
GitHub 中提及
francoisluus/KGReasoning
pytorch
GitHub 中提及
kjh9503/caqr
pytorch
GitHub 中提及
hkust-knowcomp/efo-1-qa-benchmark
pytorch
GitHub 中提及
Jacfger/simple-stuffs
pytorch
GitHub 中提及
hyren/query2box
官方
pytorch
LHY-24/KG-Compilation
pytorch
GitHub 中提及

基准测试

基准方法指标
complex-query-answering-on-fb15kQ2B
MRR 1p: 0.68
MRR 2i: 0.551
MRR 2p: 0.21
MRR 2u: 0.351
MRR 3i: 0.665
MRR 3p: 0.142
MRR ip: 0.261
MRR pi: 0.394
MRR up: 0.167
complex-query-answering-on-fb15k-237Q2B
MRR 1p: 0.406
MRR 2i: 0.295
MRR 2p: 0.094
MRR 2u: 0.113
MRR 3i: 0.423
MRR 3p: 0.068
MRR ip: 0.126
MRR pi: 0.212
MRR up: 0.076
complex-query-answering-on-nell-995Q2B
MRR 1p: 0.422
MRR 2i: 0.333
MRR 2p: 0.140
MRR 2u: 0.113
MRR 3i: 0.445
MRR 3p: 0.112
MRR ip: 0.168
MRR pi: 0.224
MRR up: 0.1103

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
Query2Box:使用盒状嵌入在向量空间中对知识图谱进行推理 | 论文 | HyperAI超神经