Sign-Aware Gated Sparse Autoencoders: Modeling Anticorrelated Features with Bi-Jump-ReLU Activations
Not on the map. Why?
Abstract (arXiv)
Sparse Autoencoders (SAEs) extract interpretable features from Large Language Model activations, but standard variants enforce non-negative latents, so a bidirectional semantic axis (e.g., "pressure too high" vs. "pressure too low") must be split across two latents, wasting dictionary capacity on anticorrelated features. We propose the Sign-Aware Gated SAE (SA-GSAE), which combines two-sided gated sparsity, signed shrinkage-free magnitudes, and auxiliary gate supervision in a new Bi-Jump-ReLU activation, so that a single latent carries both polarities of one decoder direction; parameter accounting shows sign-awareness stays parameter-efficient even when anticorrelated pairs are rare. Across three mid-depth hookpoints on Pythia-1B and SmolLM3-3B (six cells, three seeds), a half-width SA-GSAE empirically dominates the aggregate mean frontier of a full-width Gated SAE on three of six cells, matches its R^2 within 0.025 on the remaining three, and cuts dead fraction by 0.35-0.82 absolute at matched L_0 = 64 on all six. Ablations show the two-sided gate and the auxiliary loss are essential whereas per-polarity asymmetry is not; we recommend the fully tied symmetric variant as the default. A blinded semantic audit finds nameable opposition between a latent's two sides is rare for SA-GSAE and all tested baselines, while sign-conditioned interventions show a single signed latent acts as a bidirectional causal dial where a pair of "opposite" non-negative latents does not; we scope interpretability claims accordingly. At full width, SA-GSAE is over-parameterized and its reported configuration exhibits a reproducible reconstruction collapse at the SmolLM3-3B residual-stream site; the recommended configuration (small threshold initialization with dead-latent threshold resets) prevents it.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 29 of its 44 references that are in the map. First posted on arXiv 27 May 2026.
Builds on
29 of 44 references.
References · 29
- 2017Attention Is All You Needcited by 59,932
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2022Toy Models of Superpositioncited by 928
- 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 381
- 2024Jumping Ahead: Improving Reconstruction Fidelity with JumpReLU Sparse Autoencoderscited by 283
- 2013k-Sparse Autoencoderscited by 209
- 2024Not All Language Model Features Are One-Dimensionally Linearcited by 201
- 2024Improving Dictionary Learning with Gated Sparse Autoencoderscited by 171
17 more
- 2024A is for Absorption: Studying Feature Splitting and Absorption in Sparse Autoencoderscited by 148
- 2025Learning Multi-Level Features with Matryoshka Sparse Autoencoderscited by 136
- 2024Llama Scope: Extracting Millions of Features from Llama-3.1-8B with Sparse Autoencoderscited by 133
- 2024BatchTopK Sparse Autoencoderscited by 126
- 2025SAEBench: A Comprehensive Benchmark for Sparse Autoencoders in Language Model Interpretabilitycited by 118
- 2021An Interpretability Illusion for BERTcited by 94
- 2025Sparse Autoencoders Trained on the Same Data Learn Different Featurescited by 86
- 2024Towards Principled Evaluations of Sparse Autoencoders for Interpretability and Controlcited by 83
- 2025Sparse Autoencoders Do Not Find Canonical Units of Analysiscited by 80
- 2024Measuring Progress in Dictionary Learning for Language Model Interpretability with Board Game Modelscited by 51
- 2024Efficient Dictionary Learning with Switch Sparse Autoencoderscited by 38
- 2024Compute Optimal Inference and Provable Amortisation Gap in Sparse Autoencoderscited by 17
- 2025Feature Hedging: Correlated Features Break Narrow Sparse Autoencoderscited by 15
- 2025On the Limits of Sparse Autoencoders: A Theoretical Framework and Reweighted Remedycited by 12
- 2025Evaluating Adversarial Robustness of Concept Representations in Sparse Autoencoderscited by 11
- 2025AbsTopK: Rethinking Sparse Autoencoders For Bidirectional Featurescited by 8
- 2025Evaluating SAE interpretability without explanationscited by 5
Built on by
Not cited yet.