Learning from Language Feedback via Variational Policy Distillation
Summary
Variational Policy Distillation (VPD) formalizes learning from language feedback as a variational EM problem, co-training teacher and student networks to improve policy learning in reinforcement learning from verifiable rewards. It shows consistent improvements over baselines on code generation and scientific reasoning tasks.
View Cached Full Text
Cached at: 05/21/26, 10:12 PM
Paper page - Learning from Language Feedback via Variational Policy Distillation
Source: https://huggingface.co/papers/2605.15113 Variational Policy Distillation (VPD) addresses a key limitation of reinforcement learning from verifiable rewards (RLVR): the binary reward signal discards all information from near-miss failures. A coding solution that fails 1 test out of 50 gets the same reward as random noise, even though the compiler error tells you exactly what went wrong.
VPD formalizes learning from language feedback (compiler errors, LLM critiques, self-corrections) as a variational EM problem. Unlike prior self-distillation methods that treat the feedback-conditioned teacher as a frozen function, VPD co-trains the teacher and student in an alternating loop:
- E-step: refine the teacher’s ability to interpret feedback via preference optimization
- M-step: distill the improved teacher into the student on its own rollouts
Both share a single network, so there’s zero additional memory overhead.
We evaluate on 3 model families (Qwen3-4B, Qwen3-8B, Llama-3.1-8B) across code generation (LiveCodeBench) and scientific reasoning (SciKnowEval). VPD consistently improves over GRPO and self-distillation baselines, with notably more stable training dynamics. We also characterize where the approach has limitations — on strict mathematical reasoning where error feedback is less informative, standard RL remains stronger.
Happy to discuss — feedback welcome!
Similar Articles
Teaching the Way, Not the Answer: Privileged Tutoring Distillation for Multimodal Policy Optimization
This paper proposes PTD-PO, a privileged tutoring distillation framework that provides dense token-level supervision for reinforcement learning with verifiable rewards in multimodal reasoning tasks, without exposing the answer. It uses structured hints and a Top-K JS divergence objective to stabilize training, consistently outperforming existing methods on 2B-8B LVLMs.
Self-Boosting Vision-Language Models with Noisy Student On-Policy Self-Distillation
Proposes NOPD, a self-distillation method that improves vision-language models without external supervision by leveraging prediction discrepancies between clean and corrupted inputs. Achieves significant gains on visual reasoning tasks, matching or exceeding RL and distillation from external models.
KL for a KL: On-Policy Distillation with Control Variate Baseline
Proposes vOPD, which stabilizes on-policy distillation for LLMs by introducing a control variate baseline from reinforcement learning, achieving performance comparable to expensive full-vocabulary methods at lower computational cost.
The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes
This paper presents a comprehensive empirical study on on-policy distillation for large language models, identifying failure mechanisms like distribution mismatch and optimization instability, and proposing fixes such as stop-gradient objectives and RLVR-adapted teachers.
DOPD: Dual On-policy Distillation
DOPD proposes a dual on-policy distillation paradigm that dynamically routes token-level supervision between privileged teacher and student policies based on advantage gaps and probabilities, addressing privilege illusion and improving capability transfer in LLMs and VLMs.