TOPReward: Token Probabilities as Hidden Zero-Shot Rewards for Robotics

Shirui Chen, Cole Harrison, Ying-Chun Lee, Angela Jin Yang, Zhongzheng Ren +4 more
2/22/2026
cs.ROcs.AIcs.LG

No official code found. Best implementation: TOPReward/TOPReward (113 stars, MIT, last commit 2026-03-06).

Abstract

While Vision-Language-Action (VLA) models have seen rapid progress in pretraining, their advancement in Reinforcement Learning (RL) remains hampered by low sample efficiency and sparse rewards in real-world settings. Developing generalizable process reward models is essential for providing the fine-grained feedback necessary to bridge this gap, yet existing temporal value functions often fail to generalize beyond their training domains. We introduce TOPReward, a novel, probabilistically grounded temporal value function that leverages the latent world knowledge of pretrained video Vision-Language Models (VLMs) to estimate robotic task progress. Unlike prior methods that prompt VLMs to directly output progress values, which are prone to numerical misrepresentation, TOPReward extracts task progress directly from the VLM's internal token logits. In zero-shot evaluations across 130+ distinct real-world tasks and multiple robot platforms (e.g., Franka, YAM, SO-100/101), TOPReward achieves 0.947 mean Value-Order Correlation (VOC) on Qwen3-VL, dramatically outperforming the state-of-the-art GVL baseline which achieves near-zero correlation on the same open-source model. We further demonstrate that TOPReward serves as a versatile tool for downstream applications, including success detection and reward-aligned behavior cloning.

View on arXivView PDF

Code Implementations(1)

MIT

TOPReward: Token Probabilities as Hidden Zero-Shot Rewards for Robotics

11316Shell, PythonFeb 21, 20267 months agoMIT

Similar Papers

Cite this paper

@article{chen2026topreward,
  title  = {TOPReward: Token Probabilities as Hidden Zero-Shot Rewards for Robotics},
  author = {Shirui Chen and Cole Harrison and Ying-Chun Lee and Angela Jin Yang and Zhongzheng Ren and Lillian J. Ratliff and Jiafei Duan and Dieter Fox and Ranjay Krishna},
  year   = {2026},
  eprint = {2602.19313},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.19313v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2602.19313.svg)](https://codeofpaper.com/paper/2602.19313)

Discussion