arXiv:2605.28839v1 Announce Type: new Abstract: Knowledge editing methods such as ROME and MEMIT update factual associations in transformer models by modifying MLP weights. While evaluated mainly by output behavior, their internal mechanism remains underexplored. We investigate whether edits rely on a common mechanism, regardless of which fact is modified. Despite fact-specific weight changes, we argue that ROME and MEMIT target the same subset of weights critical for maintaining edits. To isolate this subset, we train a compact binary mask over the edited weights. The mask reverses 80% of edits on the training set and over 70% on the test set, confirming that diverse edits share a common functional structure. Our analysis reveals that the mask reverses edits by eliminating overattention in later layers. Additionally, we show that injecting the mask during editing drops editing success from 98% to 38%, demonstrating that this mechanism is necessary for edits to succeed. Our finding that edits suppress rather than overwrite knowledge explains why ROME and MEMIT fail to propagate changes to related facts. The identified common functional subspace informs detection and defense against unwanted edits.
One Mask to Rule Them All: On Hidden Facts after Editing and How to Find Them
Researchers at arXiv have discovered that knowledge editing methods ROME and MEMIT, which modify factual associations in transformer models, rely on a shared functional mechanism rather than fact-specific weight changes. By training a binary mask over edited weights, the team reversed over 70% of edits on test sets and found that edits suppress rather than overwrite knowledge, explaining why these methods fail to propagate changes to related facts. The findings reveal a common functional subspace that could inform detection and defense against unwanted edits in AI systems.
Run your AI side-project on zahid.host
EU-based hosting, git-push deploys, automatic HTTPS, no cold starts. Free tier with a custom domain — perfect for shipping the agent you just read about.