
摘要
神经链接预测器在识别大规模知识图谱中的缺失边方面具有极大的实用性。然而,如何利用这些模型来回答多个领域中出现的更复杂的查询仍然是一个未解之谜,尤其是涉及逻辑合取($\land$)、析取($\lor$)和存在量词($\exists$)的查询,同时还要考虑缺失边的问题。在这项工作中,我们提出了一种框架,用于高效地回答不完整知识图谱上的复杂查询。我们将每个查询转换为端到端可微的目标函数,其中每个原子的真值由预训练的神经链接预测器计算得出。随后,我们分析了两种解决优化问题的方法,包括基于梯度的搜索和组合搜索。在实验中,所提出的这种方法在无需大量且多样的复杂查询训练的情况下,比现有方法——即在数百万生成查询上训练的黑盒神经模型——产生了更准确的结果。使用数量级较少的训练数据,我们在包含事实信息的不同知识图谱上获得了从8%到40%不等的Hits@3相对提升。最后,我们展示了可以解释我们的模型对每个复杂查询原子所识别的中间解决方案的结果。我们的所有源代码和数据集均可在线获取,网址为 https://github.com/uclnlp/cqd。
代码仓库
dice-group/dice-embeddings
pytorch
GitHub 中提及
Blidge/KGReasoning
pytorch
GitHub 中提及
pminervini/kgreasoning
pytorch
GitHub 中提及
LHY-24/KG-Compilation
pytorch
GitHub 中提及
uclnlp/cqd
官方
pytorch
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| complex-query-answering-on-fb15k | CQD-CO | Hits@3 1p: 0.918 Hits@3 2i: 0.796 Hits@3 2p: 0.454 Hits@3 2u: 0.816 Hits@3 3i: 0.837 Hits@3 3p: 0.191 Hits@3 ip: 0.336 Hits@3 pi: 0.513 Hits@3 up: 0.319 |
| complex-query-answering-on-fb15k | CQD | MRR 1p: 0.892 MRR 2i: 0.771 MRR 2p: 0.653 MRR 2u: 0.723 MRR 3i: 0.806 MRR ip: 0.716 |
| complex-query-answering-on-fb15k | CQD-Beam | Hits@3 1p: 0.918 Hits@3 2i: 0.796 Hits@3 2p: 0.779 Hits@3 2u: 0.839 Hits@3 3i: 0.837 Hits@3 3p: 0.577 Hits@3 ip: 0.375 Hits@3 pi: 0.658 Hits@3 up: 0.345 |
| complex-query-answering-on-fb15k-237 | CQD | MRR 3i: 0.486 |
| complex-query-answering-on-fb15k-237 | CQD-CO | Hits@3 1p: 0.512 Hits@3 2i: 35.2 Hits@3 2p: 0.213 Hits@3 2u: 0.281 Hits@3 3i: 0.457 Hits@3 3p: 0.131 Hits@3 ip: 0.146 Hits@3 pi: 0.222 Hits@3 up: 0.132 |
| complex-query-answering-on-fb15k-237 | CQD-Beam | Hits@3 1p: 0.512 Hits@3 2i: 0.352 Hits@3 2p: 0.288 Hits@3 2u: 0.284 Hits@3 3i: 0.457 Hits@3 3p: 0.221 Hits@3 ip: 0.129 Hits@3 pi: 0.249 Hits@3 up: 0.121 |
| complex-query-answering-on-nell-995 | CQD | MRR 1p: 0.604 MRR 2i: 0.436 MRR ip: 0.256 |
| complex-query-answering-on-nell995 | CQD-CO | Hits@3 1p: 0.667 Hits@3 2i: 0.410 Hits@3 2p: 0.265 Hits@3 2u: 0.531 Hits@3 3i: 0.529 Hits@3 3p: 0.220 Hits@3 ip: 0.196 Hits@3 pi: 0.302 Hits@3 up: 0.194 |
| complex-query-answering-on-nell995 | CQD-Beam | Hits@3 2p: 0.350 Hits@3 3p: 0.288 Hits@3 ip: 0.171 Hits@3 pi: 0.277 Hits@3 up: 0.156 |