mlsanityData to 7 Oct 2026

Diagnosing Generalization Failures from Representational Geometry Markers

arXiv 2603.01879 ↗First posted 2 Mar 2026cs.LG cs.AIcited by 2

Not on the map. Why?

Abstract (arXiv)

Generalization, the ability to perform well beyond the training context, is a hallmark of biological and artificial intelligence, yet anticipating unseen failures remains a central challenge. Conventional approaches often take a ``bottom-up'' mechanistic route by reverse-engineering interpretable features or circuits to build explanatory models. While insightful, these methods often struggle to provide the high-level, predictive signals for anticipating failure in real-world deployment. Here, we propose using a ``top-down'' approach to studying generalization failures inspired by medical biomarkers: identifying system-level measurements that serve as robust indicators of a model's future performance. Rather than mapping out detailed internal mechanisms, we systematically design and test network markers to probe structure, function links, identify prognostic indicators, and validate predictions in real-world settings. In image classification, we find that task-relevant geometric properties of in-distribution (ID) object manifolds consistently forecast poor out-of-distribution (OOD) generalization. In particular, reductions in two geometric measures, effective manifold dimensionality and utility, predict weaker OOD performance across diverse architectures, optimizers, and datasets. We apply this finding to transfer learning with ImageNet-pretrained models. We consistently find that the same geometric patterns predict OOD transfer performance more reliably than ID accuracy. This work demonstrates that representational geometry can expose hidden vulnerabilities, offering more robust guidance for model selection and AI interpretability.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 52 of its 77 references that are in the map. First posted on arXiv 2 Mar 2026.

Builds on

52 of 77 references.

References · 52

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
  3. 2016Densely Connected Convolutional Networkscited by 6,691
  4. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
  5. 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,827
  6. 2016Wide Residual Networkscited by 3,057
  7. 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,857
  8. 2016Aggregated Residual Transformations for Deep Neural Networkscited by 2,540
  9. 2013Describing Textures in the Wildcited by 2,110
  10. 2020Shortcut Learning in Deep Neural Networkscited by 1,928
  11. 2013Fine-Grained Visual Classification of Aircraftcited by 1,701
  12. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
38 more
  1. 2020Energy-based Out-of-distribution Detectioncited by 1,296
  2. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  3. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
  4. 2020Designing Network Design Spacescited by 954
  5. 2021Generalized Out-of-Distribution Detection: A Surveycited by 796
  6. 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
  7. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
  8. 2018Recognition in Terra Incognitacited by 655
  9. 2018Do Better ImageNet Models Transfer Better?cited by 560
  10. 2020Underspecification Presents Challenges for Credibility in Modern Machine Learningcited by 488
  11. 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
  12. 2018Rethinking ImageNet Pre-trainingcited by 376
  13. 2020An Investigation of Why Overparameterization Exacerbates Spurious Correlationscited by 310
  14. 2019Intrinsic dimension of data representations in deep neural networkscited by 276
  15. 2015Net2Net: Accelerating Learning via Knowledge Transfercited by 245
  16. 2023Causal Abstraction: A Theoretical Foundation for Mechanistic Interpretabilitycited by 194
  17. 2024Transcoders Find Interpretable LLM Feature Circuitscited by 186
  18. 2021Predicting with Confidence on Unseen Distributionscited by 116
  19. 2021Salient ImageNet: How to discover spurious features in Deep Learning?cited by 114
  20. 2021Transformer visualization via dictionary learning: contextualized embedding as a linear superposition of transformer factorscited by 106
  21. 2021On the Role of Neural Collapse in Transfer Learningcited by 100
  22. 2018On the importance of single directions for generalizationcited by 99
  23. 2021On the geometry of generalization and memorization in deep neural networkscited by 71
  24. 2017Classification and Geometry of General Perceptual Manifoldscited by 62
  25. 2021Neural population geometry: An approach for understanding biological and artificial neural networkscited by 52
  26. 2017Linking connectivity, dynamics and computations in low-rank recurrent neural networkscited by 51
  27. 2023NECO: NEural Collapse Based Out-of-distribution detectioncited by 41
  28. 2020Emergence of Separable Manifolds in Deep Language Representationscited by 39
  29. 2023The Tunnel Effect: Building Data Representations in Deep Neural Networkscited by 34
  30. 2025Everything, Everywhere, All at Once: Is Mechanistic Interpretability Identifiable?cited by 32
  31. 2022Understanding Rare Spurious Correlations in Neural Networkscited by 27
  32. 2025The Geometry of Prompting: Unveiling Distinct Mechanisms of Task Adaptation in Language Modelscited by 14
  33. 2023Variance-Covariance Regularization Improves Representation Learningcited by 13
  34. 2024What Variables Affect Out-of-Distribution Generalization in Pretrained Models?cited by 13
  35. 2025Controlling Neural Collapse Enhances Out-of-Distribution Detection and Transfer Learningcited by 11
  36. 2022Linear Classification of Neural Manifolds with Correlated Variabilitycited by 9
  37. 2025Feature Learning beyond the Lazy-Rich Dichotomy: Insights from Representational Geometrycited by 9
  38. 2025Can Interpretation Predict Behavior on Unseen Data?cited by 8

50 most cited of 52 All 52 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.