mlsanityData to 7 Oct 2026

Reward-Informed Sparse Autoencoders and the Solution-Completeness Confound

arXiv 2608.26136 ↗First posted 25 Jun 2026cs.CLcited by 0

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

  1. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  3. 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 381
  4. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
  5. 2024Jumping Ahead: Improving Reconstruction Fidelity with JumpReLU Sparse Autoencoderscited by 283
  6. 2024Llama Scope: Extracting Millions of Features from Llama-3.1-8B with Sparse Autoencoderscited by 133
  7. 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

  1. 2026SAEExplainer: Interpreting SAE Features with Activation-Guided Preference Optimization
  2. 2025A Comparative Analysis of Sparse Autoencoder and Activation Difference in Language Model Steering
  3. 2025Interpretable Reward Model via Sparse Autoencoder
  4. 2026Do Sparse Autoencoders Identify Reasoning Features in Language Models?
  5. 2025I Have Covered All the Bases Here: Interpreting Reasoning Features in Large Language Models via Sparse Autoencoders
  6. 2026From Input to Output: A Flexible Agent for Dual-End Interpretation of Sparse Autoencoder Features
  7. 2025Evaluating Sparse Autoencoders for Monosemantic Representation
  8. 2025Beyond Redundancy: Diverse and Specialized Multi-Expert Sparse Autoencoder