Exploring the Limits of Model Compression in LLMs: A Knowledge Distillation Study on QA Tasks
By: Joyeeta Datta , Niclas Doll , Qusai Ramadan and more
Potential Business Impact:
Makes smart computer programs smaller and faster.
Large Language Models (LLMs) have demonstrated outstanding performance across a range of NLP tasks, however, their computational demands hinder their deployment in real-world, resource-constrained environments. This work investigates the extent to which LLMs can be compressed using Knowledge Distillation (KD) while maintaining strong performance on Question Answering (QA) tasks. We evaluate student models distilled from the Pythia and Qwen2.5 families on two QA benchmarks, SQuAD and MLQA, under zero-shot and one-shot prompting conditions. Results show that student models retain over 90% of their teacher models' performance while reducing parameter counts by up to 57.1%. Furthermore, one-shot prompting yields additional performance gains over zero-shot setups for both model families. These findings underscore the trade-off between model efficiency and task performance, demonstrating that KD, combined with minimal prompting, can yield compact yet capable QA systems suitable for resource-constrained applications.
Similar Papers
Knowledge Distillation of Domain-adapted LLMs for Question-Answering in Telecom
Computation and Language
Makes big AI models smaller and smarter.
An Empirical Study of Knowledge Distillation for Code Understanding Tasks
Software Engineering
Makes smart computer code understand faster.
Does Knowledge Distillation Matter for Large Language Model based Bundle Generation?
Computation and Language
Makes smart computer programs smaller and faster.