Researchers introduce Relay On-Policy Distillation (Relay-OPD) to address prefix failure in standard on-policy distillation, where student errors lead to unreliable supervision. The method uses a teacher-student continuation asymmetry as a trigger for the teacher to briefly take over and redirect the trajectory before the student resumes.

  • Utilizes a Qwen3-4B-Instruct-2507 teacher with Qwen3-0.6B/1.7B-Non-Thinking students on eight mathematical reasoning benchmarks.
  • Achieves the best or second-best results on every benchmark, outperforming standard OPD by +5.73% and FastOPD by +1.49% on average for the 1.7B model.
  • Consistent gains are observed at the 0.6B scale, while training trajectory length is reduced by over 50%.

This approach concentrates intervention on critical early positions via a limited relay budget, limiting departure from the student policy while significantly improving efficiency and accuracy.