IC-525GPT-2 small's residual stream at layer 8 decomposes into two sub-spaces of approximately 25% and 75% of the dimensionality

Report a problem with this record

Leo Gao, Tom Dupre la Tour, Henk Tillman, Gabriel Goh, Rajan Troll, Alec Radford, Ilya Sutskever, Jan Leike, Jeffrey Wu

SourceScaling and evaluating sparse autoencoders

By training a sparse autoencoder on GPT-2 small's layer 8 post-MLP residual stream and performing SVD on the learned encoding matrix, the authors find that the residual stream is composed of two distinct linear sub-spaces. Approximately 25% of the autoencoder latents project primarily onto a sub-space spanning 25% of the dimensions, while the remaining 75% project onto the complementary 75%. Latents in the smaller sub-space have larger encoder norms and activate on fewer vocabulary tokens (less than 100 effective tokens) compared to the larger sub-space (up to 1000 effective tokens). The separation is revealed by ordering singular vectors via a 1D UMAP embedding of pairwise distances between normalized left singular vectors.

Evidence
observational
Key metric
"about 25% of latents mostly project on a sub-space using 25% of dimensions"; smaller cluster activates on "less than 100" effective vocabulary tokens, larger cluster "up to 1000"
Caveat
The sub-space separation is 'fuzzily' defined; the authors note that the difference in encoder norm between clusters 'can significantly guide the svd to separate these two sub-spaces,' suggesting the separation may be partly an artefact of the SVD decomposition rather than a hard boundary in the model.
Model
GPT-2
Methods
UMAP [primary], Sparse autoencoder / Sparse autoencoders / K-sparse autoencoder / Topk sparse autoencoder / Scaling and Evaluating Sparse Autoencoders / Cunningham et al. 2023 (sparse autoencoders) [supporting]
Related work
Elhage et al. 2022 (Toy models of superposition) [context]
Related findings
IC-526, IC-527
Extraction
automatic-extraction