From Data Statistics to Feature Geometry: How Correlations Shape Superposition

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Prieto, Lucas, Stevinson, Edward, Barsbey, Melih, Birdal, Tolga, Mediano, Pedro A. M.
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917330348408832
author Prieto, Lucas
Stevinson, Edward
Barsbey, Melih
Birdal, Tolga
Mediano, Pedro A. M.
author_facet Prieto, Lucas
Stevinson, Edward
Barsbey, Melih
Birdal, Tolga
Mediano, Pedro A. M.
contents A central idea in mechanistic interpretability is that neural networks represent more features than they have dimensions, arranging them in superposition to form an over-complete basis. This framing has been influential, motivating dictionary learning approaches such as sparse autoencoders. However, superposition has mostly been studied in idealized settings where features are sparse and uncorrelated. In these settings, superposition is typically understood as introducing interference that must be minimized geometrically and filtered out by non-linearities such as ReLUs, yielding local structures like regular polytopes. We show that this account is incomplete for realistic data by introducing Bag-of-Words Superposition (BOWS), a controlled setting to encode binary bag-of-words representations of internet text in superposition. Using BOWS, we find that when features are correlated, interference can be constructive rather than just noise to be filtered out. This is achieved by arranging features according to their co-activation patterns, making interference between active features constructive, while still using ReLUs to avoid false positives. We show that this kind of arrangement is more prevalent in models trained with weight decay and naturally gives rise to semantic clusters and cyclical structures which have been observed in real language models yet were not explained by the standard picture of superposition. Code for this paper can be found at https://github.com/LucasPrietoAl/correlations-feature-geometry.
format Preprint
id arxiv_https___arxiv_org_abs_2603_09972
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle From Data Statistics to Feature Geometry: How Correlations Shape Superposition
Prieto, Lucas
Stevinson, Edward
Barsbey, Melih
Birdal, Tolga
Mediano, Pedro A. M.
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
A central idea in mechanistic interpretability is that neural networks represent more features than they have dimensions, arranging them in superposition to form an over-complete basis. This framing has been influential, motivating dictionary learning approaches such as sparse autoencoders. However, superposition has mostly been studied in idealized settings where features are sparse and uncorrelated. In these settings, superposition is typically understood as introducing interference that must be minimized geometrically and filtered out by non-linearities such as ReLUs, yielding local structures like regular polytopes. We show that this account is incomplete for realistic data by introducing Bag-of-Words Superposition (BOWS), a controlled setting to encode binary bag-of-words representations of internet text in superposition. Using BOWS, we find that when features are correlated, interference can be constructive rather than just noise to be filtered out. This is achieved by arranging features according to their co-activation patterns, making interference between active features constructive, while still using ReLUs to avoid false positives. We show that this kind of arrangement is more prevalent in models trained with weight decay and naturally gives rise to semantic clusters and cyclical structures which have been observed in real language models yet were not explained by the standard picture of superposition. Code for this paper can be found at https://github.com/LucasPrietoAl/correlations-feature-geometry.
title From Data Statistics to Feature Geometry: How Correlations Shape Superposition
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2603.09972