Search papers, labs, and topics across Lattice.
This paper introduces SA-OPD, a Spurious-Signal-Aware On-Policy Distillation framework that addresses the issue of misleading token-level supervision in On-Policy Distillation (OPD). By filtering out spurious signals that are driven by input-agnostic language priors, the method enhances the optimization process for both large language models (LLMs) and vision-language models (VLMs). The results show that SA-OPD significantly outperforms traditional OPD and selective methods, highlighting the importance of input-groundedness in optimizing distillation processes.
Filtering out misleading signals can boost OPD performance by leveraging input-groundedness, leading to more effective model training.
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.
