LGAIFeb 15, 2025

K-Edit: Language Model Editing with Contextual Knowledge Awareness

Amazon
arXiv:2502.10626v24 citationsh-index: 50
Originality Incremental advance
AI Analysis

This addresses the need for efficient model updates to correct false information, though it is incremental as it builds on existing knowledge-based editing approaches.

The paper tackles the problem of updating large language models with new information without retraining, by introducing K-Edit, a method that uses knowledge graphs to generate contextually consistent edits, resulting in significant improvements in multi-hop question answering.

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.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes