Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Shirui Chen, Cole Harrison, Ying-Chun Lee, Angela Jin Yang, Zhongzheng Ren, Lillian J. Ratliff, Jiafei Duan, Dieter Fox, Ranjay Krishna• 2026

Related benchmarks

TaskDatasetResultRank
Reward Prediction10-task benchmark S1 classic
Demo L (MSE)0.09
8
Reward rollout alignment10-task benchmark T1: Folding Shorts
Rollout ρ-0.222
8
Value EstimationSuboptimal-Value-Bench AgileX
Hesitation RMSE0.36
7
Value EstimationSuboptimal-Value-Bench ARX
Hesitation RMSE0.24
7
Value EstimationSuboptimal-RoboSuite Value-Bench
Hesitation RMSE0.33
7
Value-Order CorrelationSelf-collected 3 embodiments, expert demonstrations
Value-Order Correlation (VOC)0.58
7
Reward PredictionS2 10-task benchmark unconventional
Demo L MSE0.12
7
Reward Prediction10-task benchmark Overall
Demo L (MSE)0.107
7
Reward rollout alignment10-task benchmark T2: Cleaning Whiteboard
Rollout ρ-0.444
7
Value-Order CorrelationOXE expert demonstrations
Value-Order Correlation0.19
7
Showing 10 of 36 rows

Other info

GitHub

Follow for update