Pedagogically-Inspired Data Synthesis for Language Model Knowledge Distillation

TL;DR

This paper introduces a pedagogically-inspired framework for knowledge distillation from large to smaller language models, systematically improving transfer efficiency by mimicking educational principles like curriculum learning and mastery.

Contribution

It proposes a three-stage IOA pipeline that enhances knowledge transfer by identifying deficiencies, organizing curricula, and adapting representations based on educational theories.

Findings

Achieves 94.7% of teacher performance with less than 1/10th parameters.

Significantly improves complex reasoning tasks, with 19.2% on MATH and 22.3% on HumanEval.

Outperforms baseline distillation methods across multiple benchmarks.

Abstract

Knowledge distillation from Large Language Models (LLMs) to smaller models has emerged as a critical technique for deploying efficient AI systems. However, current methods for distillation via synthetic data lack pedagogical awareness, treating knowledge transfer as a one-off data synthesis and training task rather than a systematic learning process. In this paper, we propose a novel pedagogically-inspired framework for LLM knowledge distillation that draws from fundamental educational principles. Our approach introduces a three-stage pipeline -- Knowledge Identifier, Organizer, and Adapter (IOA) -- that systematically identifies knowledge deficiencies in student models, organizes knowledge delivery through progressive curricula, and adapts representations to match the cognitive capacity of student models. We integrate Bloom's Mastery Learning Principles and Vygotsky's Zone of Proximal Development to create a dynamic distillation process where student models approach teacher model's performance on prerequisite knowledge before advancing, and new knowledge is introduced with controlled, gradual difficulty increments. Extensive experiments using LLaMA-3.1/3.2 and Qwen2.5 as student models demonstrate that IOA achieves significant improvements over baseline distillation methods, with student models retaining 94.7% of teacher performance on DollyEval while using less than 1/10th of the parameters. Our framework particularly excels in complex reasoning tasks, showing 19.2% improvement on MATH and 22.3% on HumanEval compared with state-of-the-art baselines.