A new study investigates how knowledge distillation affects the internal mechanisms of large language models, specifically comparing GPT-2 with its distilled version, DistilGPT-2. Using techniques like linear CKA and activation patching, the research found that while the Indirect Object Identification (IOI) circuit in DistilGPT-2 experienced a complete collapse in functionality, the Greater-Than (GT) circuit largely survived. This survival was attributed to the remapping of a key comparison head from GPT-2's layer 9 to DistilGPT-2's layer 4, indicating that the compressibility of specific circuits dictates their resilience during distillation. AI
IMPACT Reveals that distillation's impact on LLM circuits varies, with some critical functionalities collapsing while others adapt, suggesting a need for more nuanced evaluation beyond aggregate metrics.
RANK_REASON Academic paper detailing mechanistic interpretability study of LLM distillation. [lever_c_demoted from research: ic=1 ai=1.0]
- arXiv
- Deshan Sumanathilaka
- DistilGPT-2
- GPT-2
- Greater-Than (GT)
- Hugging Face
- Indirect Object Identification (IOI)
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →