Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Mitigating Memorization in LLMs using Activation Steering

About

The memorization of training data by Large Language Models (LLMs) poses significant risks, including privacy leaks and the regurgitation of copyrighted content. Activation steering, a technique that directly intervenes in model activations, has emerged as a promising approach for manipulating LLMs. In this work, we explore the effectiveness of activation steering in reducing memorization while preserving generalization capabilities. We conduct empirical evaluations using a controlled memorization benchmark of literary material and demonstrate that our method successfully suppresses memorized content with minimal degradation in model performance in Gemma. Additionally, we analyze the trade-offs between suppression effectiveness and linguistic fluency, highlighting the advantages and limitations of activation-based interventions. Our findings contribute to ongoing efforts in developing safer and more privacy-preserving LLMs by providing a practical and efficient mechanism to mitigate unintended memorization.

Manan Suri, Nishit Anand, Amisha Bhaskar• 2025

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningHellaSwag
HellaSwag Accuracy72.5
897
Model EditingOLMo-7B memorized sequences
Success Rate (%)73.3
11
Language ModelingWikiText-103
PPLmed16.42
11
Question AnsweringWikidata QA
Accuracy58
11
Showing 4 of 4 rows

Other info

Follow for update