mlsanityData to 14 Oct 2026

Preference Heads in Large Language Models: A Mechanistic Framework for Interpretable Personalization

arXiv 2604.22345 ↗First posted 24 Apr 2026cs.CLcited by 2

Not on the map. Why?

Abstract (arXiv)

Large Language Models (LLMs) exhibit strong implicit personalization ability, yet most existing approaches treat this behavior as a black box, relying on prompt engineering or fine tuning on user data. In this work, we adopt a mechanistic interpretability perspective and hypothesize the existence of a sparse set of Preference Heads, attention heads that encode user specific stylistic and topical preferences and exert a causal influence on generation. We introduce Differential Preference Steering (DPS), a training free framework that (1) identifies Preference Heads through causal masking analysis and (2) leverages them for controllable and interpretable personalization at inference time. DPS computes a Preference Contribution Score (PCS) for each attention head, directly measuring its causal impact on user aligned outputs. During decoding, we contrast model predictions with and without Preference Heads, amplifying the difference between personalized and generic logits to selectively strengthen preference aligned continuations. Experiments on widely used personalization benchmarks across multiple LLMs demonstrate consistent gains in personalization fidelity while preserving content coherence and low computational overhead. Beyond empirical improvements, DPS provides a mechanistic explanation of where and how personalization emerges within transformer architectures. Our implementation is publicly available.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 19 of its 27 references that are in the map. First posted on arXiv 24 Apr 2026.

Builds on

19 of 27 references.

References · 19

  1. 2023LaMP: When Large Language Models Meet Personalizationcited by 437
  2. 2023Trusting Your Evidence: Hallucinate Less with Context-aware Decodingcited by 327
  3. 2024Iterative Reasoning Preference Optimizationcited by 246
  4. 2024Retrieval Head Mechanistically Explains Long-Context Factualitycited by 232
  5. 2024LongLaMP: A Benchmark for Personalized Long-form Text Generationcited by 97
  6. 2025A Survey of Personalized Large Language Models: Progress and Future Directionscited by 71
  7. 2024Aligning LLM Agents by Learning Latent Preference from User Editscited by 69
  8. 2025Which Attention Heads Matter for In-Context Learning?cited by 61
  9. 2024PAD: Personalized Alignment of LLMs at Decoding-Timecited by 60
  10. 2024Induction Heads as an Essential Mechanism for Pattern Matching in In-context Learningcited by 46
  11. 2025Measuring What Makes You Unique: Difference-Aware User Modeling for Enhancing LLM Personalizationcited by 45
  12. 2024Attention in Large Language Models Yields Efficient Zero-Shot Re-Rankerscited by 43
7 more
  1. 2025Personalized Text Generation with Contrastive Activation Steeringcited by 34
  2. 2025Query-Focused Retrieval Heads Improve Long-Context Reasoning and Re-rankingcited by 31
  3. 2024DeCoRe: Decoding by Contrasting Retrieval Heads to Mitigate Hallucinationscited by 25
  4. 2025Beyond Prompt Engineering: Robust Behavior Control in LLMs via Steering Target Atomscited by 24
  5. 2026OpenDecoder: Open Large Language Model Decoding to Incorporate Document Quality in RAGcited by 11
  6. 2023SPIN: Sparsifying and Integrating Internal Neurons in Large Language Models for Text Classificationcited by 10
  7. 2026ReAttn: Improving Attention-based Re-ranking via Attention Re-weightingcited by 5

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Text as a Universal Interface for Transferable Personalization
  2. 2026PrefReward: Learning User Preference Matrix for Personalized Text Generation
  3. 2024Unsupervised Human Preference Learning
  4. 2026HyperTrace: Hypothesis-Based Preference Tracing for Online LLM Personalization
  5. 2026VIBE-Bench: Evaluating Personalized Large Language Models When Profiles Don't Mean Preferences
  6. 2026Rethinking Personalization in Large Language Models at the Token Level
  7. 2025SteerX: Disentangled Steering for LLM Personalization
  8. 2026Group Preference Collapse in Personalized Multimodal Large Language Models