pith. sign in

Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation

1 Pith paper cite this work. Polarity classification is still indexing.

1 Pith paper citing it
abstract

On-policy distillation (OPD) has emerged as an efficient post-training paradigm for large language models. However, existing studies largely attribute this advantage to denser and more stable supervision, while the parameter-level mechanisms underlying OPD's efficiency remain poorly understood. In this work, we argue that OPD's efficiency stems from a form of ``foresight'': it establishes a stable update trajectory toward the final model early in training. This foresight manifests in two aspects. First, at the \textbf{Module-Allocation Level}, OPD identifies regions with low marginal utility and concentrates updates on modules that are more critical to reasoning. Second, at the \textbf{Update-Direction Level}, OPD exhibits stronger low-rank concentration, with its dominant subspaces aligning closely with the final update subspace early in training. Building on these findings, we propose \textbf{EffOPD}, a plug-and-play acceleration method that speeds up OPD by adaptively selecting an extrapolation step size and moving along the current update direction. EffOPD requires no additional trainable modules or complex hyperparameter tuning, and achieves an average training acceleration of $3\times$ while maintaining comparable final performance. Overall, our findings provide a parameter-dynamics perspective for understanding the efficiency of OPD and offer practical insights for designing more efficient post-training methods for large language models.

fields

cs.CL 1

years

2026 1

verdicts

UNVERDICTED 1

representative citing papers

Are Full Rollouts Necessary for On-Policy Distillation?

cs.CL · 2026-05-29 · unverdicted · novelty 5.0

Truncated and progressively lengthening rollouts in on-policy distillation match full-rollout performance on mathematical reasoning while using as little as 10% of the horizon and improving efficiency up to 3x.

citing papers explorer

Showing 1 of 1 citing paper.

  • Are Full Rollouts Necessary for On-Policy Distillation? cs.CL · 2026-05-29 · unverdicted · none · ref 1 · internal anchor

    Truncated and progressively lengthening rollouts in on-policy distillation match full-rollout performance on mathematical reasoning while using as little as 10% of the horizon and improving efficiency up to 3x.