Skip to content

Gradients for Interventions and Activations for Detection: Targeted Feature Learning in Language Models

Sep 2026 · 0 citations
Computer Science

TL;DR

Overall, the results show that targeted feature quality depends jointly on the model signal and estimator, with detection and intervention capturing complementary properties.

Abstract

Model-internal features can be studied through both their ability to identify a specified concept and their causal effect when manipulated, e.g., through steering or weight editing. A prominent approach to feature learning is Sparse Autoencoders (SAEs), which learn broad feature dictionaries whose relation to particular concepts is typically identified post hoc. However, many interpretability questions are instead hypothesis-driven and concern a concept specified in advance. We study this setting as targeted feature learning, where a single feature is constructed for such a predefined concept. We present a controlled comparison across three model signals (activation values, activation gradients, and parameter gradients) and two estimators (contrastive mean and a learned one-dimensional encoder-decoder), yielding six targeted methods, with CAA and GRADIEND as existing instances and four new methods covering the remaining combinations. We compare these methods against pretrained SAEs across 15 tasks and three language models, evaluating both detection and causal intervention. Across models, the strongest detection performance is achieved by contrastive activation value methods, whereas the strongest intervention performance is achieved by gradient-based methods. Overall, our results show that targeted feature quality depends jointly on the model signal and estimator, with detection and intervention capturing complementary properties.

View source

Similar papers

#machine learning Preprint Oct 2026

What Do Sparse Autoencoder Feature Discovery Scores Identify? From Activations to Interventions

Sparse autoencoder (SAE) features are typically discovered from how their activations co-vary with behaviour, yet they are often used through interventions such as steering or ablation. We take intervention attribution, the first-order effect of a specified intervention on the model output, as our estimand. For a broad...

Yi-Feng Guo, Yan-Ran Li, Xiang Chen · 0 citations
#natural language process... Preprint Sep 2026

MetaSteer: Context-Conditioned, nonlinear Steering via Attention-Projection Adaptation

Steering large language models typically relies on linear, context-independent interventions in activation space, an assumption that recent work has challenged and that can induce an information bottleneck when a fixed representation must encode many behavioral distinctions. We introduce MetaSteer, a method that learns...

Mehdi Jafari, Hao Xue, F. Salim · 0 citations
#machine learning Preprint Sep 2026

Where Decoder Cosine Similarity Fails for SAE Feature Flow Discovery

A transition atlas of triples is constructed, where a residual-state feature and an MLP-update feature jointly predict a target residual feature, and validate candidate triples by ablating the decoded update feature.

Hendrik Droste, C. M. Adriano, Kathrin Korte et al. · 0 citations
#artificial intelligence Preprint Sep 2026

CORTEX: Learning to Share and Specialize in Dense Language Models

Large language models are trained on heterogeneous data mixtures, where different knowledge domains require both shared knowledge and specialization. Existing modular approaches typically impose explicit components or discover modules through interpretability analysis after training. In this work, we propose CORTEX, a...

Chui-Yang Meng, Ming Tang, Vincent W. S. Wong · 0 citations
#machine learning Preprint Sep 2026

SharedSAE: One Feature Dictionary Across Language Models

This work shows that a single shared SAE can replace a collection of dedicated per-model SAEs, and combines a shared dictionary with model-specific encoder-decoder pairs to achieve near-dedicated-SAE reconstruction quality.

Daniil A. Ognev, Célian Vasson, Li-Jie Hu et al. · 0 citations

Related blog posts

MIT News · Artificial Intelligence Aug 27, 2026

Looking beyond natural sequences

A new machine-learning framework aims to improve the success rate of computational protein design while moving away from results that reproduce sequences found in nature.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.