atlas

Research map / Reinforcement learning / Visual and representation RL

RT-Trajectory: Robotic Task Generalization via Hindsight Trajectory Sketches

ICLR 2024 · Reinforcement learning › Visual and representation RL · approach: Vision-based robot RL · 5 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes RT-Trajectory, which conditions an RT-1-style transformer policy on 2D (and 2.5D) hindsight trajectory sketches so robots can generalize to new tasks specified by drawings, human videos, LLM code, or image generators.

Problem

Robot policies conditioned on language or goal images fail to generalize to new tasks whose motions differ from training tasks, even when the arm trajectory resembles seen motions.

Approach

Represent the task as a coarse 2D end-effector trajectory sketch projected into the camera view (with time encoded as color, optional height, and grasp/release markers), automatically extracted from demonstrations via hindsight labeling, and use it as the policy conditioning signal.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use trajectory sketches instead of language conditioningRT-Trajectory (2D) vs RT-1 · Unseen tasks (overall)success rate: 17% -> 50%Helped
Use trajectory sketches instead of goal-image conditioningRT-Trajectory (2.5D) vs RT-1-Goal · Unseen tasks (overall)success rate: 26% -> 67%Helped
Add height information to the 2D trajectory (ablation)RT-Trajectory · Unseen tasks (overall)success rate: 50% -> 67%Helped
Add height information to the 2D trajectory (ablation)RT-Trajectory · Unseen taskssuccess rate: Place Fruit: 75% -> 75%; Upright and Move: 33% -> 50%; Move within Drawer: 67% -> 100%; Restock Drawer: 92% -> 67%; PickMixed
Condition on a human-video-derived trajectory sketch instead of executing it with an IK plannerRT-Trajectory (2D) · Human video trajectories: Picksuccess rate: 42% -> 94%Helped
Condition on a human-video-derived trajectory sketch instead of executing it with an IK plannerRT-Trajectory (2.5D) · Human video trajectories: Fold Towelsuccess rate: 25% -> 75%Helped
Add height information to the 2D trajectory (ablation)RT-Trajectory · Human video trajectories: Picksuccess rate: 94% -> 100%Helped
Condition on an LLM-generated trajectory sketch instead of executing it with an IK plannerRT-Trajectory (2D) · LLM-generated trajectories: Picksuccess rate: 83% -> 89%Helped
Condition on an LLM-generated trajectory sketch instead of executing it with an IK plannerRT-Trajectory (2D) · LLM-generated trajectories: Open Drawersuccess rate: 71% -> 60%Hurt
Add height information to the 2D trajectory (ablation)RT-Trajectory · LLM-generated trajectoriessuccess rate: Pick: 89% -> 89%; Open Drawer: 60% -> 60%No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: RT-1 demonstration dataset first-person human demonstration videos (22 videos)

Benchmarks: 7 unseen real-world manipulation skills (Place Fruit, Upright and Move, Move within Drawer, Restock Drawer, Pick from Ch

Metrics: success rate

Baselines: RT-1 RT-2 RT-1-Goal IK Planner

Builds on: RT-1 RT-2 Behavior Cloning Code as Policies

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy