Energy-Regularized Sequential Model Editing on Hyperspheres
Qingyuan Liu, Jia-Chen Gu, Yunzhi Yao, Hong Wang, Nanyun Peng
TL;DR
This paper addresses the instability and knowledge degradation observed during sequential updates to large language models. It introduces hyperspherical uniformity and Hyperspherical Energy (HE) as diagnostic tools and demonstrates a strong link between HE dynamics and editing stability, supported by a theoretical bound that connects HE changes to knowledge degradation. The authors propose SPHERE, an HE-driven regularization that projects edits into a sparse subspace orthogonal to the principal directions of pretrained weights, thereby preserving prior knowledge while enabling reliable sequential edits. Empirical results on LLaMA3-8B and Qwen2.5-7B show SPHERE outperforms state-of-the-art baselines in editing capability and general ability retention, and it also provides a plug-and-play boost to existing methods, offering a principled path toward scalable, reliable knowledge editing.
Abstract
Large language models (LLMs) require constant updates to remain aligned with evolving real-world knowledge. Model editing offers a lightweight alternative to retraining, but sequential editing often destabilizes representations and induces catastrophic forgetting. In this work, we seek to better understand and mitigate performance degradation caused by sequential editing. We hypothesize that hyperspherical uniformity, a property that maintains uniform distribution of neuron weights on a hypersphere, helps the model remain stable, retain prior knowledge, while still accommodate new updates. We use Hyperspherical Energy (HE) to quantify neuron uniformity during editing, and examine its correlation with editing performance. Empirical studies across widely used editing methods reveals a strong correlation between HE dynamics and editing performance, with editing failures consistently coinciding with high HE fluctuations. We further theoretically prove that HE dynamics impose a lower bound on the degradation of pretrained knowledge, highlighting why HE stability is crucial for knowledge retention. Motivated by these insights, we propose SPHERE (Sparse Projection for Hyperspherical Energy-Regularized Editing), an HE-driven regularization strategy that stabilizes neuron weight distributions, ultimately preserving prior knowledge while enabling reliable sequential updates. Specifically, SPHERE identifies a sparse space complementary to the principal hyperspherical directions of the pretrained weight matrices and projects new knowledge onto it, attenuating perturbations on the principal directions. Extensive experiments on LLaMA3 (8B) and Qwen2.5 (7B) show that SPHERE outperforms the best baseline in editing capability by an average of 16.41%, while most faithfully preserving general model performance, thereby offering a principled path toward reliable large-scale knowledge editing.
