Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Progressive distillation induces an implicit curriculum

About

Knowledge distillation leverages a teacher model to improve the training of a student model. A persistent challenge is that a better teacher does not always yield a better student, to which a common mitigation is to use additional supervision from several ``intermediate'' teachers. One empirically validated variant of this principle is progressive distillation, where the student learns from successive intermediate checkpoints of the teacher. Using sparse parity as a sandbox, we identify an implicit curriculum as one mechanism through which progressive distillation accelerates the student's learning. This curriculum is available only through the intermediate checkpoints but not the final converged one, and imparts both empirical acceleration and a provable sample complexity benefit to the student. We then extend our investigation to Transformers trained on probabilistic context-free grammars (PCFGs) and real-world pre-training datasets (Wikipedia and Books). Through probing the teacher model, we identify an analogous implicit curriculum where the model progressively learns features that capture longer context. Our theoretical and empirical findings on sparse parity, complemented by empirical observations on more complex tasks, highlight the benefit of progressive distillation via implicit curriculum across setups.

Abhishek Panigrahi, Bingbin Liu, Sadhika Malladi, Andrej Risteski, Surbhi Goel• 2024

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningStrategyQA (test)
Accuracy54.3
119
Mathematical ReasoningGSM8K
Accuracy46
95
Mathematical ReasoningMATH
Accuracy55
44
Science Question AnsweringARC Challenge (test)
Accuracy46.3
42
Showing 4 of 4 rows

Other info

Follow for update