Source-linked AI summary
Robotic Control via Embodied Chain-of-Thought Reasoning
Michał Zawalski, William Chen, Karl Pertsch, Oier Mees, Chelsea Finn, Sergey Levine
TL;DR
Reactive robot policies can struggle to generalize to novel scenes and unfamiliar objects. ECoT trains VLAs to reason about plans and grounded scene and robot-state features before acting, increasing OpenVLA’s absolute success rate by 28% across challenging generalization tasks.
Problem
Reactive robot policies may be poorly suited to broad generalization, while CoT approaches need reasoning grounded in environment observations and robot state.
Source: §1 · pp. 1–2Method
ECoT trains VLA policies to interleave textual reasoning about sub-tasks with reasoning grounded in visual inputs and robot state before predicting actions.
Source: §1 · p. 2Results
28% absolute success-rate increase for OpenVLA across challenging generalization tasks, without additional robot training data.
Source: §1 · p. 2Takeaways & Limitations
ECoT makes policy failures more interpretable and lets humans correct behavior by modifying faulty reasoning chains through natural-language feedback.
Source: §1 · p. 3ECoT inference speed remains limiting for higher-frequency control tasks, despite runtime optimizations.
Source: §6 · p. 11
Abstract
from arXiv · showhide
A key limitation of learned robot control policies is their inability to generalize outside their training data. Recent works on vision-language-action models (VLAs) have shown that the use of large, internet pre-trained vision-language models as the backbone of learned robot policies can substantially improve their robustness and generalization ability. Yet, one of the most exciting capabilities of large vision-language models in other domains is their ability to reason iteratively through complex problems. Can that same capability be brought into robotics to allow policies to improve performance by reasoning about a given task before acting? Naive use of "chain-of-thought" (CoT) style prompting is significantly less effective with standard VLAs because of the relatively simple training examples that are available to them. Additionally, purely semantic reasoning about sub-tasks, as is common in regular CoT, is insufficient for robot policies that need to ground their reasoning in sensory observations and the robot state. To this end, we introduce Embodied Chain-of-Thought Reasoning (ECoT) for VLAs, in which we train VLAs to perform multiple steps of reasoning about plans, sub-tasks, motions, and visually grounded features like object bounding boxes and end effector positions, before predicting the robot action. We design a scalable pipeline for generating synthetic training data for ECoT on large robot datasets. We demonstrate, that ECoT increases the absolute success rate of OpenVLA, the current strongest open-source VLA policy, by 28% across challenging generalization tasks, without any additional robot training data. Additionally, ECoT makes it easier for humans to interpret a policy's failures and correct its behavior using natural language.
1 Introduction
The paper introduces Embodied Chain-of-Thought Reasoning (ECoT), which trains vision-language-action policies to reason about plans and grounded visual and robot-state features before acting. The introduction reports improved generalization for OpenVLA and says embodied reasoning makes failures easier to interpret and correct with natural language.
- Reactive policies may struggle to generalize to novel scenes or unfamiliar objects, while prompted CoT and semantic sub-task plans alone do not ensure reasoning grounded in observations and robot state.Source: §1 · pp. 1–2
- ECoT interleaves semantic sub-task plans with grounded object-box predictions and robot-state-dependent movement reasoning before action prediction, using synthetic supervision generated with detectors and language models.Source: §1 · p. 2, Figure 1 · p. 2
- 28%: ECoT increases OpenVLA’s absolute success rate across tasks involving new objects, scenes, viewpoints, and instructions, without additional robot training data [7].Source: §1 · p. 2
- Embodied CoT makes policy failures more interpretable and lets humans correct behavior by modifying faulty reasoning chains with natural-language feedback.Source: §1 · p. 3
2 Related Work
Related work traces robot-policy generalization from large, diverse datasets and pretrained vision-language models, then identifies a gap: VLAs do not sufficiently use reasoning that is grounded in visual observations and agent state.
Scaling robot learning
Vision-language models for robot generalization
- VLAs [7] offer a scalable alternative for integrating pretrained vision-language models into robot policies, with state-of-the-art generalist performance [7] and generalization to new objects and scenes.The related work notes that existing VLAs do not sufficiently leverage the underlying models’ ability to reason through the steps required for a task.Source: §2 · p. 3
Reasoning for language and control
- The paper trains a VLA to generate high- and low-level chain-of-thought reasoning and actions from instructions and observations, grounding both forms of reasoning in the agent’s environment.The authors argue that reasoning should support low-level control as well as high-level sub-task planning, and report that this grounded formulation is critical to leveraging (V)LM reasoning for control.Source: §2 · p. 3
- Prior robotics approaches use pretrained or fine-tuned LLMs to decompose tasks into high-level sub-tasks, relying on pretrained low-level policies for execution.The authors contrast this with their position that reasoning can also benefit low-level control and should be grounded in visual observations and agent state.Source: §2 · p. 3
- Step-by-step prompting improves LLM performance on complex tasks, and chain-of-thought has become standard in language and vision-language model training.Related work also applies LLM reasoning to high-level robotic task planning.Source: §2 · p. 3
3 Preliminaries: Vision-Language-Action Models
Vision-language-action models fine-tune a pre-trained vision-language model to predict robot actions from an image observation and task instruction. This work uses OpenVLA as its open-source VLA backbone.
3 Preliminaries: Vision-Language-Action Models
- VLAs autoregressively predict the next robot action from the current image observation and task instruction, converting continuous actions into discrete tokens.Each action dimension is discretized into 256 bins [7].Source: §3 · pp. 3–4
- ECoT interleaves linguistic subtask reasoning with steps grounded in the scene and robot state, which the experiments identify as key to improving policy performance with chain-of-thought reasoning.The grounded steps connect reasoning to visual features and the robot state before action prediction.Source: Figure 3 · p. 4
- OpenVLA combines a fused SigLIP and/or DinoV2 visual encoder with a Llama 2 7B backbone to map image and task tokens to action tokens [7].It builds on the Prismatic VLM and is fully open-source.Source: §3 · p. 4
4 Embodied Chain-of-Thought Reasoning for Visuomotor Policies
ECoT trains VLAs to reason about plans, sub-tasks, motions, and visually grounded scene and robot features before predicting actions. It uses automatically generated reasoning data and reuses reasoning across steps to address inference speed.
4 Embodied Chain-of-Thought Reasoning for Visuomotor Policies
- Instead of relying on prompting small VLA backbones to reason, ECoT post-hoc labels robot data with reasoning strings and trains models to predict reasoning tokens before action tokens.The labels use features extracted from pre-trained models, producing observation-reasoning-action tuples for autoregressive training.Source: §4 · p. 4
4.1 Designing Embodied Chain-of-Thought Reasoning Steps
- ECoT progresses from rephrasing the TASK and planning steps to choosing a SUBTASK, predicting a MOVE command, and grounding reasoning in GRIPPER position and OBJECT bounding boxes.The authors note that alternative reasoning tasks and orderings remain unexplored, and automatic selection of sensible chains is future work.Source: §4.1 · pp. 4–5
4.2 Generating Embodied Chain-of-Thought Data at Scale
- The synthetic-data pipeline generated ECoT annotations for Bridge v2’s more than 2.5M transitions over 7 days.It combines Prismatic-7B scene descriptions, Grounding DINO object boxes, proprioception-derived movement primitives, OWLv2/SAM gripper positions, and Gemini-generated plans and sub-tasks. Object detections require box confidence above 0.3 and text confidence above 0.2; movement directions over the next four time steps map to 729 templates, and gripper projections are fitted per trajectory with RANSAC.Source: Figure 4 · p. 5, §4.2 · pp. 5–6
4.3 Efficient Chain-of-Thought Inference for Robot Policies
- Reasoning increases tokens per timestep from 7 for OpenVLA to 350 for ECoT; synchronous and asynchronous strategies reuse high-level reasoning across steps to improve inference speed.Synchronous execution updates high-level reasoning every N steps; asynchronous execution uses one policy instance to update reasoning and another to predict lower-level reasoning and actions.Source: §4.3 · p. 6
5 Experiments
Across challenging manipulation tasks, ECoT improves generalization over several policy baselines, while its reasoning traces support failure diagnosis, natural-language correction, and more efficient inference.
5.1 Experimental Setup
- Experiments used WidowX with Bridge V2’s 60k demonstrations and 314 trials per approach across spatial-relation, unseen-object, and unseen-instruction tasks.Policies received a single third-person camera view and natural-language instruction, and predicted end-effector velocity actions.Source: §5.1 · p. 6
5.2 Embodied Chain-of-Thought Reasoning Improves Policy Generalization
- ECoT’s aggregate absolute success rate exceeded Octo, OpenVLA (Bridge), RT-2-X, and naïve CoT by 45%, 22%, 19%, and 18% in-distribution, and 48%, 34%, 16%, and 16% out-of-distribution.ECoT also outperformed OpenVLA (Bridge) on all but one generalization evaluation and surpassed RT-2-X despite its 10 additional robot datasets and 7x larger network.Source: Table 1 · p. 7, §5.2 · p. 7
- ECoT and OpenVLA (Bridge) shared the same VLM base and robot fine-tuning data; ECoT’s reasoning traces ground sub-tasks in object boxes and gripper position before action.The reported distinction between the policies was ECoT’s use of chain-of-thought reasoning.Source: §5.2 · p. 7
5.3 Diagnosing Policy Failures Through Inspecting Reasoning Chains
- Inspecting ECoT’s reasoning exposed a failure in which it mistook a hammer for a screwdriver and reached for the wrong object.Reasoning chains are not a foolproof account of an end-to-end policy’s actions, though they often correlate strongly with execution.Source: §5.3 · p. 7
5.4 Chain-of-Thought Reasoning Enables Interactive Policy Correction
- ECoT’s success rate increased by 48% on the most challenging tasks after a human provided one natural-language intervention during a rollout.ChatGPT adapted the reasoning chain to the feedback, which was held fixed for five steps; without intervention, ECoT averaged 32% success on those tasks. OpenVLA and RT-2-X did not benefit to the same degree from the correction.Source: §5.4 · p. 8
5.5 Efficient Chain-of-Thought Inference
- Both proposed inference accelerations improved speed while at least matching performance; asynchronous execution was fastest but doubled inference compute.The five-step freeze approach was used for Table 1 because it offered the best performance-speed tradeoff.Source: §5.5 · p. 9, Table 2 · p. 9
5.6 Additional Analysis
- Moving bounding-box prediction earlier in the reasoning chain sped up inference by 30–50%, although the frozen-box policy performed below base ECoT while outperforming the baselines.The authors adopted this faster structure for subsequent experiments because it also supported rollout visualization.Source: §5.6 · p. 9
- Co-training retained conversational question-answering ability but yielded no measurable evaluation-task gains; fine-tuning a pretrained VLA with ECoT data nearly matched the original ECoT model in 20k rather than 80k steps.Fine-tuning showed comparable performance after 2,500 steps, a reported 30x compute reduction. The model also generated ECoT reasoning for other robot embodiments without embodiment-specific reasoning annotations, but in SIMPLER’s real-to-sim setting its reasoning was faulty and did not improve over OpenVLA.Source: §5.6 · pp. 9–10
6 Discussion and Limitations
The discussion credits scene- and robot-state-grounded chain-of-thought with improving VLA performance without additional robot training data, while identifying fixed reasoning chains, limited scaling, and runtime as limitations.
6 Discussion and Limitations
- 69% aggregate success was achieved by the base ECoT policy; all Table 3 policies exceeded OpenVLA (Bridge) (29%), RT-2-X (46%), and Octo (14%) on the same trial subset.The table evaluates a large subset of trials in the harder out-of-distribution view setting; other design choices achieved higher performance on some tasks.Source: Table 3 · p. 10
- ECoT improves VLA performance without additional robot training data by grounding reasoning in scene and robot state, including object boxes, the end-effector, and low-level movements.The discussion contrasts these grounded reasoning steps with simply applying the language-modeling CoT recipe.Source: §6 · p. 10
- ECoT uses the same fixed reasoning-step order for every task; future work could select a subset based on robot and scene state and optimize the model to choose steps.The paper identifies the lack of task-adaptive reasoning-chain structure as a limitation.Source: §6 · p. 10
- Scaling ECoT training to a larger OXE dataset subset could improve transfer to more robots, while its still-limiting runtime constrains higher-frequency control.Runtime optimizations in Section 4.3 improve achievable control frequencies; exploring other approaches for improving LLM throughput may enable higher-frequency CoT control.Source: §6 · pp. 10–11
A Grounding DINO Detections and Prismatic Descriptions
The section illustrates Prismatic VLM-generated scene descriptions and Grounding DINO bounding boxes for Bridge observations, alongside examples of chain-of-thought reasoning and task prompts.
A Grounding DINO Detections and Prismatic Descriptions
- Figure 8 shows Prismatic VLM captions of Bridge observations alongside associated bounding boxes generated by Grounding DINO.The examples illustrate scene descriptions and detections together.Source: A Grounding DINO Detections and… · p. 18, Figure 8 · p. 18
- Figure 9 provides qualitative examples of successful and failed chain-of-thought reasoning.Source: Figure 9 · p. 19
- Figure 10 shows example starting scenes and associated prompts for all task types.Source: Figure 10 · p. 20
B List of Movement Primitives
The appendix defines movement primitives by comparing the robot’s current state with its position four steps ahead, then labeling detected motion and gripper actions. Labels omit inactive axes, and no detected movement is labeled stop.
B List of Movement Primitives
- Movement labels are assigned from state differences four steps ahead, using a 0.03 threshold on each axis.Axes below threshold are omitted; when no movement is detected, the label is stop.Source: B List of Movement Primitives · p. 21
- The primitive vocabulary combines directional movement, tilt, clockwise or counterclockwise rotation, and opening or closing the gripper.Movement directions cover forward/backward, left/right, and up/down.Source: B List of Movement Primitives · p. 21
C Prompts
The section presents prompts for synthetic data generation, step-by-step trajectory annotation, language-conditioned policies, and human interventions. Its annotation instructions require reasoning grounded in the task, executed trajectory, and environment features.
C Prompts
- Scene-description generation uses a prompt about scene contents and spatial relations, optionally prefixed with the task instruction; the Gemini labeling prompt is shown in Figure 11.The task instruction is included only when the demonstration has a corresponding instruction containing at least one space character.Source: C Prompts · p. 22, Figure 11 · p. 23
- OpenVLA policy prompts are inserted into the original authors’ template [7], after which the agent generates reasoning text when trained to do so and an action.The prompts and example starting scenes for the language-conditioned policies are provided in Figure 10.Source: C Prompts · p. 22
- The prompt used for ChatGPT during human-intervention experiments is provided in Figure 12.Source: C Prompts · p. 22, Figure 12 · p. 24
- The annotation prompt asks for a task overview, remaining high-level movements and their justifications, then precise step-by-step reasoning based on progress, relevant objects, and the executed trajectory.It also asks annotators to include relevant object locations, obstacles, movement directions, and why each action is appropriate.Source: C Prompts · p. 23
- Each step’s reasoning is specified with <task>, <plan>, <subtask>, <subtask_reason>, <move>, and <move_reason> tags.The prompt requires one separate reasoning string for each trajectory step, with the same integer keys as the trajectory features.Source: C Prompts · p. 23
- The response checklist calls for a task description, executed high-level movements, a solution plan, one precise reasoning string per trajectory step, and a final FINISHED label.The example annotation context specifies a successful cloth-unfolding task and describes the robot’s environment and primitive movements.Source: C Prompts · p. 23
Related papers
This paper cites
- Learning Transferable Visual Models From Natural Language Supervision2021
- Gemini: A Family of Highly Capable Multimodal Models2023
- BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Models2023
- OpenVLA: An Open-Source Vision-Language-Action Model2024
- Sigmoid Loss for Language Image Pre-Training2023
- On the Opportunities and Risks of Foundation Models2021
Cited by
- Decoding Geometry, Not Tokens: Efficient VLMs for Robot Affordance and Trajectory Prediction2026
- PDRVLA: A Vision-Language-Action Model with Physical Dynamics Representations2026
- Pretraining Knows What, Simulation Teaches How: Compositional Generalization in a Sim-Trained Robot Agent2026
- RoboDuo: Attempt, Check and Atomic Recovery through Code for Vision-Language-Action Models2026
- FLORA: An Open Model Suite for Scaling Robotics Data Engines2026
- Externalise the Memory: A Pre-Registered Study of Where an Embodied Agent's Experience Should Live2026