PokeMQA: Programmable knowledge editing for Multi-hop Question Answering
Hengrui Gu, Kaixiong Zhou, Xiaotian Han, Ninghao Liu, Ruobing Wang, Xin Wang
TL;DR
PokeMQA tackles the problem of updating multi-hop question answering with up-to-date facts without retraining. It decouples question decomposition from knowledge editing via a two-stage programmable scope detector and an external memory of edits, augmented by a knowledge prompt that sources entity-context from Wikidata. The approach yields superior Hop-Acc and accuracy across three LLM backbones and two MQUAKE datasets, supported by ablation showing the indispensability of the scope detector and the contextual knowledge prompt under varying edit batch sizes. This work advances reliable reasoning in knowledge editing for MQA and demonstrates practical scalability to modern LLMs, with noted limitations around detector design and security considerations for memory-based editing.
Abstract
Multi-hop question answering (MQA) is one of the challenging tasks to evaluate machine's comprehension and reasoning abilities, where large language models (LLMs) have widely achieved the human-comparable performance. Due to the dynamics of knowledge facts in real world, knowledge editing has been explored to update model with the up-to-date facts while avoiding expensive re-training or fine-tuning. Starting from the edited fact, the updated model needs to provide cascading changes in the chain of MQA. The previous art simply adopts a mix-up prompt to instruct LLMs conducting multiple reasoning tasks sequentially, including question decomposition, answer generation, and conflict checking via comparing with edited facts. However, the coupling of these functionally-diverse reasoning tasks inhibits LLMs' advantages in comprehending and answering questions while disturbing them with the unskilled task of conflict checking. We thus propose a framework, Programmable knowledge editing for Multi-hop Question Answering (PokeMQA), to decouple the jobs. Specifically, we prompt LLMs to decompose knowledge-augmented multi-hop question, while interacting with a detached trainable scope detector to modulate LLMs behavior depending on external conflict signal. The experiments on three LLM backbones and two benchmark datasets validate our superiority in knowledge editing of MQA, outperforming all competitors by a large margin in almost all settings and consistently producing reliable reasoning process.
