SAEs Are Good for Steering -- If You Select the Right Features

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Arad, Dana, Mueller, Aaron, Belinkov, Yonatan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909972902707200
author Arad, Dana
Mueller, Aaron
Belinkov, Yonatan
author_facet Arad, Dana
Mueller, Aaron
Belinkov, Yonatan
contents Sparse Autoencoders (SAEs) have been proposed as an unsupervised approach to learn a decomposition of a model's latent space. This enables useful applications such as steering - influencing the output of a model towards a desired concept - without requiring labeled data. Current methods identify SAE features to steer by analyzing the input tokens that activate them. However, recent work has highlighted that activations alone do not fully describe the effect of a feature on the model's output. In this work, we draw a distinction between two types of features: input features, which mainly capture patterns in the model's input, and output features, which have a human-understandable effect on the model's output. We propose input and output scores to characterize and locate these types of features, and show that high values for both scores rarely co-occur in the same features. These findings have practical implications: after filtering out features with low output scores, we obtain 2-3x improvements when steering with SAEs, making them competitive with supervised methods.
format Preprint
id arxiv_https___arxiv_org_abs_2505_20063
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SAEs Are Good for Steering -- If You Select the Right Features
Arad, Dana
Mueller, Aaron
Belinkov, Yonatan
Machine Learning
Artificial Intelligence
Computation and Language
Sparse Autoencoders (SAEs) have been proposed as an unsupervised approach to learn a decomposition of a model's latent space. This enables useful applications such as steering - influencing the output of a model towards a desired concept - without requiring labeled data. Current methods identify SAE features to steer by analyzing the input tokens that activate them. However, recent work has highlighted that activations alone do not fully describe the effect of a feature on the model's output. In this work, we draw a distinction between two types of features: input features, which mainly capture patterns in the model's input, and output features, which have a human-understandable effect on the model's output. We propose input and output scores to characterize and locate these types of features, and show that high values for both scores rarely co-occur in the same features. These findings have practical implications: after filtering out features with low output scores, we obtain 2-3x improvements when steering with SAEs, making them competitive with supervised methods.
title SAEs Are Good for Steering -- If You Select the Right Features
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2505.20063