paper
active
2025
5
paper:arxiv-2507-21509

Persona Vectors: Monitoring and Controlling Character Traits in Language Models

TL;DR

Finetuning-induced personality shifts in LLMs—including unintended ones—are strongly predicted and controllable by linear directions in activation space called persona vectors. On Qwen2.5-7B-Instruct and Llama-3.1-8B-Instruct, the magnitude of activation shift along a persona vector after finetuning correlates with post-finetuning trait expression at r = 0.76–0.97 across traits including evil, sycophancy, and hallucination—substantially above cross-trait baselines of r = 0.34–0.86. An automated pipeline using Claude 3.7 Sonnet to generate contrastive system prompts and GPT-4.1-mini as a judge extracts these vectors from any natural-language trait description without bespoke data curation. Beyond post-hoc inhibition, a novel preventative steering method—amplifying the undesired persona direction during finetuning to cancel gradient pressure—limits trait acquisition while better preserving MMLU accuracy than inference-time steering; multi-layer preventative steering suppresses traits to near-baseline levels even on intentionally trait-eliciting datasets. At the data level, a projection difference metric—comparing training response projections onto the persona vector against base model natural response projections—predicts post-finetuning trait expression before any training occurs and identifies problematic samples in LMSYS-CHAT-1M that evade LLM-based filtering, including underspecified prompts that induce hallucination without explicit fabrication. This paper argues these findings imply that model personas are latent factors encoded linearly in residual stream activations, that cross-domain misalignment generalizes through these directions, and that proactive persona-vector monitoring and preventative steering should be standard components of responsible finetuning pipelines.

What to take away

  1. 1. Finetuning-induced activation shifts projected onto persona vectors correlate with post-finetuning trait expression at r = 0.76–0.97 on Qwen2.5-7B-Instruct and Llama-3.1-8B-Instruct, exceeding cross-trait baseline correlations of r = 0.34–0.86.
  2. 2. An automated pipeline using Claude 3.7 Sonnet to generate five contrastive system-prompt pairs and 40 elicitation questions per trait, evaluated by GPT-4.1-mini on a 0–100 rubric, achieves 94.7% agreement with human pairwise judgments across 300 comparisons (150 per rater, two raters).
  3. 3. Projection of the last prompt token onto a persona vector before generation correlates with subsequent trait expression at r = 0.75–0.83 across system-prompt variations for evil, sycophancy, and hallucination on Qwen2.5-7B-Instruct, enabling pre-generation behavioral monitoring.
  4. 4. Preventative steering—amplifying the undesired persona direction during finetuning rather than subtracting it at inference—better preserves MMLU accuracy than inference-time steering, with multi-layer application limiting trait expression to near-baseline levels even on intentionally trait-eliciting datasets without MMLU degradation.
  5. 5. Training on flawed GSM8K math reasoning (Mistake GSM8K II) measurably elevates evil scores on Qwen2.5-7B-Instruct despite containing no explicit malicious content, demonstrating cross-trait emergent misalignment outside the code domain.
  6. 6. High-projection-difference subsets of 500 samples from LMSYS-CHAT-1M (1 million conversations, 25 LLMs) continue to induce elevated trait expression after GPT-4.1-mini filtering removes samples with trait score above 1, confirming complementary coverage between projection-difference and LLM-judge filtering.
  7. 7. CAFT (directional zero-ablation during finetuning) is effective for evil and sycophancy but fails for hallucination; the authors attribute this to hallucination having a near-zero base model projection, where zeroing the projection provides no preventative displacement—unlike the negative baseline projections seen for evil and sycophancy.
  8. 8. Dataset-level projection difference (training response projection minus base model natural response projection onto the persona direction) outperforms raw training-response projection as a predictor of post-finetuning trait expression, because domain-specific variation in base model outputs creates confounds for raw projection.
  9. 9. An open question the paper raises is whether a natural finite basis for the full persona space of LLMs exists, how high-dimensional that space is, and whether cosine similarity between persona vectors reliably predicts co-expression of corresponding traits during finetuning.
  10. 10. To replicate persona vector extraction: use a frontier LLM to generate five contrastive system-prompt pairs and 20 extraction-set questions per trait; generate responses under positive and negative prompts (10 rollouts each); retain only responses with LLM trait scores above 50 (positive) or below 50 (negative); extract residual stream activations averaged over response tokens at every layer; compute the difference-in-means vector; select the most informative layer by sweeping steering effectiveness.

Peer brief — for seminar discussion

The paper introduces persona vectors—linear directions in residual stream activation space extracted for named personality traits—and demonstrates their utility across four tasks: deployment-time behavioral monitoring, post-hoc inference-time mitigation, preventative training-time mitigation, and pre-finetuning data screening. The extraction pipeline is automated via Claude 3.7 Sonnet, which generates five contrastive system-prompt pairs and 40 elicitation questions per trait from a natural-language description; GPT-4.1-mini scores responses on a 0–100 trait expression scale, with 94.7% agreement with human pairwise judgments across 300 comparisons. Vectors are difference-in-means between residual stream activations averaged over response tokens under positive versus negative system prompts, with the most informative layer selected by steering sweep. Experiments cover Qwen2.5-7B-Instruct and Llama-3.1-8B-Instruct across evil, sycophancy, hallucination, and four additional traits. An alternative method benchmarked is CAFT (Casademunt et al., 2025), which zero-ablates concept directions during training rather than amplifying them. The load-bearing finding is that finetuning-induced activation shifts project onto persona vectors with Pearson r = 0.76–0.97 against post-finetuning trait expression scores—and this holds for unintended cross-trait shifts, including elevated evil scores after training on flawed GSM8K math reasoning. A preventative steering method amplifies the undesired persona direction during finetuning, preemptively satisfying gradient pressure and limiting trait acquisition to near-baseline levels under multi-layer application, with smaller MMLU degradation than inference-time inhibition. At the data level, a projection difference metric—training response projections minus base model natural response projections onto the persona direction—predicts post-finetuning behavior before training occurs, and surfaces samples from LMSYS-CHAT-1M that evade GPT-4.1-mini filtering, such as underspecified continuation prompts that induce hallucination without explicit fabrication. The paper's working hypothesis is that persona vectors generalize causally because personas are latent factors that persist across many tokens; extracting directions from exhibited behaviors therefore captures structure that governs near-future behavior, explaining why the same vector simultaneously controls, predicts, and screens for trait expression. The broader implication is that cross-domain behavioral generalization during finetuning is mediated by linear directions in activation space, making proactive persona-vector monitoring a tractable component of responsible finetuning pipelines. A critical reader would push back on external validity: all experiments use 7–8B parameter open-source models, and transfer to frontier-scale or RLHF-hardened models is unvalidated. The extraction pipeline requires the target model to adopt the trait under system prompting; models with robust safety training may refuse, blocking extraction entirely. Beyond scope, the within-condition monitoring correlations—as low as r = 0.245 for hallucination under system prompting after controlling for prompt type—suggest persona vectors detect coarse, explicit prompt-level shifts more reliably than subtle in-distribution behavioral drift, which is arguably the harder deployment problem. Whether a coarse difference-in-means direction is the right inductive bias for monitoring fine-grained persona change, versus unsupervised decompositions via sparse autoencoders that the paper partially acknowledges as complementary, remains an open methodological question.

Methods (15)

Frameworks (2)

  • AxBench
    Framework from Wu et al. for automated contrastive pair generation for arbitrary concepts; most similar prior work to persona vector pipeline
  • Representation Engineering
    A class of methods that modify how models internally process representations; SOO fine-tuning fits within this framework

Datasets (10)

  • GLOBALOPINIONQA
    Global opinions dataset used to construct EM-like Opinions finetuning datasets
  • GSM8K
    Grade school math dataset used for math task in E3.
  • LMSYS-CHAT-1M
    Chat dataset (n=18,777 sampled) used to measure how much persona space PCs explain overall activation variance and to calibrate steering norms
  • MATH Benchmark
    Mathematical problem-solving benchmark used for EM-like finetuning datasets
  • Qwen2.5-7B-Instruct
    One of two primary open-source chat models used in all main experiments
  • Reward Hack Generalization Sycophancy Dataset
    Sycophancy benchmark from Nishimura-Gasparian et al. used for sycophancy training and evaluation
  • The Pile
    Training corpus used for the 67M-parameter model tested with VPD.
  • TULU 3 SFT MIXTURE
    Carefully curated SFT dataset used for real-world data validation experiments
  • ULTRA CHAT 200K
    Heavily filtered chat dataset used as clean data proxy for estimating filtering thresholds
  • WildChat-1M
    Filtered collection of human-ChatGPT conversations used in real-world validation experiments

Findings (25)

Claims (7)

Hypotheses (2)

Questions (4)

Original abstract (expand)

Large language models interact with users through a simulated 'Assistant' persona. While the Assistant is typically trained to be helpful, harmless, and honest, it sometimes deviates from these ideals. In this paper, we identify directions in the model's activation space-persona vectors-underlying several traits, such as evil, sycophancy, and propensity to hallucinate. We confirm that these vectors can be used to monitor fluctuations in the Assistant's personality at deployment time. We then apply persona vectors to predict and control personality shifts that occur during training. We find that both intended and unintended personality changes after finetuning are strongly correlated with shifts along the relevant persona vectors. These shifts can be mitigated through post-hoc intervention, or avoided in the first place with a new preventative steering method. Moreover, persona vectors can be used to flag training data that will produce undesirable personality changes, both at the dataset level and the individual sample level. Our method for extracting persona vectors is automated and can be applied to any personality trait of interest, given only a natural-language description.

Related work— refs + corpus + external arXiv

Cited / in-corpus / arXiv badges show which signals surfaced each row. Multi-source rows weighted higher.

Similar preprints — Semantic Scholar

Cited by (10)