On the Relationship Between Representation Geometry and Generalization in Deep Neural Networks
Not on the map. Why?
Abstract (arXiv)
We investigate the relationship between representation geometry and neural network performance. Analyzing 52 pretrained ImageNet models across 13 architecture families, we show that effective dimension -- an unsupervised geometric metric -- strongly predicts accuracy. Output effective dimension achieves partial r=0.75 (p < 10^(-10)) after controlling for model capacity, while total compression achieves partial r=-0.72. These findings replicate across ImageNet and CIFAR-10, and generalize to NLP: effective dimension predicts performance for 8 encoder models on SST-2/MNLI and 15 decoder-only LLMs on AG News (r=0.69, p=0.004), while model size does not (r=0.07). We establish bidirectional causality: degrading geometry via noise causes accuracy loss (r=-0.94, p < 10^(-9)), while improving geometry via PCA maintains accuracy across architectures (-0.03pp at 95% variance). This relationship is noise-type agnostic -- Gaussian, Uniform, Dropout, and Salt-and-pepper noise all show |r| > 0.90. These results establish that effective dimension provides domain-agnostic predictive and causal information about neural network performance, computed entirely without labels.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 46 of its 68 references that are in the map. First posted on arXiv 28 Jan 2026.
Builds on
46 of 68 references.
References · 46
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2016Densely Connected Convolutional Networkscited by 6,720
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2022A ConvNet for the 2020scited by 3,579
34 more
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2023Qwen Technical Reportcited by 3,344
- 2016Wide Residual Networkscited by 3,070
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,682
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
- 2019Searching for MobileNetV3cited by 1,994
- 2019Similarity of Neural Network Representations Revisitedcited by 1,755
- 2016Understanding deep learning requires rethinking generalizationcited by 1,674
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
- 2014How transferable are features in deep neural networks?cited by 1,225
- 2018ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Designcited by 1,136
- 2021EfficientNetV2: Smaller Models and Faster Trainingcited by 1,135
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,130
- 2016SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model sizecited by 1,060
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2015Deep Learning and the Information Bottleneck Principlecited by 802
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
- 2017Exploring Generalization in Deep Learningcited by 600
- 2023Textbooks Are All You Needcited by 569
- 2017Opening the Black Box of Deep Neural Networks via Informationcited by 503
- 2024The Platonic Representation Hypothesiscited by 382
- 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 370
- 2018Insights on representational similarity in neural networks with canonical correlationcited by 324
- 2018Stronger generalization bounds for deep nets via a compression approachcited by 307
- 2019Intrinsic dimension of data representations in deep neural networkscited by 282
- 2018Estimating the intrinsic dimension of datasets by a minimal neighborhood informationcited by 250
- 2017Resurrecting the sigmoid in deep learning through dynamical isometry: theory and practicecited by 132
- 2018Tropical Geometry of Deep Neural Networkscited by 95
- 2018An elementary introduction to information geometrycited by 76
- 2025Harnessing the Universal Geometry of Embeddingscited by 55
- 2018On Characterizing the Capacity of Neural Networks using Algebraic Topologycited by 38
- 2025Deep sequence models tend to memorize geometrically; it is unclear whycited by 9
Built on by
Not cited yet.