References & Citations
Computer Science > Computation and Language
Title: SQUIRE: A Sequence-to-sequence Framework for Multi-hop Knowledge Graph Reasoning
(Submitted on 17 Jan 2022 (v1), last revised 31 Oct 2022 (this version, v3))
Abstract: Multi-hop knowledge graph (KG) reasoning has been widely studied in recent years to provide interpretable predictions on missing links with evidential paths. Most previous works use reinforcement learning (RL) based methods that learn to navigate the path towards the target entity. However, these methods suffer from slow and poor convergence, and they may fail to infer a certain path when there is a missing edge along the path. Here we present SQUIRE, the first Sequence-to-sequence based multi-hop reasoning framework, which utilizes an encoder-decoder Transformer structure to translate the query to a path. Our framework brings about two benefits: (1) It can learn and predict in an end-to-end fashion, which gives better and faster convergence; (2) Our Transformer model does not rely on existing edges to generate the path, and has the flexibility to complete missing edges along the path, especially in sparse KGs. Experiments on standard and sparse KGs show that our approach yields significant improvement over prior methods, while converging 4x-7x faster.
Submission history
From: Yushi Bai [view email][v1] Mon, 17 Jan 2022 04:22:54 GMT (364kb,D)
[v2] Fri, 28 Oct 2022 17:27:53 GMT (460kb,D)
[v3] Mon, 31 Oct 2022 10:57:11 GMT (461kb,D)
Link back to: arXiv, form interface, contact.