English

GSS: Gated Subspace Steering for Selective Memorization Mitigation in LLMs

Machine Learning 2026-02-10 v1

Abstract

Large language models (LLMs) can memorize and reproduce training sequences verbatim -- a tendency that undermines both generalization and privacy. Existing mitigation methods apply interventions uniformly, degrading performance on the majority of tokens that generalize normally. We show empirically that memorization is sparse, intermittent, and token-conditioned, suggesting that effective mitigation requires context-aware intervention rather than static parameter modification. To this end, we propose a novel and effective selective memorization mitigation method -- Gated Subspace Steering (GSS), which decomposes intervention into a probe (detecting memorization-relevant activations) and a steer (applying targeted correction only when the probe exceeds a threshold). The optimal probe-steer pair emerges from a principled optimization framework based on optimal subspace steering. Experiments on four benchmarks show GSS matches or exceeds state-of-the-art memorization reduction while requiring 1001000×100-1000 \times less compute than optimization-based alternatives. Furthermore, we provide new theoretical insights into the geometry of memorization in neural representations.

Keywords

Cite

@article{arxiv.2602.08901,
  title  = {GSS: Gated Subspace Steering for Selective Memorization Mitigation in LLMs},
  author = {Xuanqi Zhang and Haoyang Shang and Xiaoxiao Li},
  journal= {arXiv preprint arXiv:2602.08901},
  year   = {2026}
}

Comments

34 pages, 12 figures

R2 v1 2026-07-01T10:28:18.913Z