Search papers, labs, and topics across Lattice.
This paper introduces Prism-GRPO, an enhancement to the GRPO method for optimizing vision-language-action (VLA) policies by incorporating a weighted trajectory-level execution-quality score to better utilize rollout data. By splitting same-outcome groups into a quality spectrum, the method effectively recovers training signal and reduces the number of rollouts needed to achieve target success rates by up to 56%. The results demonstrate improved success and quality across various RoboTwin tasks, while also mitigating reward-hacking behaviors during deployment on real robots.
Splitting same-outcome groups by execution quality can cut rollout requirements by over half while enhancing policy performance in complex VLA tasks.
GRPO is increasingly used for reinforcement learning of vision-language-action (VLA) policies because, unlike PPO, it does not require training a critic. This simplification comes with a sampling cost: group-relative advantages require multiple rollouts from each scene. Under binary success rewards, groups whose rollouts all succeed or all fail have zero advantage and are discarded by dynamic sampling. These groups are especially common early in training, when most rollouts fail, wasting much of the expensive robotic rollout budget. We introduce Prism-GRPO, which augments binary outcome reward with a weighted trajectory-level execution-quality score. By splitting same-outcome groups into a quality spectrum, Prism-GRPO recovers training signal while ensuring that every success still outranks every failure. Quality scores can be derived from simulator contacts, executed actions, or visual observations, avoiding task-specific progress rewards. We prove that Prism-GRPO never increases the probability that a sampled group is discarded for having zero advantages, and derive a gradient-alignment condition under which its combined update remains a local ascent direction for task success. Across four RoboTwin tasks spanning different horizons and coordination patterns, Prism-GRPO improves success and quality at matched rollout budgets and reaches target success rates with up to 56% fewer rollouts. It also suppresses a reward-hacking shortcut, with the cleaner behavior transferring under direct deployment to a real robot. Through ablations, we show consistent gains across contact-, smoothness-, and VLM-derived quality signals.