Probe-Geometry Alignment: Erasing the Cross-Sequence Memorization Signature Below Chance
On the map: LLM safety and security → Machine unlearning
Abstract (arXiv)
Recent attacks show that behavioural unlearning of large language models leaves internal traces recoverable by adversarial probes. We characterise where this retention lives and show it can be surgically removed without measurable capability cost. Our central protocol is a leave-one-out cross-sequence probe that tests whether a memorisation signature generalises across held-out sequences. The signature is real and consistent across scale: memorisation-specific gaps of +0.32, +0.19, +0.30 on Pythia-70M, GPT-2 medium, and Mistral-7B; on Pythia-70M, the random-initialisation control collapses to -0.04 at the deepest layer where the pretrained signature peaks. The probe direction is causally separable from recall -- projecting it out collapses the signature locally (+0.44 -> -0.19) while behavioural recall barely changes -- and a probe trained on naturally memorised content does not classify fine-tuning-injected secrets, marking two representationally distinct regimes. We then introduce probe-geometry alignment (PGA), a surgical erasure that aligns activations along the probe's live readout direction at each depth. PGA drives the cross-sequence probe below random chance at all four scales tested (toy depth-4: 0.17; Pythia-70M: 0.07; Mistral-7B: 0.45; GPT-2 medium: 0.06 via MD-PGA k=2) and remains robust to six adversarial probe variants. Against a re-fitting attacker who trains a fresh probe on PGA-treated activations, we extend PGA adversarially, defeating the re-fit probe at every memorisation-relevant depth while preserving five zero-shot capability benchmarks within 2.8 percentage points per task (mean Deltaacc = +0.2pp). The cross-sequence signature is a real, causally separable, regime-specific property of pretrained representations -- removable below chance with a single rank-one intervention per depth at no measurable capability cost.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 55 of its 63 references that are in the map. First posted on arXiv 3 May 2026.
Builds on
55 of 63 references.
Closest ancestors · 17
Same direction, newest first
- 2025The Erasure Illusion: Stress-Testing the Generalization of LLM Forgetting Evaluation22 Dec
- 2025Unlearning Isn't Invisible: Detecting Unlearning Traces in LLMs from Model Outputs16 Jun
- 2025SoK: Machine Unlearning for Large Language Models10 Jun
- 2025Invariance Makes LLM Unlearning Resilient Even to Unanticipated Downstream Fine-Tuning2 Jun
- 2025Does Machine Unlearning Truly Remove Knowledge?29 May
- 2025Unlearning Isn't Deletion: Investigating Reversibility of Machine Unlearning in LLMs22 May
- 2024Catastrophic Failure of LLM Unlearning via Quantization21 Oct
- 2024Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning9 Oct
- 2024MUSE: Machine Unlearning Six-Way Evaluation for Language Models8 Jul
- 2024Unlearning or Obfuscating? Jogging the Memory of Unlearned LLMs via Benign Relearning19 Jun
- 2024Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearning8 Apr
- 2024Eight Methods to Evaluate Robust Unlearning in LLMs26 Feb
5 more
- 2024TOFU: A Task of Fictitious Unlearning for LLMs11 Jan
- 2023Large Language Model Unlearning14 Oct
- 2023Who's Harry Potter? Approximate Unlearning in LLMs3 Oct
- 2023Right to be Forgotten in the Era of Large Language Models: Implications, Challenges, and Solutions8 Jul
- 2022Can Bad Teaching Induce Forgetting? Unlearning in Deep Networks using an Incompetent Teacher17 May
Other references · 38
- 2017Attention Is All You Needcited by 59,932
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
- 2023Mistral 7Bcited by 3,065
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2020Extracting Training Data from Large Language Modelscited by 2,144
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
26 more
- 2014Unsupervised Domain Adaptation by Backpropagationcited by 1,906
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2014FitNets: Hints for Thin Deep Netscited by 1,619
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2019Machine Unlearningcited by 1,020
- 2022Mass-Editing Memory in a Transformercited by 976
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 951
- 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
- 2023Steering Language Models With Activation Engineeringcited by 824
- 2022Quantifying Memorization Across Neural Language Modelscited by 755
- 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 733
- 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 711
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 673
- 2019Designing and Interpreting Probes with Control Taskscited by 608
- 2019Eternal Sunshine of the Spotless Net: Selective Forgetting in Deep Networkscited by 599
- 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 488
- 2020Null It Out: Guarding Protected Attributes by Iterative Nullspace Projectioncited by 442
- 2023Scalable Extraction of Training Data from (Production) Language Modelscited by 415
- 2020Closing the AI Accountability Gap: Defining an End-to-End Framework for Internal Algorithmic Auditingcited by 380
- 2023Detecting Pretraining Data from Large Language Modelscited by 375
- 2023LEACE: Perfect linear concept erasure in closed formcited by 254
- 2020CLUB: A Contrastive Log-ratio Upper Bound of Mutual Informationcited by 233
- 2022Memorization Without Overfitting: Analyzing the Training Dynamics of Large Language Modelscited by 218
- 2024Lessons from the Trenches on Reproducible Evaluation of Language Modelscited by 172
- 2022Linear Adversarial Concept Erasurecited by 98
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Rethinking LLM Memorization through the Lens of Adversarial Compression
- 2026Why Some Models Resist Unlearning: A Linear Stability Perspective
- 2026Learning the Signature of Memorization in Autoregressive Language Models
- 2026Control-Plane Placement Shapes Forgetting: An Architectural Study of Agent Memory Across Thirteen System Configurations
- 2026Linear Probes Detect Task Format, Not Reasoning Mode in Language Model Hidden States
- 2025Analyzing Memorization in Large Language Models through the Lens of Model Attribution
- 2026Beyond Fixed Directions: Adaptive Representation Analysis of Reasoning and Memorization in LLMs
- 2024A Geometric Framework for Understanding Memorization in Generative Models