ZPPO让小模型通过提示词向大模型学习,避免训练漂移。
Zone of Proximal Policy Optimization: Teacher in Prompts, Not Gradients

- 用提示词引入教师答案,而非修改梯度,保持策略更新的稳定性。
- 在0.8B到9B的小模型上,相比基线提升12%-23%准确率。
- 特别适合小模型在复杂任务中突破能力瓶颈,如视觉语言理解。
知识蒸馏将大模型的知识传递给小模型,但在小模型阶段易失效:强制学生模仿大模型的输出概率会使其过度聚焦于教师最突出的模式,损害在训练集外基准上的泛化能力。强化学习虽避免了对输出概率的模仿,但当所有采样轨迹均失败(优势为零)时,直接引入教师响应会破坏在线策略假设并导致策略漂移。本文提出近端策略优化区(ZPPO),受维果茨基最近发展区理论启发,将教师置于提示中而非梯度中。针对难题,ZPPO构造两类重构提示:二选一候选提示(BCQ)将一个正确教师答案与一个错误学生答案作为匿名候选供学生区分;负向候选提示(NCQ)将学生的错误轨迹聚合为单一提示,暴露其共性缺陷。提示回放缓冲区持续重播难题,直至学生平均准确率达到50%(毕业)或因容量限制被剔除,从而在学生当前发展区内放大学习信号。在四个不同规模(0.8B-9B)的Qwen3.5模型上,以27B教师进行后训练,并在包含16个视觉语言、10个语言模型和5个视频任务的31项基准测试中评估,ZPPO显著优于传统及在线/离线策略蒸馏方法,且小模型收益最大。
原文摘要 · Abstract (English)
Knowledge distillation transfers a teacher's competence to a small student but is brittle in the small-student regime: forcing the student to imitate logits from a much larger teacher concentrates it on the teacher's sharpest modes, hurting generalization on benchmark families beyond the training corpus. Reinforcement learning (RL) avoids logit imitation by training on the student's own rollouts. However, on questions where every rollout fails-yielding zero advantage and being silently discarded-injecting a stronger teacher's response into the policy gradient breaks the on-policy assumption and induces drift. We introduce Zone of Proximal Policy Optimization (ZPPO), inspired by Vygotsky's zone of proximal development, which keeps the teacher inside the prompt rather than the policy gradient. On hard questions, ZPPO constructs two reformulated prompts: a Binary Candidate-included Question (BCQ) pairs one correct teacher response with one incorrect student response as anonymized candidates the student must discriminate, and a Negative Candidate-included Question (NCQ) aggregates the student's wrong rollouts into a single prompt to surface their shared failure modes. A prompt replay buffer recirculates each hard question until it either graduates-the student's mean rollout accuracy on it reaches half- or is FIFO-evicted under finite capacity, amplifying BCQ and NCQ inside the student's current zone of proximal development. On the Qwen3.5 family at four student scales (0.8B-9B) with a 27B teacher, post-trained as vision-language models and evaluated on a 31-benchmark suite (16 VLM, 10 LLM, 5 Video), ZPPO outperforms off/on-policy distillation and GRPO, with the largest gains at the smallest scale.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。