Jungyu Lee
2026
EAIR: Entity-aware Inference-Time Knowledge Routing for Multi-Hop Knowledge Editing
Jungyu Lee | Kunhui Lee | Gyun Lee | Seung-Hoon Na
Findings of the Association for Computational Linguistics: ACL 2026
Jungyu Lee | Kunhui Lee | Gyun Lee | Seung-Hoon Na
Findings of the Association for Computational Linguistics: ACL 2026
Existing in-context editing (ICE) methods for multi-hop knowledge editing commonly suffer from paraphrase sensitivity, which refers to the phenomenon where these methods are not sufficiently robust to paraphrased multi-hop questions. To improve retrieval accuracy and knowledge routing to address paraphrase sensitivity, this paper proposes a novel entity-aware inference-time knowledge routing method, referred to as EAIR, which consists of four major steps: 1) Entity-referential query decomposition, which decomposes the original question into multiple entity-referential sub-question instructions; 2) Entity-aware retrieval, which leverages the previously reference-resolved topic entity in the retrieval step; 3) Evidence-conditioned contrastive decoding, which discourages the model from relying on its parametric knowledge and steers the model toward following retrieved edits; 4) Reflection-based knowledge routing, which additionally filters decoding results using refusal-style reflection to mitigate the risk introduced by contrastive decoding. Experimental results across the MQuAKE benchmark family and model scales show that EAIR achieves the highest strict case accuracy in 11 of 12 settings, substantially reducing paraphrase sensitivity.