Zhaocheng Zhu, Mikhail Galkin, Zuobai Zhang, Jian Tang
Answering complex first-order logic (FOL) queries on knowledge graphs is a fundamental task for multi-hop reasoning. Traditional symbolic methods traverse a complete knowledge graph to extract the answers, which provides good interpretation for each step. Recent neural methods learn geometric embeddings for complex queries. These methods can generalize to incomplete knowledge graphs, but their reasoning process is hard to interpret. In this paper, we propose Graph Neural Network Query Executor (GNN-QE), a neural-symbolic model that enjoys the advantages of both worlds. GNN-QE decomposes a complex FOL query into relation projections and logical operations over fuzzy sets, which provides interpretability for intermediate variables. To reason about the missing links, GNN-QE adapts a graph neural network from knowledge graph completion to execute the relation projections, and models the logical operations with product fuzzy logic. Experiments on 3 datasets show that GNN-QE significantly improves over previous state-of-the-art models in answering FOL queries. Meanwhile, GNN-QE can predict the number of answers without explicit supervision, and provide visualizations for intermediate variables.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Knowledge Graphs | FB15k | MRR 1p | 0.885 | GNN-QE |
| Knowledge Graphs | FB15k | MRR 2i | 0.797 | GNN-QE |
| Knowledge Graphs | FB15k | MRR 2p | 0.693 | GNN-QE |
| Knowledge Graphs | FB15k | MRR 2u | 0.741 | GNN-QE |
| Knowledge Graphs | FB15k | MRR 3i | 0.835 | GNN-QE |
| Knowledge Graphs | FB15k | MRR 3p | 0.587 | GNN-QE |
| Knowledge Graphs | FB15k | MRR ip | 0.704 | GNN-QE |
| Knowledge Graphs | FB15k | MRR pi | 0.699 | GNN-QE |
| Knowledge Graphs | FB15k | MRR up | 0.61 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR 1p | 0.533 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR 2i | 0.424 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR 2p | 0.189 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR 2u | 0.159 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR 3i | 0.525 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR 3p | 0.149 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR ip | 0.189 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR pi | 0.308 | GNN-QE |
| Knowledge Graphs | NELL-995 | MRR up | 0.126 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR 1p | 0.428 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR 2i | 0.383 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR 2p | 0.147 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR 2u | 0.162 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR 3i | 0.541 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR 3p | 0.118 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR ip | 0.189 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR pi | 0.311 | GNN-QE |
| Knowledge Graphs | FB15k-237 | MRR up | 0.134 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR 1p | 0.885 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR 2i | 0.797 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR 2p | 0.693 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR 2u | 0.741 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR 3i | 0.835 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR 3p | 0.587 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR ip | 0.704 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR pi | 0.699 | GNN-QE |
| Knowledge Graph Completion | FB15k | MRR up | 0.61 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR 1p | 0.533 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR 2i | 0.424 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR 2p | 0.189 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR 2u | 0.159 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR 3i | 0.525 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR 3p | 0.149 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR ip | 0.189 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR pi | 0.308 | GNN-QE |
| Knowledge Graph Completion | NELL-995 | MRR up | 0.126 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR 1p | 0.428 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR 2i | 0.383 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR 2p | 0.147 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR 2u | 0.162 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR 3i | 0.541 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR 3p | 0.118 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR ip | 0.189 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR pi | 0.311 | GNN-QE |
| Knowledge Graph Completion | FB15k-237 | MRR up | 0.134 | GNN-QE |
| Large Language Model | FB15k | MRR 1p | 0.885 | GNN-QE |
| Large Language Model | FB15k | MRR 2i | 0.797 | GNN-QE |
| Large Language Model | FB15k | MRR 2p | 0.693 | GNN-QE |
| Large Language Model | FB15k | MRR 2u | 0.741 | GNN-QE |
| Large Language Model | FB15k | MRR 3i | 0.835 | GNN-QE |
| Large Language Model | FB15k | MRR 3p | 0.587 | GNN-QE |
| Large Language Model | FB15k | MRR ip | 0.704 | GNN-QE |
| Large Language Model | FB15k | MRR pi | 0.699 | GNN-QE |
| Large Language Model | FB15k | MRR up | 0.61 | GNN-QE |
| Large Language Model | NELL-995 | MRR 1p | 0.533 | GNN-QE |
| Large Language Model | NELL-995 | MRR 2i | 0.424 | GNN-QE |
| Large Language Model | NELL-995 | MRR 2p | 0.189 | GNN-QE |
| Large Language Model | NELL-995 | MRR 2u | 0.159 | GNN-QE |
| Large Language Model | NELL-995 | MRR 3i | 0.525 | GNN-QE |
| Large Language Model | NELL-995 | MRR 3p | 0.149 | GNN-QE |
| Large Language Model | NELL-995 | MRR ip | 0.189 | GNN-QE |
| Large Language Model | NELL-995 | MRR pi | 0.308 | GNN-QE |
| Large Language Model | NELL-995 | MRR up | 0.126 | GNN-QE |
| Large Language Model | FB15k-237 | MRR 1p | 0.428 | GNN-QE |
| Large Language Model | FB15k-237 | MRR 2i | 0.383 | GNN-QE |
| Large Language Model | FB15k-237 | MRR 2p | 0.147 | GNN-QE |
| Large Language Model | FB15k-237 | MRR 2u | 0.162 | GNN-QE |
| Large Language Model | FB15k-237 | MRR 3i | 0.541 | GNN-QE |
| Large Language Model | FB15k-237 | MRR 3p | 0.118 | GNN-QE |
| Large Language Model | FB15k-237 | MRR ip | 0.189 | GNN-QE |
| Large Language Model | FB15k-237 | MRR pi | 0.311 | GNN-QE |
| Large Language Model | FB15k-237 | MRR up | 0.134 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR 1p | 0.885 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR 2i | 0.797 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR 2p | 0.693 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR 2u | 0.741 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR 3i | 0.835 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR 3p | 0.587 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR ip | 0.704 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR pi | 0.699 | GNN-QE |
| Inductive knowledge graph completion | FB15k | MRR up | 0.61 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR 1p | 0.533 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR 2i | 0.424 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR 2p | 0.189 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR 2u | 0.159 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR 3i | 0.525 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR 3p | 0.149 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR ip | 0.189 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR pi | 0.308 | GNN-QE |
| Inductive knowledge graph completion | NELL-995 | MRR up | 0.126 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR 1p | 0.428 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR 2i | 0.383 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR 2p | 0.147 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR 2u | 0.162 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR 3i | 0.541 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR 3p | 0.118 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR ip | 0.189 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR pi | 0.311 | GNN-QE |
| Inductive knowledge graph completion | FB15k-237 | MRR up | 0.134 | GNN-QE |