Projecting Assumptions: The Duality Between Sparse Autoencoders and Concept Geometry

Sai Sumedh R. Hindupur, Ekdeep Singh Lubana, Thomas Fel, Demba Ba
3/3/2025
cs.LGcs.AI

No official code found. Best implementation: EkdeepSLubana/spadeFormalGrammars (5 stars, MIT, last commit 2025-03-08). 1 more below.

Abstract

Sparse Autoencoders (SAEs) are widely used to interpret neural networks by identifying meaningful concepts from their representations. However, do SAEs truly uncover all concepts a model relies on, or are they inherently biased toward certain kinds of concepts? We introduce a unified framework that recasts SAEs as solutions to a bilevel optimization problem, revealing a fundamental challenge: each SAE imposes structural assumptions about how concepts are encoded in model representations, which in turn shapes what it can and cannot detect. This means different SAEs are not interchangeable -- switching architectures can expose entirely new concepts or obscure existing ones. To systematically probe this effect, we evaluate SAEs across a spectrum of settings: from controlled toy models that isolate key variables, to semi-synthetic experiments on real model activations and finally to large-scale, naturalistic datasets. Across this progression, we examine two fundamental properties that real-world concepts often exhibit: heterogeneity in intrinsic dimensionality (some concepts are inherently low-dimensional, others are not) and nonlinear separability. We show that SAEs fail to recover concepts when these properties are ignored, and we design a new SAE that explicitly incorporates both, enabling the discovery of previously hidden concepts and reinforcing our theoretical insights. Our findings challenge the idea of a universal SAE and underscores the need for architecture-specific choices in model interpretability. Overall, we argue an SAE does not just reveal concepts -- it determines what can be seen at all.

View on arXivView PDF

Code Implementations(2)

MIT

Codebase for the paper "Projecting Assumptions: The Duality Between Sparse Autoencoders and Concept Geometry"

51Python, Jupyter NotebookMar 8, 20251 years agoMIT
MIT

Codebase for synthetic experiments and figures in the paper "Projecting Assumptions: The Duality Between Sparse Autoencoders and Concept Geometry"

11Python, Jupyter NotebookApr 2, 20251 years agoMIT

Similar Papers

Cite this paper

@article{hindupur2025projecting,
  title  = {Projecting Assumptions: The Duality Between Sparse Autoencoders and Concept Geometry},
  author = {Sai Sumedh R. Hindupur and Ekdeep Singh Lubana and Thomas Fel and Demba Ba},
  year   = {2025},
  eprint = {2503.01822},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2503.01822v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2503.01822.svg)](https://codeofpaper.com/paper/2503.01822)

Discussion