Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2503.00177 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866912253180116992 |
|---|---|
| author | Bayat, Reza Rahimi-Kalahroudi, Ali Pezeshki, Mohammad Chandar, Sarath Vincent, Pascal |
| author_facet | Bayat, Reza Rahimi-Kalahroudi, Ali Pezeshki, Mohammad Chandar, Sarath Vincent, Pascal |
| contents | A key challenge in AI alignment is guiding large language models (LLMs) to follow desired behaviors at test time. Activation steering, which modifies internal model activations during inference, offers a potential solution. However, prior work in dense activation spaces struggles with superposition, wherein multiple features become entangled, limiting interpretability and precise control. In contrast, sparse representations provide an untapped opportunity for more interpretable behavior modulation. In this work, we introduce sparse activation steering (SAS), a method that leverages sparse autoencoders (SAEs) to steer LLM behavior in sparse spaces. By isolating behavior-specific features through a contrastive prompt-pairing approach, we define a set of features that can selectively reinforce or suppress behaviors. Experiments on Gemma 2 LLMs show that SAS vectors enable nuanced behavioral modulation and finer-grained control. Furthermore, scaling SAEs improves monosemanticity of SAS vectors, suggesting more reliable and interpretable interventions. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2503_00177 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Steering Large Language Model Activations in Sparse Spaces Bayat, Reza Rahimi-Kalahroudi, Ali Pezeshki, Mohammad Chandar, Sarath Vincent, Pascal Machine Learning Artificial Intelligence A key challenge in AI alignment is guiding large language models (LLMs) to follow desired behaviors at test time. Activation steering, which modifies internal model activations during inference, offers a potential solution. However, prior work in dense activation spaces struggles with superposition, wherein multiple features become entangled, limiting interpretability and precise control. In contrast, sparse representations provide an untapped opportunity for more interpretable behavior modulation. In this work, we introduce sparse activation steering (SAS), a method that leverages sparse autoencoders (SAEs) to steer LLM behavior in sparse spaces. By isolating behavior-specific features through a contrastive prompt-pairing approach, we define a set of features that can selectively reinforce or suppress behaviors. Experiments on Gemma 2 LLMs show that SAS vectors enable nuanced behavioral modulation and finer-grained control. Furthermore, scaling SAEs improves monosemanticity of SAS vectors, suggesting more reliable and interpretable interventions. |
| title | Steering Large Language Model Activations in Sparse Spaces |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2503.00177 |