mlsanityData to 7 Oct 2026

Teacher Supervision over Representation Equivalence Classes

arXiv 2607.03572 ↗First posted 3 Jul 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Knowledge distillation is usually framed as a choice of what to match in the teacher - its logits, hidden features, or sample relations - which presupposes that the teacher's representation has absolute coordinates to match. It does not: a pretrained representation is identifiable only up to an orthogonal-and-isotropic-scaling equivalence class, so a student should learn the teacher's equivalence class, not its features. The organizing fact is that capability is the teacher's output function, a class invariant that factors through the quotient by the class action, so an objective recovers capability exactly when it is defined there. This makes absolute feature matching ill-posed, and admissible supervision a matter of targeting class invariants (Gram structure, CKA, principal subspaces) or aligning coordinates first, unifying feature matching, relational distillation, alignment, and grafting in one geometric account. We validate our framework on Qwen2.5 and Llama-3.1. A restoration study recovers a corrupted model's representation (CKA ~ 0.99) but not its capability, and an ablation isolates the cause: output-function (logit) matching drives capability, while matching hidden representations aligns geometry without restoring function. Recovery is confined to the corpus-covered region, and a graft study confirms that boundary overlap predicts transplant success but is necessary, not sufficient.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 17 of its 23 references that are in the map. First posted on arXiv 3 Jul 2026.

Builds on

17 of 23 references.

References · 17

  1. 2024The Llama 3 Herd of Modelscited by 14,192
  2. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  3. 2024Qwen2.5 Technical Reportcited by 4,713
  4. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  5. 2014FitNets: Hints for Thin Deep Netscited by 1,619
  6. 2022Toy Models of Superpositioncited by 928
  7. 2022STaR: Bootstrapping Reasoning With Reasoningcited by 895
  8. 2019Relational Knowledge Distillationcited by 873
  9. 2019Similarity-Preserving Knowledge Distillationcited by 507
  10. 2023Reinforced Self-Training (ReST) for Language Modelingcited by 486
  11. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
  12. 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 285
5 more
  1. 2014Understanding image representations by measuring their equivariance and equivalencecited by 240
  2. 2021Revisiting Model Stitching to Compare Neural Representationscited by 203
  3. 2023Similarity of Neural Network Models: A Survey of Functional and Representational Measurescited by 128
  4. 2026AgentArk: Distilling Multi-Agent Intelligence into a Single LLM Agentcited by 12
  5. 2025ProST: Progressive Sub-task Training for Pareto-Optimal Multi-agent Systems Using Small Language Modelscited by 1

Built on by

Not cited yet.