Publications

K-Edit: Language Model Editing with Contextual Knowledge Awareness

Abstract

As the world changes, we need to be able to update our models and correct false information without costly retraining. Knowledge-based model editing enables precise modifications to the weights of large language models in order to modify the information encoded within. Recent approaches have seen success in enabling recall of edited information for thousands of edits at once. However, these approaches fail to produce edits that account for associated contextual information. We present K-Edit, an effective approach to generating contextually consistent knowledge edits. By using knowledge graphs, which maintain contextual consistency when an edge is edited, we are able to generate additional \textit{contextual edits} that ensure consistency of related information in the language model. Our experiments demonstrate significant improvements in multi-hop question answering while maintaining the general effectiveness and scalability of model edits.

Metadata

publication
arXiv preprint arXiv:2502.10626, 2025
year
2025
publication date
2025/2/15
authors
Elan Markowitz, Anil Ramakrishna, Ninareh Mehrabi, Charith Peris, Rahul Gupta, Kai-Wei Chang, Aram Galstyan
link
https://arxiv.org/abs/2502.10626
resource_link
https://arxiv.org/pdf/2502.10626
journal
arXiv preprint arXiv:2502.10626