When Does Knowledge Distillation Hurt? Reliability-Aware Distillation for Low-Resource Language Summarization
Summary
New research on knowledge distillation (KD) for low-resource language summarization reveals that standard KD offers only marginal ROUGE-L gains (+0.0003) on the BanSum Bangla summarization benchmark, with 51.3% of training samples actively harming student validation loss. Two novel reliability-aware distillation methods are introduced: CHAD (Counterfactual Harm-Aware Distillation), which uses gradient alignment to gate per-sample KD usefulness, and EWAD+CPDP, combining token-level entropy-weighted distillation with a cross-vocabulary geometric constraint from a second teacher. On BanSum, CHAD achieved +0.0173 ROUGE-L and EWAD+CPDP achieved +0.0219 ROUGE-L over standard KD. Both 60M-parameter methods significantly outperformed a fine-tuned Qwen-2.5-3B model (50x larger). EWAD+CPDP also improved performance on 10 out of 15 typologically diverse XL-Sum languages, particularly when teachers offered complementary signals.
Key takeaway
For machine learning engineers optimizing sequence-to-sequence models in low-resource settings, relying on standard knowledge distillation is often counterproductive. You should instead implement reliability-aware distillation methods like CHAD or EWAD+CPDP. These approaches, which selectively apply teacher signals based on measured usefulness, can yield substantial performance gains, even enabling 60M-parameter models to outperform 3B-parameter LLMs. Prioritize methods that leverage complementary teacher signals, especially when a significant student-teacher capacity gap exists.
Key insights
Standard knowledge distillation often harms student performance; selective, reliability-aware methods significantly improve low-resource summarization.
Principles
- Uniform KD can harm student validation loss for over 50% of samples.
- Teacher confidence and ROUGE agreement are weak predictors of KD usefulness.
- KD is most effective with substantial teacher-student capacity gaps.
Method
CHAD uses gradient alignment to train a gate for per-sample KD weighting. EWAD+CPDP combines token-level confidence gating with a cross-vocabulary geometric constraint from a second teacher.
In practice
- Measure per-sample KD usefulness via gradient alignment.
- Modulate token-level KD based on teacher's top-1 probability.
Topics
- Knowledge Distillation
- Low-Resource NLP
- Abstractive Summarization
- Gradient Alignment
- Multi-Teacher Models
- Model Compression
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 cs.CL updates on arXiv.org.