ArXiv

Beyond the Hard Budget: Sparsity Regularizers for More Interpretable Top-k Sparse Autoencoders

Authors
Nathanaël Jacquier, Maria Vakalopoulou, Mahdi S. Hosseini
Categories
cs.LG, cs.AI
arXiv
https://arxiv.org/abs/2606.27321v1
PDF
https://arxiv.org/pdf/2606.27321v1

Brief

Top-k sparse autoencoders, which fix a hard budget k, are augmented with two compatible sparsity regularizers: an ℓ1 penalty on off-support units and a scale-invariant ℓ1/ℓ2 ratio on batch-active units, both applied before Top-k. Evaluated on two datasets and three vision foundation models, these regularizers consistently improve monosemanticity without harming reconstruction, and the ℓ1/ℓ2 variant further concentrates code and improves robustness to inference-time k and small-budget linear probing. Summary based on the paper abstract.

Why it matters

Top-k SAE augmented with two sparsity regularizers—an ℓ1 penalty on off-support (unselected) units and a scale-invariant ℓ1/ℓ2 ratio on batch-active units—by Jacquier, Vakalopoulou, and Hosseini (published 2026-06-25) increases monosemanticity across two datasets, three vision foundation models, and a range of k while preserving reconstruction quality.

Key details

  • The ℓ1/ℓ2 ratio concentrates information into fewer effective latents, making reconstruction more robust to the inference-time choice of k and improving small-budget linear probing; both penalties act on activations before Top-k and are applied only to units selected at least once within the batch.
Source evidence

Abstract

Sparse autoencoders (SAEs) have become a leading tool for interpreting the representations of vision foundation models, decomposing their polysemantic activations into a larger set of sparse, more monosemantic features. The Top-$k$ SAE, a now-standard variant, enforces sparsity architecturally through its activation function, retaining only the $k$ most active latents per input. Because it was designed precisely to avoid the $\ell1$ penalty used by earlier SAEs and its known drawbacks, it has not been combined with an explicit sparsity regularizer, despite retaining limitations of its own, such as a budget $k$ that is fixed regardless of input complexity and a tendency to overfit to the training value of $k$. We introduce two sparsity regularizers compatible with the Top-$k$ architecture, both acting on the activations before the Top-$k$ selection: an $\ell1$ penalty on the unselected (off-support) units, and a scale-invariant $\ell1/\ell2$-ratio penalty that concentrates the code onto fewer effective units. Both penalties are applied only to the batch-active units, those selected by the Top-$k$ operator at least once within the batch. Across two datasets, three vision foundation models, and a range of $k$, both regularizers consistently improve monosemanticity at no cost to reconstruction quality. The $\ell1/\ell2$ penalty further concentrates information into fewer latents, making reconstruction more robust to the inference-time choice of $k$ and improving small-budget linear probing. Our central finding is that hard architectural sparsity and soft sparsity regularization are complementary rather than mutually exclusive.