KLOD: Locality-Preserving Knowledge Editing via Non-Target Distribution Preservation

Hojun Jeong, Gyunyeop Kim, Sangwoo Kang


Abstract
Fine-tuning-based knowledge editing is simple and architecture-agnostic, but standard cross-entropy increases the edited target probability without explicitly constraining changes in the non-target output distribution. In sequential editing, such unconstrained redistribution can accumulate as distributional drift and contribute to locality degradation. We propose KLOD, a bounded and distribution-preserving objective for fine-tuning-based knowledge editing that separates the intended target update from distributions that should remain stable. KLOD stops target amplification once a probability threshold is reached, while preserving the target-excluded non-target distribution at target positions and the full next-token distribution at prefix positions. Experiments on CounterFact and ZsRE with LLaMA3-8B-Instruct and Qwen2.5-7B-Instruct show that KLOD substantially mitigates locality degradation while maintaining high edit reliability. The target probability threshold further provides a controllable Generalization–Locality trade-off. Ablation, multi-seed, and distributional KL analyses support the interpretation that KLOD’s locality gains are associated with preserving output distributions rather than simply weakening the edit.
Anthology ID:
2026.findings-emnlp.1241
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
26264–26286
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1241/
DOI:
10.18653/2026.findings-emnlp.1241
Bibkey:
Cite (ACL):
Hojun Jeong, Gyunyeop Kim, and Sangwoo Kang. 2026. KLOD: Locality-Preserving Knowledge Editing via Non-Target Distribution Preservation. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 26264–26286, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
KLOD: Locality-Preserving Knowledge Editing via Non-Target Distribution Preservation (Jeong et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1241.pdf