paper-with-me

홈 › Papers

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

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

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.

📄 PDF Abstract BibTeX arXiv:2602.19313

Code (0)

등록된 구현이 없습니다.

Tasks

Reinforcement Learning

Similar Papers 제목 키워드 기반

Beyond the Next Token: Towards Prompt-Robust Zero-Shot Classification via Efficient Multi-Token Prediction

2025-04-04 · Junlang Qian, Zixiao Zhu, Hanzhang Zhou, Zijian Feng 외

Zero-shot text classification typically relies on prompt engineering, but the inherent prompt brittleness of large language models undermines its reliability. Minor changes in prompt can cause significant discrepancies i…

AttributeLanguage ModelingLanguage ModellingPrompt Engineering+5

Unleashing the Multilingual Encoder Potential: Boosting Zero-Shot Performance via Probability Calibration

2023-10-08 · Ercong Nie, Helmut Schmid, Hinrich Schütze

Pretrained multilingual encoder models can directly perform zero-shot multilingual tasks or linguistic probing by reformulating the input examples into cloze-style prompts. This is accomplished by predicting the probabil…

Position

Prompt-Induced Score Variance in Zero-Shot Binary Vision-Language Safety Classification

2026-05-01 · Charles Weng, Dingwen Li, Alexander Martin arxiv

Single-prompt first-token probabilities from zero-shot vision-language model (VLM) safety classifiers are treated as decision scores, but we show they are unreliable under semantically equivalent prompt reformulation: ev…

Correcting Suppressed Log-Probabilities in Language Models with Post-Transformer Adapters

2026-03-25 · Bryan Sanchez arxiv

Alignment-tuned language models frequently suppress factual log-probabilities on politically sensitive topics despite retaining the knowledge in their hidden representations. We show that a 786K-parameter (approximately …

Token-based Decision Criteria Are Suboptimal in In-context Learning

2024-06-24 · Hakaze Cho, Yoshihiro Sakai, Mariko Kato, Kenshiro Tanaka 외

In-Context Learning (ICL) typically utilizes classification criteria from output probabilities of manually selected label tokens. However, we argue that such token-based classification criteria lead to suboptimal decisio…

ClassificationIn-Context Learning