When Does Knowledge Distillation Hurt? Reliability-Aware Distillation for Low-Resource Language Summarization
Summary
Knowledge distillation (KD), a common method for compressing sequence-to-sequence models, often has unexamined per-sample effects. Research on the BanSum Bangla summarization benchmark reveals that standard KD improves ROUGE-L by only +0.0003, and an estimated 51.3% of training samples actively harm student validation loss. To address this, two reliability-aware distillation methods are proposed: CHAD (Counterfactual Harm-Aware Distillation) and EWAD+CPDP (Entropy-Weighted Adaptive Distillation with Capacity-Proportional Geometric Constraint). 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, vocabulary-incompatible teacher. Both methods significantly outperform standard KD on BanSum, with CHAD achieving +0.0173 ROUGE-L and EWAD+CPDP +0.0219 ROUGE-L. Despite using only 60M parameters, they surpass a fine-tuned Qwen 2.5-3B model. EWAD+CPDP also beats the cross-entropy baseline on 10 of 15 XL-Sum languages. Code and trained models are released.
Key takeaway
For Machine Learning Engineers optimizing sequence-to-sequence models in low-resource language settings, be aware that standard knowledge distillation can actively degrade performance on over 50% of training samples. You should instead consider implementing reliability-aware methods like CHAD or EWAD+CPDP. These approaches offer substantial ROUGE-L improvements, outperforming much larger models (e.g., Qwen 2.5-3B) with only 60M parameters, making them critical for efficient and effective model compression and performance.
Key insights
Reliability-aware knowledge distillation methods overcome standard KD's per-sample harm, significantly boosting low-resource language summarization performance.
Principles
- Standard KD can harm student validation loss.
- Complementary teacher signals enhance distillation.
- Gradient alignment measures per-sample KD usefulness.
Method
CHAD uses gradient alignment to gate distillation. EWAD+CPDP combines token-level entropy-weighted adaptive distillation with a capacity-proportional geometric constraint from a second, vocabulary-incompatible teacher.
In practice
- Implement CHAD or EWAD+CPDP for summarization.
- Utilize multiple teachers for complementary signals.
- Achieve strong performance with 60M parameter models.
Topics
- Knowledge Distillation
- Low-Resource NLP
- Text Summarization
- Sequence-to-Sequence Models
- Model Compression
- Distillation Reliability
Best for: AI Engineer, Research Scientist, 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 Computation and Language.