EAIR: Entity-aware Inference-Time Knowledge Routing for Multi-Hop Knowledge Editing

Jungyu Lee, Kunhui Lee, Gyun Lee, Seung-Hoon Na


Abstract
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.
Anthology ID:
2026.findings-acl.873
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
17622–17645
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.873/
DOI:
Bibkey:
Cite (ACL):
Jungyu Lee, Kunhui Lee, Gyun Lee, and Seung-Hoon Na. 2026. EAIR: Entity-aware Inference-Time Knowledge Routing for Multi-Hop Knowledge Editing. In Findings of the Association for Computational Linguistics: ACL 2026, pages 17622–17645, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
EAIR: Entity-aware Inference-Time Knowledge Routing for Multi-Hop Knowledge Editing (Lee et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.873.pdf
Checklist:
 2026.findings-acl.873.checklist.pdf