mlsanityData to 7 Oct 2026

Sparse Semantic Dimension as a Generalization Certificate for LLMs

arXiv 2602.11388 ↗First posted 11 Feb 2026cs.LG cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Standard statistical learning theory predicts that Large Language Models (LLMs) should overfit because their parameter counts vastly exceed the number of training tokens. Yet, in practice, they generalize robustly. We propose that the effective capacity controlling generalization lies in the geometry of the model's internal representations: while the parameter space is high-dimensional, the activation states lie on a low-dimensional, sparse manifold. To formalize this, we introduce the Sparse Semantic Dimension (SSD), a complexity measure derived from the active feature vocabulary of a Sparse Autoencoder (SAE) trained on the model's layers. Treating the LLM and SAE as frozen oracles, we utilize this framework to attribute the model's generalization capabilities to the sparsity of the dictionary rather than the total parameter count. Empirically, we validate this framework on GPT-2 Small and Gemma-2B, demonstrating that our bound provides non-vacuous certificates at realistic sample sizes. Crucially, we uncover a counter-intuitive "feature sharpness" scaling law: despite being an order of magnitude larger, Gemma-2B requires significantly fewer calibration samples to identify its active manifold compared to GPT-2, suggesting that larger models learn more compressible, distinct semantic structures. Finally, we show that this framework functions as a reliable safety monitor: out-of-distribution inputs trigger a measurable "feature explosion" (a sharp spike in active features), effectively signaling epistemic uncertainty through learned feature violation. Code is available at: https://github.com/newcodevelop/sparse-semantic-dimension.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 10 of its 14 references that are in the map. First posted on arXiv 11 Feb 2026.

Builds on

10 of 14 references.

References · 10

  1. 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
  2. 2016Understanding deep learning requires rethinking generalizationcited by 1,657
  3. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
  4. 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 527
  5. 2021The Intrinsic Dimension of Images and Its Impact on Learningcited by 307
  6. 2018Stronger generalization bounds for deep nets via a compression approachcited by 306
  7. 2019Intrinsic dimension of data representations in deep neural networkscited by 276
  8. 2019Uniform convergence may be unable to explain generalization in deep learningcited by 223
  9. 2023Non-Vacuous Generalization Bounds for Large Language Modelscited by 53
  10. 2024Unlocking Tokens as Data Points for Generalization Bounds on Larger Language Modelscited by 25

Built on by

Not cited yet.