TOPReward: Token Probabilities as Hidden Zero-Shot Rewards for Robotics
About
General-purpose robot learning requires dense, instruction-conditioned feedback that can distinguish meaningful task progress from stalled, failed, or partially completed behavior. Yet obtaining such feedback at scale remains difficult, since existing approaches often rely on manual progress annotations, task-specific demonstrations, or reward models trained on curated robot datasets. We introduce TOPReward, a training-free progress reward method that probes pretrained Video-Language Models (VLMs) through their internal token probabilities rather than asking them to generate numerical progress values. Given a video prefix and a language instruction, TOPReward measures the model's likelihood that the instructed task has been completed, converting latent video-language understanding into a dense reward signal without task-specific reward-model training or manually annotated progress labels. We evaluate TOPReward on ManiRewardBench, our real-world manipulation benchmark spanning 130 unique tasks and four robot platforms, as well as on Open X-Embodiment datasets. Across these settings, TOPReward substantially outperforms prior training-free VLM reward methods on open-source models and is competitive with a trained reward-model baseline on progress-estimation metrics, while requiring no reward-model training. Additional analyses show that the reward is sensitive to the specified instruction and is not explained by time index alone. Finally, TOPReward supports downstream applications including success detection and offline reward-weighted behavior cloning.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Reward Prediction | 10-task benchmark S1 classic | Demo L (MSE)0.09 | 8 | |
| Reward rollout alignment | 10-task benchmark T1: Folding Shorts | Rollout ρ-0.222 | 8 | |
| Value Estimation | Suboptimal-Value-Bench AgileX | Hesitation RMSE0.36 | 7 | |
| Value Estimation | Suboptimal-Value-Bench ARX | Hesitation RMSE0.24 | 7 | |
| Value Estimation | Suboptimal-RoboSuite Value-Bench | Hesitation RMSE0.33 | 7 | |
| Value-Order Correlation | Self-collected 3 embodiments, expert demonstrations | Value-Order Correlation (VOC)0.58 | 7 | |
| Reward Prediction | S2 10-task benchmark unconventional | Demo L MSE0.12 | 7 | |
| Reward Prediction | 10-task benchmark Overall | Demo L (MSE)0.107 | 7 | |
| Reward rollout alignment | 10-task benchmark T2: Cleaning Whiteboard | Rollout ρ-0.444 | 7 | |
| Value-Order Correlation | OXE expert demonstrations | Value-Order Correlation0.19 | 7 |