Adaptive Multi-Subspace Representation Steering for Attribute Alignment in Large Language Models
About
Activation steering offers a promising approach to controlling the behavior of Large Language Models by directly manipulating their internal activations. However, most existing methods struggle to jointly steer multiple attributes, often resulting in interference and undesirable trade-offs. To address this challenge, we propose Multi-Subspace Representation Steering (MSRS), a novel framework for effective multi-attribute steering via subspace representation fine-tuning. MSRS reduces inter-attribute interference by allocating orthogonal subspaces to each attribute, isolating their influence within the model's representation space. MSRS also incorporates a hybrid subspace composition strategy: it combines attribute-specific subspaces for unique steering directions with a shared subspace for common steering directions. A dynamic weighting function learns to efficiently integrate these components for precise control. During inference, MSRS introduces a token-level steering mechanism that dynamically identifies and intervenes on the most semantically relevant tokens, enabling fine-grained behavioral modulation. Experimental results show that MSRS significantly reduces attribute conflicts, surpasses existing methods across a range of attributes, and generalizes effectively to diverse downstream tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Commonsense Reasoning | HellaSwag | HellaSwag Accuracy87.4 | 711 | |
| Multi-task Language Understanding | MMLU | MMLU Accuracy70.2 | 442 | |
| Instruction Following | Alpaca | -- | 173 | |
| Bias Evaluation | BBQ | Accuracy64.5 | 171 | |
| General Language Understanding | GLUE | Accuracy83.2 | 75 | |
| Reading Comprehension | RACE | Accuracy68.3 | 59 | |
| Attribute Steering | HelpSteer | Helpfulness3.89 | 22 | |
| Refusal Behavior | Refusal | Sorry Rate69.3 | 22 | |
| Question Answering | TruthfulQA | MC1 Score34.91 | 22 | |
| Natural Language Understanding | GLUE | SST-2 Accuracy97.99 | 12 |