正在加载视频...
视频加载失败
Excited to introduce SOLE-R1, a video-language reasoning model for zero-shot reward prediction for robot manipulation tasks! SOLE-R1 reasoning can serve as the SOLE signal for learning new tasks (completely from scratch) through online RL - i.e., robots start with random actions and learn previously unseen tasks guided only by... show more
18,957 次观看 • 3 个月前 •via X (Twitter)
7 条评论

(2/6) Given a natural-language goal and a video stream of observations, SOLE-R1 produces (i) per-timestep chain-of-thought reasoning about what has changed since the last timestep and (ii) a dense scalar progress estimate used as a reward for online RL. Attached video shows examples of SOLE-R1 zero-shot reward prediction for the RoboSuite pick cube task (both successful and failed trajectories).

(3/6) SOLE-R1 outperforms strong baselines (e.g., Robometer, RoboReward, TOPReward, GPT-5, Gemini-3-Pro) on zero-shot reward prediction for online RL. Baselines show greater vulnerability to reward hacking - i.e., the policy tricks the reward model into predicting high rewards without actually doing the task. Example videos below of baselines frequently over-estimating progress on failed trajectories.

(4/6) Training data: We build the SOLE-R1 training data in two stages: (1) foundational reasoning over space (single-image + depth) and time (multi-image/video), and (2) robot-video spatiotemporal reasoning specialized for dense progress estimation. For (1), we carefully curate a diverse collection of general spatial and multi-frame temporal reasoning data (e.g., from SSR-CoT, SpatialVLM, Spot-the-diff, Embodied CoT, RoboVQA, Robo2VLM-Reasoning) to serve as a foundational layer of our training mixture. For (2), we generate over 1 million chain-of-thought reasoning and ground-truth progress prediction examples from more than 40,000 real-world and simulated videos containing both successful and failed trajectories. Training recipe: Given this data, SOLE-R1 is trained with a two-stage hybrid recipe: (i) SFT to develop high-quality spatiotemporal CoT reasoning across the full training dataset, (ii) RLVR (GRPO) to further develop accurate progress prediction, boosted by the strong CoT reasoning developed during SFT

(5/6) Evaluation setting: We evaluate whether SOLE-R1 can serve as the SOLE supervision signal for learning manipulation skills from scratch via online RL. We run experiments across 4 sim benchmark suites (LIBERO, ManiSkill, Meta-World, and RoboSuite) and in a real-world tabletop manipulation setting with a Franka arm. Across all settings, we evaluate a total of 40 tasks, spanning pick-and-place, articulation, button/lever/knob interactions, and mobile manipulation. Results: SOLE-R1 achieves ≥50% success rate on 24 tasks, substantially outperforming all baselines. The strongest baselines include GPT-5 and Gemini, but they reach 50% success on only 7 and 5 tasks, respectively. The non-reasoning models achieve near-zero success on all tasks, with the exception of Meta-World tasks, where Robometer, RoboReward, and ReWiND achieve above 40% success rate on 4 tasks.

(6/6) 🌐 Website: 📄 Paper: 👩🏻💻 Code: Thank you to my co-authors: @thomas_weng, Karl Schmeckpeper, @_ericrosen, Stephen Hart And special thanks to @BizaOndrej for supervising this work from the very start! More to come soon!

Really interesting work. How well does SOLE-R1 handle failure cases where the visual signal is ambiguous - does the reward model ever confidently guide the robot in the wrong direction?

Open source is great, mind if I test on my robot?
