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

· Source: Takara TLDR - Daily AI Papers · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Natural Language Processing · Depth: Expert, quick

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

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

Topics

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.