28  Embodied AI & Vision-Language-Action Models

TipTL;DR

Vision-Language-Action (VLA) models extend the multimodal Transformer into physical control. The same architecture that reads text and images now outputs robot actions, by treating an action as just another token to predict, grounding language instructions in perception and continuous motor commands.

Depends on: Multimodal · BERT & GPT

28.1 Why this matters

Every architecture in this book so far consumes and produces symbols: text tokens, image patches, denoised pixels. None of them act in a physical environment. A robot arm following the instruction “pick up the red block” must go from pixels and language to a sequence of motor torques or end-effector positions, a fundamentally different output space from anything covered so far.

The VLA insight is that this does not require a new architecture family. If actions can be represented as tokens, the same Transformer that predicts the next word can predict the next action, reusing every mechanism this book has built (self-attention, autoregressive decoding, large-scale pretraining) for physical control.

28.2 The mechanism

28.2.1 The perceive-reason-act loop

An embodied agent runs a repeating cycle:

  1. Perceive the current state (camera images, sometimes proprioception).
  2. Reason about what to do given the instruction and history.
  3. Act by emitting a motor command.
  4. Perceive the new state and repeat.

This loop is the physical-world analogue of an LLM’s autoregressive generation loop (Decoding): each action, like each token, conditions on everything that came before.

28.2.2 Action tokenization

The key architectural move is representing continuous robot actions as discrete tokens drawn from the same vocabulary space as language tokens. A typical action is 7-dimensional: a 3D position delta, a 3D rotation delta, and a gripper open/close state. Each dimension is uniformly binned (e.g. into 256 bins) and mapped to a token ID, often reusing the least-used tokens in an existing language tokenizer’s vocabulary so no architecture change is needed at all, only a vocabulary reinterpretation:

\[ a \in \mathbb{R}^{7} \;\longrightarrow\; \text{discretize} \;\longrightarrow\; (a_1, \ldots, a_7) \in \{0, \ldots, 255\}^7 \;\longrightarrow\; \text{token IDs}. \]

Once actions are token IDs, the model trains with the exact next-token-prediction objective from BERT & GPT. The loss function does not know or care whether a token represents a word piece or a robot joint delta.

import numpy as np

def discretize_action(a, n_bins=256, lo=-1.0, hi=1.0):
    bins = np.linspace(lo, hi, n_bins)
    return np.digitize(a, bins)

action = np.array([0.42, -0.87, 0.05, 0.0, 0.0, 0.33, 1.0])  # 7-DoF continuous action
tokens = discretize_action(action)
print("continuous action:", action)
print("discretized tokens:", tokens)
continuous action: [ 0.42 -0.87  0.05  0.    0.    0.33  1.  ]
discretized tokens: [182  17 134 128 128 170 256]

28.2.3 The VLA architecture end to end

Camera image "pick up the red block" Vision encoder (ViT) Language tokenizer VLA Transformer (vision + language tokens in one context, same as a multimodal LLM) Predicted action tokens Detokenize → continuous action Robot actuator Environment changes new camera frame closes the loop

The loop shown above is the perceive-reason-act cycle made concrete: vision and language enter through the same kind of encoders as any multimodal LLM (Multimodal), the Transformer backbone processes them in one shared context, and the only genuinely new component is the last mile: detokenizing predicted action tokens into a continuous command the actuator executes, after which the changed environment produces a new camera frame and the cycle repeats.

28.2.4 RT-2 and OpenVLA

RT-2 (Brohan et al., 2023) fine-tunes a pretrained vision-language model (the same ViT + LLM combination from Multimodal) by adding action tokens to its output vocabulary and training on robot-demonstration data interleaved with the web-scale vision-language data the base model already saw. This co-training is the central finding: mixing in the original web data, not just robot data, lets the model transfer general visual and semantic knowledge (recognizing an object it never saw a robot manipulate, for example) into the control task, an emergent-generalization result analogous to emergent abilities in language models.

OpenVLA (Kim et al., 2024) is an open-source replication of this recipe: a Llama-2 backbone with a vision encoder, fine-tuned on a large open cross-embodiment robot dataset. It demonstrates that the VLA recipe is reproducible outside closed labs and generalizes across different robot hardware (“embodiments”) without redesigning the architecture per robot.

28.2.5 Sim-to-real and data challenges

Robot data is expensive to collect; there is no equivalent of “scrape the web” for real-world manipulation trajectories. Two mitigations dominate practice:

  • Simulation: train in a physics simulator, then transfer to real hardware. The “sim-to-real gap” is the mismatch between simulated and real dynamics or visuals that this transfer has to survive.
  • Cross-embodiment datasets: pool demonstrations across many robot types so the model sees enough diversity to generalize, the strategy behind OpenVLA’s training data.

28.2.6 What to observe

  • Nothing here required a new attention mechanism, a new loss function, or a new training procedure, only a reinterpretation of the output vocabulary as including motor-command tokens.
  • Co-training on the original web-scale data (not just robot data) is what transfers general knowledge into the control task, nearly identical in spirit to how pretraining transfers into narrow fine-tuning tasks throughout BERT & GPT.
  • Data scarcity, not architecture, is the binding constraint on embodied AI today, the opposite bottleneck from language modeling.
WarningPitfall: action tokenization is lossy

Discretizing a continuous action into a fixed number of bins caps precision: fine motor control (threading a needle, for example) may need resolution finer than the token vocabulary allows. This is an active area of VLA research; some newer approaches predict continuous action chunks directly rather than discretizing.

28.3 Application & impact

Component Role Lives on as
Vision-language backbone Perception + instruction understanding Reused unchanged from Multimodal VLMs
Discretized action tokens Bridge continuous control into the token vocabulary RT-2, OpenVLA, and most current VLA systems
Autoregressive decoding Action sequence generation Same mechanism as Decoding, applied to motor commands
Web-scale co-training Transfer general knowledge into control The generalization story behind RT-2’s headline results
  • The architecture question is settled, for now. VLA work treats “what network computes the policy” as solved (a Transformer) and focuses effort on data, tokenization granularity, and co-training recipes instead.
  • The frontier open question is what comes next. Whether next-token prediction over discretized actions is sufficient for long-horizon physical reasoning, or whether a genuinely different objective, such as predicting future world states rather than action tokens, is needed. That debate is picked up directly in World Models & JEPA.
NoteKey takeaway

VLA models close the “text and image in, more text and images out” loop this book has followed since the Transformer, by re-treating physical actions as tokens in the same autoregressive framework. It is the same architecture, applied one level further out into the world, and it sets up the book’s final, most open question: is next-token prediction the right objective for an agent that must plan and act, or does that require a different kind of model entirely?

→ Next: Research Landscape — Transformer Era