Track manipulation, locomotion and embodied systems.
Search papers, labs, and topics across Lattice.
Investigating on-policy distillation for multi-turn visual agents, the authors reveal that performance gains stem almost entirely from counterfactual rollbacks at the "pivot step"—the first unrecoverable action in an episode. To bypass the severe computational overhead of physical environment rollbacks, PIVOT internalizes this recovery process directly into token-level parameter updates using a unified Analyzer-Teacher-Student architecture. The approach resolves zero-gradient silence in GRPO, boosting agent accuracy to 90% on Qwen2.5-VL-3B (+8% over SFT+GRPO) and 92% on Qwen3-VL-2B (+12%) across five multi-turn embodied control and reasoning benchmarks.
The real driver of on-policy distillation gains in multi-turn agents is counterfactual rollback at the first fatal mistake—an insight leveraged here to achieve double-digit RLVR gains by internalizing rollback dynamics directly into gradient updates without resetting the environment.
Reinforcement learning with verifiable rewards (RLVR) via Group-Relative Policy Optimization (GRPO) is widely used for multi-turn VLM agent training, yet it suffers from zero-gradient silence on uniform failures and coarse episode-level credit assignment. While On-Policy Distillation (OPD) and On-Policy Self-Distillation (OPSD) mitigate sparse rewards using hindsight information, their underlying mechanisms remain poorly understood. Through controlled counterfactual rollback probes across five multi-turn VLM agent benchmarks, we reveal that performance gains in OPSD/OPD are largely driven by physical state rollback at the pivot step, defined as the first unrecoverable action without remaining step budget. However, physical state rollbacks are computationally prohibitive and infeasible in real-world environments. To bridge this gap, we present Pivot-Aware Internalized Visual On-Policy Training (PIVOT), an RL framework that internalizes pivot localization and state restoration directly into token-level parameter updates, eliminating environment rollbacks during RL training and additional skill hints at test time. PIVOT unifies three functional roles within a single architecture: a failure Analyzer non-invasively localizes the pivot step and diagnoses failure modes from visual trajectory collages and action logs; a detached Teacher re-scores failed tokens under this privileged diagnostic context; and a Student optimizes joint GRPO and confidence-gated OPD objectives. At test time, both Teacher and Analyzer branches are stripped. Evaluated on five multi-turn VLM agent tasks across cognitive grid puzzles, 3D embodied control and navigation, and generative reasoning, PIVOT achieves 0.90 overall accuracy on Qwen2.5-VL-3B (+8% over SFT+GRPO baseline and +5% over previous SOTA) and scales to 0.92 on Qwen3-VL-2B (+12% over SFT+GRPO baseline).