Reward-Informed Sparse Autoencoders and the Solution-Completeness Confound
Not on the map. Why?
Abstract (arXiv)
Sparse autoencoders (SAEs) decompose language-model activations into sparse, interpretable features, and an appealing way to aim them at reasoning is to curate their data with a signal reinforcement learning already produces: the reward. We build such a reward-informed SAE (RI-SAE): we split GRPO trajectories into high-reward ("good") and low-reward ("bad") reasoning continuations, train a standard JumpReLU SAE on their activations, and then ask what the resulting good/bad separation actually measures. On Llama-3.1-8B a sparse subset of the 16,384 features does separate the classes (silhouette 0.79 on the selected features versus 0.005 for the full code), but a control battery shows the separation is largely solution completeness rather than reasoning quality: a TF-IDF text classifier already splits the classes (AUC 0.75--0.83), and three structural cues alone (length, a closed reasoning block, and a boxed answer) reach AUC 0.70 (99% of good versus 69% of bad completions are boxed). A generic SAE that never saw the reward does not separate the classes at all (silhouette 0.01, no discriminative features), so the 0.79 is in-sample fitting of this curated signal rather than structure that a reward-blind dictionary recovers. We therefore present the recipe and its control battery together: reward filtering is a cheap, label-free way to reuse RL signals for interpretability, but most of what it surfaces is completion form. Two discriminative features are still readable (symbolic mathematics; procedural and evaluative language), which we take as illustrative rather than as isolated reasoning.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 7 of its 8 references that are in the map. First posted on arXiv 25 Jun 2026.
Builds on
7 of 8 references.
References · 7
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 381
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
- 2024Jumping Ahead: Improving Reconstruction Fidelity with JumpReLU Sparse Autoencoderscited by 283
- 2024Llama Scope: Extracting Millions of Features from Llama-3.1-8B with Sparse Autoencoderscited by 133
- 2025Archetypal SAE: Adaptive and Stable Dictionary Learning for Concept Extraction in Large Vision Modelscited by 67
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026SAEExplainer: Interpreting SAE Features with Activation-Guided Preference Optimization
- 2025A Comparative Analysis of Sparse Autoencoder and Activation Difference in Language Model Steering
- 2025Interpretable Reward Model via Sparse Autoencoder
- 2026Do Sparse Autoencoders Identify Reasoning Features in Language Models?
- 2025I Have Covered All the Bases Here: Interpreting Reasoning Features in Large Language Models via Sparse Autoencoders
- 2026From Input to Output: A Flexible Agent for Dual-End Interpretation of Sparse Autoencoder Features
- 2025Evaluating Sparse Autoencoders for Monosemantic Representation
- 2025Beyond Redundancy: Diverse and Specialized Multi-Expert Sparse Autoencoder