When Does Knowledge Distillation Hurt? Reliability-Aware Distillation for Low-Resource Language Summarization
Summary
A study on knowledge distillation (KD) for sequence-to-sequence models reveals that standard KD minimally improves ROUGE-L by +0.0003 on the BanSum Bangla summarization benchmark, with an estimated 51.3% of training samples actively harming student validation loss. Researchers propose two reliability-aware distillation methods: CHAD (Counterfactual Harm-Aware Distillation) and EWAD+CPDP. CHAD uses gradient alignment to measure per-sample KD usefulness, while EWAD+CPDP combines token-level entropy-weighted adaptive distillation with a capacity-proportional geometric constraint from a second teacher. On BanSum, CHAD achieves +0.0173 ROUGE-L and EWAD+CPDP achieves +0.0219 ROUGE-L, both substantially outperforming standard KD and a 50x larger Qwen 2.5-3B model with only 60M parameters. EWAD+CPDP also beats the cross-entropy baseline on 10 of 15 XL-Sum languages.
Key takeaway
For NLP engineers optimizing sequence-to-sequence models, especially in low-resource language summarization, you should re-evaluate standard knowledge distillation practices. Your current KD implementation might be degrading performance on over half your training data. Instead, consider adopting reliability-aware methods like CHAD or EWAD+CPDP, which have demonstrated significant ROUGE-L gains and can outperform much larger models, offering a path to more efficient and effective model compression.
Key insights
Standard knowledge distillation can harm over half of training samples; reliability-aware methods significantly improve performance.
Principles
- Per-sample KD usefulness can be quantified via gradient alignment.
- Complementary signals from multiple teachers enhance distillation reliability.
- Standard KD's per-sample effects are often detrimental.
Method
CHAD measures per-sample KD usefulness via gradient alignment and trains a lightweight gate. EWAD+CPDP combines token-level entropy-weighted adaptive distillation with a capacity-proportional geometric constraint from a vocabulary-incompatible teacher.
In practice
- Assess per-sample KD impact using gradient alignment.
- Implement EWAD+CPDP for low-resource summarization tasks.
- Consider multi-teacher setups for diverse language coverage.
Topics
- Knowledge Distillation
- Low-Resource NLP
- Text Summarization
- Sequence-to-Sequence Models
- ROUGE-L
- CHAD
- EWAD+CPDP
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.