English

Geometric Limits of Knowledge Distillation: A Minimum-Width Theorem via Superposition Theory

Machine Learning 2026-04-08 v2 Artificial Intelligence

Abstract

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 dSd_S can encode at most dSg(α)d_S \cdot g(\alpha) features, where g(α)=1/((1α)ln11α)g(\alpha) = 1/((1-\alpha)\ln\frac{1}{1-\alpha}) 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 F28,700F \approx 28{,}700 features at α0.992\alpha \approx 0.992 (critical width dS1,065d_S^* \approx 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.993R^2 = 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.

Keywords

Cite

@article{arxiv.2604.04037,
  title  = {Geometric Limits of Knowledge Distillation: A Minimum-Width Theorem via Superposition Theory},
  author = {Nilesh Sarkar and Dawar Jyoti Deka},
  journal= {arXiv preprint arXiv:2604.04037},
  year   = {2026}
}
R2 v1 2026-07-01T11:54:21.661Z