Discovering Universal Activation Directions for PII Leakage in Language Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Marchyok, Leo, Coalson, Zachary, Keum, Sungho, Son, Sooel, Hong, Sanghyun
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911456839073792
author Marchyok, Leo
Coalson, Zachary
Keum, Sungho
Son, Sooel
Hong, Sanghyun
author_facet Marchyok, Leo
Coalson, Zachary
Keum, Sungho
Son, Sooel
Hong, Sanghyun
contents Modern language models exhibit rich internal structure, yet little is known about how privacy-sensitive behaviors, such as personally identifiable information (PII) leakage, are represented and modulated within their hidden states. We present UniLeak, a mechanistic-interpretability framework that identifies universal activation directions: latent directions in a model's residual stream whose linear addition at inference time consistently increases the likelihood of generating PII across prompts. These model-specific directions generalize across contexts and amplify PII generation probability, with minimal impact on generation quality. UniLeak recovers such directions without access to training data or groundtruth PII, relying only on self-generated text. Across multiple models and datasets, steering along these universal directions substantially increases PII leakage compared to existing prompt-based extraction methods. Our results offer a new perspective on PII leakage: the superposition of a latent signal in the model's representations, enabling both risk amplification and mitigation.
format Preprint
id arxiv_https___arxiv_org_abs_2602_16980
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Discovering Universal Activation Directions for PII Leakage in Language Models
Marchyok, Leo
Coalson, Zachary
Keum, Sungho
Son, Sooel
Hong, Sanghyun
Machine Learning
Cryptography and Security
Modern language models exhibit rich internal structure, yet little is known about how privacy-sensitive behaviors, such as personally identifiable information (PII) leakage, are represented and modulated within their hidden states. We present UniLeak, a mechanistic-interpretability framework that identifies universal activation directions: latent directions in a model's residual stream whose linear addition at inference time consistently increases the likelihood of generating PII across prompts. These model-specific directions generalize across contexts and amplify PII generation probability, with minimal impact on generation quality. UniLeak recovers such directions without access to training data or groundtruth PII, relying only on self-generated text. Across multiple models and datasets, steering along these universal directions substantially increases PII leakage compared to existing prompt-based extraction methods. Our results offer a new perspective on PII leakage: the superposition of a latent signal in the model's representations, enabling both risk amplification and mitigation.
title Discovering Universal Activation Directions for PII Leakage in Language Models
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2602.16980