mlsanityData to 14 Oct 2026

Distilled Circuits: A Mechanistic Study of Internal Restructuring in Knowledge Distillation

arXiv 2505.10822 ↗First posted 16 May 2025cs.LGcited by 0

On the map: Knowledge distillation → Knowledge distillation for LLMs

Abstract (arXiv)

Knowledge distillation compresses a larger neural model (teacher) into smaller, faster student models by training them to match teacher outputs. However, the internal computational transformations that occur during this process remain poorly understood. We apply techniques from mechanistic interpretability to analyze how internal circuits, representations, and activation patterns differ between teachers and students. Focusing on GPT2 and its distilled counterpart DistilGPT2, and generalizing our findings to both bidirectional architectures and larger model pairs, we find that student models can reorganize, compress, and discard teacher components, often resulting in a stronger reliance on fewer individual components. To quantify functional alignment beyond output similarity, we introduce an alignment metric based on influence-weighted component similarity, validated across multiple tasks. Our findings reveal that while knowledge distillation preserves broad functional behaviors, it also causes significant shifts in internal computation, with important implications for the robustness and generalization capacity of distilled models.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Knowledge distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 35 of its 47 references that are in the map. First posted on arXiv 16 May 2025.

Builds on

35 of 47 references.

Closest ancestors · 5

Same direction, newest first

  1. 2025Towards Understanding Distilled Reasoning Models: A Representational Approach5 Mar
  2. 2024Survey on Knowledge Distillation for Large Language Models: Methods, Evaluation, and Application2 Jul
  3. 2024A Survey on Knowledge Distillation of Large Language Models20 Feb
  4. 2023Towards Understanding and Improving Knowledge Distillation for Neural Machine Translation14 May
  5. 2023Neural Architecture Search for Effective Teacher-Student Knowledge Transfer in Language Models16 Mar

Other references · 30

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2024The Llama 3 Herd of Modelscited by 14,715
  4. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  5. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  6. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  7. 2019Similarity of Neural Network Representations Revisitedcited by 1,755
  8. 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
  9. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
  10. 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
  11. 2019Contrastive Representation Distillationcited by 748
  12. 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 724
18 more
  1. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  2. 2024Mechanistic Interpretability for AI Safety -- A Reviewcited by 422
  3. 2019On the Efficacy of Knowledge Distillationcited by 381
  4. 2024Measuring short-form factuality in large language modelscited by 373
  5. 2020Pretrained Transformers Improve Out-of-Distribution Robustnesscited by 294
  6. 2023How does GPT-2 compute greater-than?: Interpreting mathematical abilities in a pre-trained language modelcited by 202
  7. 2021Does Knowledge Distillation Really Work?cited by 190
  8. 2021Towards Understanding Knowledge Distillationcited by 158
  9. 2019Knowledge Distillation from Internal Representationscited by 87
  10. 2024LLM Pruning and Distillation in Practice: The Minitron Approachcited by 82
  11. 2023Successor Heads: Recurring, Interpretable Attention Heads In The Wildcited by 77
  12. 2018Knowledge Transfer with Jacobian Matchingcited by 76
  13. 2022Measuring the Mixing of Contextual Information in the Transformercited by 65
  14. 2024A Multimodal Automated Interpretability Agentcited by 63
  15. 2023FIND: A Function Description Benchmark for Evaluating Interpretability Methodscited by 32
  16. 2025Towards Understanding Fine-Tuning Mechanisms of LLMs via Circuit Analysiscited by 20
  17. 2023Towards Interpretable Sequence Continuation: Analyzing Shared Circuits in Large Language Modelscited by 15
  18. 2024Adaptive Circuit Behavior and Generalization in Mechanistic Interpretabilitycited by 7

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Circuit Distillation
  2. 2026Knowledge Distillation under Teacher Misspecification: An Order-Parameter Analysis of the Gap between Teacher Mimicry and Task Performance
  3. 2025A Functional Perspective on Knowledge Distillation in Neural Networks
  4. 2023Supervision Complexity and its Role in Knowledge Distillation
  5. 2026H'an D=an Xu'e B`u (Mimicry) or Q=ing Ch=u Y'u L'an (Mastery)? A Cognitive Perspective on Reasoning Distillation in Large Language Models
  6. 2024Quantifying Knowledge Distillation Using Partial Information Decomposition
  7. 2026DistillLens: Symmetric Knowledge Distillation Through Logit Lens
  8. 2026Representational Difference Distillation