Sparse Prototype Code Underlies Classification and Prediction Across Modalities
Not on the map. Why?
Abstract (arXiv)
Neural representations have become a central tool for studying the internal mechanisms of modern AI models, yet their complex high-dimensional structure makes them difficult to interpret. We show that classification tasks give rise to a universal representational geometry, shared across state-of-the-art models in vision, audio, and language processing. The key structure is that within-class variability is not random in representation space. Instead, its classifier-relevant component has strong and structured correlations with the class's own centroid and with the centroids of its competing classes. Building on this observation, we derive an analytical mean-field theory governed mainly by the variability along true-class and rival-class centroid coordinates, together with a global renormalization of the class radius that compensates for the non-Gaussian statistics of real representations. The theory accurately predicts classification accuracy across architectures and modalities. The relevant geometric quantities improve systematically with model scale, mirroring the observed gains in accuracy. A striking feature of the theory is its sparsity: accurate prediction requires only a small set of centroid coordinates associated with the true class and its strongest rivals - connecting our framework to sparse-feature extraction approaches such as sparse autoencoders. Together, these results provide a parsimonious predictive theory of neural representations and suggest that classification in deep networks is governed by a sparse, centroid-aligned structure embedded within the full high-dimensional representation space.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 41 of its 53 references that are in the map. First posted on arXiv 16 Aug 2026.
Builds on
41 of 53 references.
References · 41
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2025Qwen3 Technical Reportcited by 8,195
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2021On the Opportunities and Risks of Foundation Modelscited by 3,961
- 2022A ConvNet for the 2020scited by 3,545
- 2017Prototypical Networks for Few-shot Learningcited by 3,001
29 more
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2025DINOv3cited by 1,297
- 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,046
- 2022Toy Models of Superpositioncited by 928
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2020VGGSound: A Large-scale Audio-Visual Datasetcited by 641
- 2021AST: Audio Spectrogram Transformercited by 634
- 2018Do Better ImageNet Models Transfer Better?cited by 560
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2024Smarter, Better, Faster, Longer: A Modern Bidirectional Encoder for Fast, Memory Efficient, and Long Context Finetuning and Inferencecited by 527
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 425
- 2019Linguistic Knowledge and Transferability of Contextual Representationscited by 382
- 20242 OLMo 2 Furiouscited by 302
- 2025Layer by Layer: Uncovering Hidden Representations in Language Modelscited by 293
- 2021Unsolved Problems in ML Safetycited by 291
- 2024Nomic Embed: Training a Reproducible Long Context Text Embeddercited by 255
- 2013k-Sparse Autoencoderscited by 209
- 2026Scaling Monosemanticity: Extracting Interpretable Features from Claude 3 Sonnetcited by 175
- 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 156
- 2018Dissecting Contextual Word Embeddings: Architecture and Representationcited by 146
- 2021On the Role of Neural Collapse in Transfer Learningcited by 100
- 2021On the geometry of generalization and memorization in deep neural networkscited by 71
- 2017Classification and Geometry of General Perceptual Manifoldscited by 62
- 2023CED: Consistent ensemble distillation for audio taggingcited by 29
- 2025SSLAM: Enhancing Self-Supervised Models with Audio Mixtures for Polyphonic Soundscapescited by 16
- 2022Linear Classification of Neural Manifolds with Correlated Variabilitycited by 9
- 2023Multilingual Nonce Dependency Treebanks: Understanding how Language Models represent and process syntactic structurecited by 5
- 2023Probing Biological and Artificial Neural Networks with Task-dependent Neural Manifoldscited by 4
- 2026Directional Neural Collapse Explains Few-Shot Transfer in Self-Supervised Learningcited by 3
Built on by
Not cited yet.