Fugu-MT: arxivの論文翻訳

What Does Privileged Information Add to On-Policy Self-Distillation?

Authors XiuYu Zhang, Wei Chow, Junfeng Fang, Zhenkai Liang, Tat-Seng Chua
Affiliations National University of Singapore
Categories Method / Self-Distillation / On-policy distillation with privileged information, Evaluation / Mathematical Reasoning Benchmarks / Reusable math problem suite, Application / Language Model Improvement / Teacher-guided rollout supervision
License CC BY 4.0

Abstract Overview

This paper investigates the degree to which privileged information (PI) contributes to on-policy self-distillation (OPSD) beyond the baseline gains of cross-mode distillation itself. To isolate these effects, the authors develop AMPLE-Math, a suite of 5,319 mathematical problems featuring six answer-matched reasoning views ranging from Answer Only to Full Trace. Using a frozen, thinking-enabled teacher to supervise student rollouts, they evaluate student performance across checkpoints in Qwen3-1.7B and SmolLM3-3B. The findings reveal that substantial gains occur even in reference-free distillation, with privileged references offering varying and model-dependent additional benefits. Furthermore, teacher-student dynamics demonstrate that the supervision's effectiveness depends heavily on student response trajectories and training duration rather than token-level loss tweaks alone.

Novelty

The study introduces a rigorous experimental framework to disentangle baseline cross-mode distillation benefits from the incremental value of privileged teacher references in OPSD. It also contributes AMPLE-Math, a dataset providing six audited, answer-matched reasoning representations per problem to evaluate reasoning density while holding ground-truth answers constant.

Results

In Qwen3-1.7B, reference-free cross-mode distillation captures most of the performance gain under thinking-enabled evaluation, with Clean Solution offering a modest additional advantage of 1.30 points at step 100. In SmolLM3-3B, Full Trace adds 2.0 points over the reference-free baseline at step 50, though performance across configurations deteriorates below the base model by step 100. Across both model families, replacing direct-response training rollouts with thinking-enabled rollouts reverses gains into performance losses under an identical evaluation setup.

Key Points

  1. AMPLE-Math provides 5,319 math problems with six answer-matched reasoning views to isolate the impact of reference reasoning format on teacher supervision.
  2. A major portion of OPSD performance gains stems from cross-mode distillation between direct-response rollouts and thinking-enabled teacher scoring rather than access to full solution traces.
  3. Switching student rollouts from direct-response to thinking-enabled trajectories reverses distillation gains, and token-level loss interventions show that apparent loss-coverage benefits are largely driven by early stopping.

References

This page was created using generative AI such as GPT-5, Claude Opus 4, Gemini 3, Gemini 3.1 Flash Image, and their higher-end successor versions. No guarantee can be made regarding its contents.