Output Vector Editing for Memorization Mitigation in Large Language Models
Not on the map. Why?
Abstract (arXiv)
Large language models memorize and reproduce sequences from their training data, creating privacy, copyright, and security risks. Existing neuron-level mitigation methods equate editing with zeroing out neuron activations, but the activation only controls whether a neuron engages; the output vector is what writes to the residual stream and, through superposition, encodes multiple features. We propose output vector editing, a constrained-optimization weight edit that locates a small set of MLP neurons responsible for a memorized continuation and minimally modifies their output vectors to introduce a distractor in vocabulary space, redirecting their residual-stream contributions while leaving activations unchanged. Evaluating on four models from 360M to 7B parameters (SmolLM-360M, OLMo-1B, OLMo-7B, Llama2-7B), we center on OLMo-7B (whose open weights and pretraining corpus enable systematic mining) and mine 6831 memorized sequences, achieving up to 87.9% suppression. The 2.7× gap over zero ablation on the same located neurons shows the suppression comes from the output-vector edit, not localization alone. Four edit modes span a spectrum from aggressive suppression to minimal redirection; in ensemble they cover 96.5% of memorized sequences, while our recommended single-mode configuration reaches 81.5% with no catastrophic locality failures. We further identify a mechanistic boundary at ~14% of sequences unreachable by MLP-only editing; while these failures are not attention-driven overall, ablating the top contributing attention heads recovers 60--64% of them, with stronger recovery on continuations that copy tokens from the prefix, positioning attention as a complementary fallback rather than a primary mechanism. Edit mode ordering and the success-locality trade-off transfer across all four models, with success rates scaling with model size rather than family.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 39 of its 44 references that are in the map. First posted on arXiv 17 Jun 2026.
Builds on
39 of 44 references.
References · 39
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2020Extracting Training Data from Large Language Modelscited by 2,189
- 2022Beyond the Imitation Game: Quantifying and extrapolating the capabilities of language modelscited by 1,998
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2022Mass-Editing Memory in a Transformercited by 1,001
- 2022Quantifying Memorization Across Neural Language Modelscited by 763
- 2021Deduplicating Training Data Makes Language Models Bettercited by 714
- 2024OLMo: Accelerating the Science of Language Modelscited by 613
- 2021Knowledge Neurons in Pretrained Transformerscited by 601
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 532
27 more
- 2024Dolma: an Open Corpus of Three Trillion Tokens for Language Model Pretraining Researchcited by 483
- 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 334
- 2022Deduplicating Training Data Mitigates Privacy Risks in Language Modelscited by 314
- 2023Large Language Model Unlearningcited by 302
- 2022Preventing Verbatim Memorization in Language Models Gives a False Sense of Privacycited by 100
- 2024Information Flow Routes: Automatically Interpreting Language Models at Scalecited by 95
- 2024Model Editing at Scale leads to Gradual and Catastrophic Forgettingcited by 93
- 2024Unlearning or Obfuscating? Jogging the Memory of Unlearned LLMs via Benign Relearningcited by 84
- 2023Can Neural Network Memorization Be Localized?cited by 69
- 2022Understanding Transformer Memorization Recall Through Idiomscited by 62
- 2024Demystifying Verbatim Memorization in Large Language Modelscited by 56
- 2024The Butterfly Effect of Model Editing: Few Edits Can Trigger Large Language Models Collapsecited by 54
- 2024Be like a Goldfish, Don't Memorize! Mitigating Memorization in Generative LLMscited by 50
- 2023Adversarial Attacks and Defenses in Large Language Models: Old and New Threatscited by 49
- 2023SoK: Memorization in General-Purpose Large Language Modelscited by 37
- 2024Copyright Traps for Large Language Modelscited by 37
- 2024Localizing Paragraph Memorization in Language Modelscited by 31
- 2023Do Localization Methods Actually Localize Memorized Data in LLMs? A Tale of Two Benchmarkscited by 29
- 2025Modality-Aware Neuron Pruning for Unlearning in Multimodal Large Language Modelscited by 26
- 2024Mitigating Memorization In Language Modelscited by 12
- 2025Lifelong Knowledge Editing requires Better Regularizationcited by 10
- 2025Mitigating Memorization in LLMs using Activation Steeringcited by 8
- 2025GeoEdit: Geometric Knowledge Editing for Large Language Modelscited by 6
- 2025ParaPO: Aligning Language Models to Reduce Verbatim Reproduction of Pre-training Datacited by 6
- 2025Understanding Verbatim Memorization in LLMs Through Circuit Discoverycited by 4
- 2025SCOPE: Intrinsic Semantic Space Control for Mitigating Copyright Infringement in LLMscited by 3
- 2025Low-Perplexity LLM-Generated Sequences and Where To Find Themcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024WISE: Rethinking the Knowledge Memory for Lifelong Model Editing of Large Language Models
- 2025Cross-Modal Unlearning via Influential Neuron Path Editing in Multimodal Large Language Models
- 2024Navigating the Dual Facets: A Comprehensive Evaluation of Sequential Memory Editing in Large Language Models
- 2026Beyond the Covariance Trap: Unlocking Generalization in Same-Subject Knowledge Editing for Large Language Models
- 2026Gated DeltaNet-2: Decoupling Erase and Write in Linear Attention
- 2026Not Every Change Is Necessary: Recoverable Drift in Large Language Model Unlearning
- 2026SCAN: Sparse Circuit Anchor Interpretable Neuron for Lifelong Knowledge Editing
- 2026Erase-then-Delta Attention: Decoupling Erase and Write Addresses in Delta-Rule Linear Attention