4 个月前

基于神经链接预测器的复杂查询回答

基于神经链接预测器的复杂查询回答

摘要

神经链接预测器在识别大规模知识图谱中的缺失边方面具有极大的实用性。然而,如何利用这些模型来回答多个领域中出现的更复杂的查询仍然是一个未解之谜,尤其是涉及逻辑合取($\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-fb15kCQD-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-fb15kCQD
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-fb15kCQD-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-237CQD
MRR 3i: 0.486
complex-query-answering-on-fb15k-237CQD-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-237CQD-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-995CQD
MRR 1p: 0.604
MRR 2i: 0.436
MRR ip: 0.256
complex-query-answering-on-nell995CQD-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-nell995CQD-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

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于神经链接预测器的复杂查询回答 | 论文 | HyperAI超神经