SAE-SSV: Supervised Steering in Sparse Representation Spaces for Reliable Control of Language Models
By: Zirui He , Mingyu Jin , Bo Shen and more
Potential Business Impact:
Guides AI to follow instructions better.
Large language models (LLMs) have demonstrated impressive capabilities in natural language understanding and generation, but controlling their behavior reliably remains challenging, especially in open-ended generation settings. This paper introduces a novel supervised steering approach that operates in sparse, interpretable representation spaces. We employ sparse autoencoders (SAEs)to obtain sparse latent representations that aim to disentangle semantic attributes from model activations. Then we train linear classifiers to identify a small subspace of task-relevant dimensions in latent representations. Finally, we learn supervised steering vectors constrained to this subspace, optimized to align with target behaviors. Experiments across sentiment, truthfulness, and politics polarity steering tasks with multiple LLMs demonstrate that our supervised steering vectors achieve higher success rates with minimal degradation in generation quality compared to existing methods. Further analysis reveals that a notably small subspace is sufficient for effective steering, enabling more targeted and interpretable interventions.
Similar Papers
Identifiable Steering via Sparse Autoencoding of Multi-Concept Shifts
Machine Learning (CS)
Teaches AI to change its answers without retraining.
Steering Large Language Model Activations in Sparse Spaces
Machine Learning (CS)
Teaches AI to follow instructions better.
Towards LLM Guardrails via Sparse Representation Steering
Cryptography and Security
Makes AI say helpful, safe, and honest things.