When Teachers Mislead: Spurious-Signal-Aware On-Policy Distillation
Abstract
On-Policy distillation (OPD) transfers teacher capabilities by supervising student-sampled trajectories with dense token-level teacher signals. Recent selective OPD methods improve this process by prioritizing signals that are confident, informative, or learnable. However, the assumptions overlook a fundamental failure mode of language models: their token-level judgments can be driven by input-agnostic language priors, formatting conventions, or stereotyped reasoning templates rather than task-specific evidence. We refer to such optimization-relevant but weakly input-grounded supervision as spurious signals in OPD, which may produce large gradients while contributing little task-improving direction. To mitigate this issue, we propose SA-OPD, a Spurious-Signal-Aware On-Policy Distillation framework that identifies and filters misleading token-level supervision based on input-groundedness and optimization impact. SA-OPD introduces a lightweight input-groundedness proxy estimating whether a token-level distillation signal truly depends on the input. It then filters only tokens that simultaneously exhibit low input-groundedness and extreme distillation divergence, thereby removing high-impact spurious updates and achieving fine-grained OPD optimization. Extensive experiments on both large language model (LLM) and vision-language model (VLM) settings demonstrate that SA-OPD consistently outperforms Vanilla OPD and competitive selective methods. These results establish input-groundedness as a key dimension for OPD supervision selection and offer a simple, effective strategy for mitigating spurious updates.
Community
SA-OPD addresses a previously overlooked failure mode in on-policy distillation: teacher signals can appear confident, informative, and learnable while being driven by input-agnostic language priors, formatting conventions, or stereotyped reasoning templates, producing large gradients with little task-improving direction. By comparing token-level teacher–student divergence under the original input and a residual no-prompt context, SA-OPD provides a lightweight proxy for input-groundedness—without requiring external verification labels or auxiliary judges. It then filters only tokens that combine low input-groundedness with high absolute distillation divergence, targeting the high-impact portion of dense OPD supervision most likely to induce harmful updates. Across LLM and VLM distillation settings, SA-OPD consistently improves over Vanilla OPD and strong selective OPD baselines on mathematical reasoning, visual understanding, and visual reasoning benchmarks, offering a practical direction for more reliable, input-grounded distillation signal selection in on-policy distillation.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Blockwise Policy-Drift Gating for On-Policy Distillation (2026)
- Reward-Gated On-Policy Distillation (2026)
- Distill What the Student Can See: Fisher-Projected On-Policy Distillation for Vision-Language Models (2026)
- Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rollouts (2026)
- Distill Where You Fail: Recovering Learning Signals of Negative RL-Groups from Adaptive Teacher Guidance (2026)
- On the Position Bias of On-Policy Distillation (2026)
- Trace-Based On-Policy Distillation for Masked Diffusion Language Models (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2608.03632 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper