PACED: Distillation and On-Policy Self-Distillation at the Frontier of Student Competence
Abstract
Standard LLM distillation treats all training problems equally -- wasting compute on problems the student has already mastered or cannot yet solve. We empirically show that this inefficiency has a precise gradient-level signature: the cross-problem gradient signal-to-noise ratio (SNR) follows a bell curve over student pass rate, collapsing at both extremes. We propose PACED, which weights each problem by where is the student's empirical pass rate -- concentrating training on the zone of proximal development. This requires only student rollouts, no architectural changes, and no hyperparameters. We prove the Beta kernel is the leading-order optimal weight family arising from the SNR boundary-collapse structure, and is minimax-robust under misspecification (worst-case efficiency loss ). Across Qwen3, Qwen2.5, and Llama-3 families, PACED sets a new state of the art in our experimental setting on MATH-500, AIME~2024, and AIME~2025, improving over unweighted distillation by up to and over the strong AKL baseline by up to , while reducing forgetting to and in distillation and self-distillation. A two-stage forward-then-reverse KL schedule pushes gains further to over standard forward KL on the hardest benchmark.
Keywords
Cite
@article{arxiv.2603.11178,
title = {PACED: Distillation and On-Policy Self-Distillation at the Frontier of Student Competence},
author = {Yuanda Xu and Hejian Sang and Zhengze Zhou and Ran He and Zhipeng Wang},
journal= {arXiv preprint arXiv:2603.11178},
year = {2026}
}