When Does Knowledge Distillation Hurt? Reliability-Aware Distillation for Low-Resource Language Summarization

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Natural Language Processing · Depth: Expert, quick

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

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

Topics

Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.