Steering at the Source: Style Modulation Heads for Robust Persona Control

2026-02-24

AI Summary

This work addresses the degradation in output coherence and safety commonly induced by existing activation intervention methods that modulate large language model personas through perturbations of residual streams. To overcome this limitation, the authors propose a novel geometric analysis–based approach that leverages inter-layer cosine similarity and attention head contribution scores to identify—and for the first time explicitly name—three sparse, independent “style-modulating heads” responsible for disentangled control of persona and stylistic attributes. By precisely intervening on these specific attention heads, the method achieves effective persona modulation while substantially mitigating the coherence deterioration typical of conventional techniques, thereby significantly enhancing the robustness and safety of behavioral interventions in large language models.

Abstract

Activation steering offers a computationally efficient mechanism for controlling Large Language Models (LLMs) without fine-tuning. While effectively controlling target traits (e.g., persona), coherency degradation remains a major obstacle to safety and practical deployment. We hypothesize that this degradation stems from intervening on the residual stream, which indiscriminately affects aggregated features and inadvertently amplifies off-target noise. In this work, we identify a sparse subset of attention heads (only three heads) that independently govern persona and style formation, which we term Style Modulation Heads. Specifically, these heads can be localized via geometric analysis of internal representations, combining layer-wise cosine similarity and head-wise contribution scores. We demonstrate that intervention targeting only these specific heads achieves robust behavioral control while significantly mitigating the coherency degradation observed in residual stream steering. More broadly, our findings show that precise, component-level localization enables safer and more precise model control.

Problem

• activation steering
• coherency degradation
• persona control
• Large Language Models
• model safety

Innovation

• Style Modulation Heads
• activation steering
• persona control
• attention heads
• coherency preservation