Geometric Limits of Knowledge Distillation: A Minimum-Width Theorem via Superposition Theory
Not on the map. Why?
Abstract (arXiv)
Knowledge distillation compresses large teachers into smaller students, but performance saturates at a loss floor that persists across training methods and objectives. We argue this floor is geometric: neural networks represent far more features than dimensions through superposition, and a student of width dS can encode at most dS · g(α) features, where g(α) = 1/((1-α)lnfrac11-α) is a sparsity-dependent capacity function. Features beyond this budget are permanently lost, yielding an importance-weighted loss floor. We validate on a toy model (48 configurations, median accuracy >93%) and on Pythia-410M, where sparse autoencoders measure F ≈ 28,700 features at α ≈ 0.992 (critical width dS^* ≈ 1,065). Distillation into five student widths confirms the predicted monotonic floor ordering. The observed floor decomposes into a geometric component and a width-independent architectural baseline (R2 = 0.993). Linear probing shows coarse concepts survive even 88% feature loss, revealing the floor arises from aggregate loss of fine-grained features in the importance distribution's long tail. Our results connect representation geometry to distillation limits and provide a practical tool for predicting distillation performance from SAE measurements alone.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 8 of its 16 references that are in the map. First posted on arXiv 5 Apr 2026.
Builds on
8 of 16 references.
References · 8
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2014FitNets: Hints for Thin Deep Netscited by 1,619
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2022Toy Models of Superpositioncited by 928
- 2019Contrastive Representation Distillationcited by 734
- 2022Polysemanticity and Capacity in Neural Networkscited by 74
Built on by
Citing papers per year: 2026: 1.
1 paper.