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

PEEK: Guiding and Minimal Image Representations for Zero-Shot Generalization of Robot Manipulation Policies

About

Robotic manipulation policies often fail to generalize because they must simultaneously learn where to attend, what actions to take, and how to execute them. We argue that high-level reasoning about where and what can be offloaded to vision-language models (VLMs), leaving policies to specialize in how to act. We present PEEK (Policy-agnostic Extraction of Essential Keypoints), which fine-tunes VLMs to predict a unified point-based intermediate representation: 1. end-effector paths specifying what actions to take, and 2. task-relevant masks indicating where to focus. These annotations are directly overlaid onto robot observations, making the representation policy-agnostic and transferable across architectures. To enable scalable training, we introduce an automatic annotation pipeline, generating labeled data across 20+ robot datasets spanning 9 embodiments. In real-world evaluations, PEEK consistently boosts zero-shot generalization, including a 41.4x real-world improvement for a 3D policy trained only in simulation, and 2-3.5x gains for both large VLAs and small manipulation policies. By letting VLMs absorb semantic and visual complexity, PEEK equips manipulation policies with the minimal cues they need--where, what, and how. Website at https://peek-robot.github.io/.

Jesse Zhang, Marius Memmel, Kevin Kim, Dieter Fox, Jesse Thomason, Fabio Ramos, Erdem B{\i}y{\i}k, Abhishek Gupta, Anqi Li• 2025

Related benchmarks

TaskDatasetResultRank
Spatial ReasoningCVBench--
15
Visual Trace GenerationVABench-V
RMSE188.9
13
4D TaskST-Human-Planning
Accuracy88
7
Next-Step Trajectory PlanningST-Human-Planning
RMSE178.3
7
Previous Instruction PredictionST-Human-Planning
BERTScore0.8
7
Task Progress EstimationST-Human-Planning
Step Accuracy14
7
3D TaskST-Human-Spatial
Accuracy50
7
Next Instruction PredictionST-Human-Planning
BERTScore0.77
7
Spatial ReasoningST-Human-Spatial
Accuracy50
7
Trajectory GenerationRLBench
RMSE81.59
7
Showing 10 of 28 rows

Other info

Follow for update