Lifelong Knowledge Editing requires Better Regularization

Akshat Gupta, Phudish Prateepamornkul, Maochuan Lu, Ahmed Alaa, Thomas Hartvigsen +1 more
2/3/2025
cs.CLcs.AIcs.LG

No official code found. Best implementation: scalable-model-editing/knowledge-editing-regularization (1 star, last commit 2025-05-23).

Abstract

Knowledge editing is a promising way to improve factuality in large language models, but recent studies have shown significant model degradation during sequential editing. In this paper, we formalize the popular locate-then-edit methods as a two-step fine-tuning process, allowing us to precisely identify the root cause of this degradation. We show that model degradation occurs due to (1) over-optimization of internal activations and (2) continuous norm-growth of edited matrices. To mitigate these issues, we introduce two regularization techniques: (1) Most-Probable Early Stopping (MPES) and (2) explicit Frobenius norm-constraint. We demonstrate that applying these simple yet effective regularization techniques at key points in the editing process can substantially mitigate model degradation. Combining these regularization methods enables scaling locate-then-edit methods to 10,000 edits while reducing editing time by 42-61%. These results show that targeted regularization is essential for lifelong knowledge editing.

View on arXivView PDF

Code Implementations(1)

10Shell, PythonMay 21, 20251 years ago

Similar Papers

Cite this paper

@article{gupta2025lifelong,
  title  = {Lifelong Knowledge Editing requires Better Regularization},
  author = {Akshat Gupta and Phudish Prateepamornkul and Maochuan Lu and Ahmed Alaa and Thomas Hartvigsen and Gopala Anumanchipalli},
  year   = {2025},
  eprint = {2502.01636},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2502.01636v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2502.01636.svg)](https://codeofpaper.com/paper/2502.01636)

Discussion