Lifelong Knowledge Editing requires Better Regularization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Gupta, Akshat, Prateepamornkul, Phudish, Lu, Maochuan, Alaa, Ahmed, Hartvigsen, Thomas, Anumanchipalli, Gopala
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909618426347520
author Gupta, Akshat
Prateepamornkul, Phudish
Lu, Maochuan
Alaa, Ahmed
Hartvigsen, Thomas
Anumanchipalli, Gopala
author_facet Gupta, Akshat
Prateepamornkul, Phudish
Lu, Maochuan
Alaa, Ahmed
Hartvigsen, Thomas
Anumanchipalli, Gopala
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2502_01636
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Lifelong Knowledge Editing requires Better Regularization
Gupta, Akshat
Prateepamornkul, Phudish
Lu, Maochuan
Alaa, Ahmed
Hartvigsen, Thomas
Anumanchipalli, Gopala
Computation and Language
Artificial Intelligence
Machine Learning
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.
title Lifelong Knowledge Editing requires Better Regularization
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2502.01636