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

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Natural Language Processing · Depth: Expert, extended

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

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

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 cs.CL updates on arXiv.org.