Robot Wiki

Knowledge Insulation

Training the VLM backbone on discrete tokens while a flow-matching expert learns actions behind a stop-gradient.

Last reviewed
Reading time
7 min
Citations
7

RT-2 represents robot actions as tokens, using the discretization introduced for RT-1: each continuous action dimension is divided into 256 uniform bins. Its robot inputs pair a camera image with a task instruction Brohan 2023. π0 instead adds a continuous flow-matching action expert to a pretrained VLM Black 2024. Knowledge Insulation, studied by Physical Intelligence in 2025, combines the two: discrete actions train the backbone’s representations while a continuous-action expert trains alongside it without sending gradients back into the backbone.

Driess 2025 The paper distinguishes the earlier two-stage π0.5 recipe from simultaneous single-stage training.

Driess 2025 The research note calls this single-stage model π0.5 + KI.

The barrier is below. Switch the pass direction to backward and then turn the stop-gradient off: the damage the recipe exists to prevent runs straight back into the pretrained backbone.

language model (2B)action expert (300M)out: FAST token logitsout: continuous actionsin: image, text, state tokensin: action tokens + noise
Language following (illustrative)92 / 100

backbone supervision: no gradients (inference)

measured effect: 7.5x fewer training steps π0.5 + KI generalist vs π0, to similar table-bussing performance

Forward pass at depth 8 of 8 keeps backbone supervision on no gradients (inference), language following at 92 of 100, and the separately reported 7.5x fewer training steps for the π0.5 + KI generalist versus π0 at similar table-bussing performance; the stop-gradient is on so expert gradients stay inside the action expert.

Current MoT pass
pass
forward
depth
8 / 8
stop-gradient
on
supervision
no gradients (inference)
language score
92 / 100
blue: token and activation flowdashed blue: sideways attention (forward)dash-dot arrow: corrupting gradient (backward)long-dashed vertical line: gradient barrier

Schematic: 8 layers drawn per stack for legibility, and the 0-100 language-following score is an illustrative rendering of the paper's qualitative example (π0 is told to bus a spoon into a bin but grasps trash), not a published curve. Appendix B specifies the 2B language model and 300M expert. The 7.5x training-step comparison is for the π0.5 + KI generalist versus π0 at similar table-bussing performance. Source: Driess 2025

7.5x fewer
Training steps
π0.5 + KI generalist on table bussing, versus π0
3B params
VLM backbone
complete VLM in the research-note diagram
300M params
Action expert
paper configuration, trained with flow matching
Continuous
Runtime output
FAST action tokens provide training supervision

The 7.5x figure compares training steps to similar table-bussing performance, not inference latency or equal wall-clock training budgets. The paper reports about 20% extra training cost from the combined objectives. Its Appendix B describes a 2B language-model backbone within the complete VLM and a 300M action expert; the research note labels the complete VLM as 3B Driess 2025 Driess 2025.

What naive joint training breaks

The paper studies a pretrained VLM with a newly initialized continuous-action expert. It finds that expert gradients can interfere with the image encoder and language backbone during adaptation, slowing learning and weakening language following. The proposed explanation is interference with pretrained representations, not a measured claim that every early gradient is large or that the model can never recover Driess 2025.

Figure 2 illustrates the problem with π0: instructed to bus a spoon into the bin, it grasps trash instead. The research note describes the target as the dish container and explicitly calls its explanation a hypothesis: disrupted language processing may lead the model to learn other correlations first Driess 2025 Driess 2025. The note’s phrase "not suitable for high-frequency, precise, or fluent motions" refers to the earlier simple fixed-bin action representation, not to every possible tokenized-action method Driess 2025.

The recipe, in three parts

Knowledge Insulation separates the learning signals Driess 2025:

  1. The action expert trains with flow matching to generate continuous action chunks at inference.
  2. The backbone learns from discrete FAST action tokens, alongside general vision-language and robot-planning data. FAST normalizes actions, applies a DCT separately along time for each action dimension, scales and rounds the coefficients, then uses byte-pair encoding to compress the flattened sequence. Coefficient quantization is lossy; BPE losslessly compresses the resulting integer sequence Pertsch 2025.
  3. Stop-gradient blocks the expert’s updates to the backbone. The expert can attend to the image/language/state prefix and its own continuous-action tokens. It does not attend to FAST action tokens, and the backbone does not attend to the expert. The attention equations apply stop-gradient to the backbone keys and values read by the expert, leaving forward information available while blocking that backward path Driess 2025.

The figure at the top of this module walks both passes through the two stacks, including the failure mode with the barrier removed.

Why the barrier alone is not enough

Stopping expert gradients is not the same as adapting the backbone for robot control. Without another robot-learning signal, the pretrained backbone’s representations remain unadapted to the task. In the paper’s frozen-backbone ablation, shirt-folding performance is reported as 0%; that is an outcome for this tested configuration, not proof that all frozen-backbone policies are impossible Driess 2025.

FAST-token supervision supplies the missing action-learning signal while the gradient barrier blocks updates from the newly initialized expert. The paper also finds that vision-language co-training can give the joint-training baseline good language following even without stop-gradient. The evidence supports the combined recipe, not a claim that continuous gradients always destroy knowledge or that cross-entropy guarantees perfect preservation Driess 2025.

What it buys

The paper separates two table-bussing comparisons. For a generalist trained on many embodiments, π0 needs 7.5 times as many training steps as π0.5 + KI to reach similar performance. For specialist models trained on one embodiment, the authors report that π0-FAST needs twice the wall-clock time to solve the task. That second result is task-completion time, not a generic twofold inference-latency multiplier or a comparison across all robot tasks Driess 2025 Driess 2025.

The table-bussing task starts with 12 objects and scores correct placement into the dish and trash receptacles. The paper reports ten episodes per real-world task and policy. In a separate mobile-manipulation object-generalization experiment, the robot moves unseen objects from a counter into an already-open drawer; the research note says web-data co-training boosted generalization most in that comparison. These are different evaluation populations, not one combined benchmark Driess 2025 Driess 2025.

Language following also improves in the items-in-drawer comparison, but remains imperfect. The paper notes that VLM co-training can help the joint-training baseline follow language even without stop-gradient Driess 2025.

Both, in different places

In the π0.5 + KI recipe, action representation has two roles: FAST tokens provide training supervision for the backbone, while the flow-matching expert generates continuous actions at inference. The expert’s gradients do not update the backbone Driess 2025. A comparison matrix should distinguish training representation from runtime output rather than force this model into a single binary category.

The π0.6 model card describes a Gemma3 4B backbone and an approximately 860M action expert with the same number of layers; its architecture figure also labels a SigLIP 400M vision encoder. The backbone predicts FAST tokens, while the expert predicts continuous actions without sending its gradients back into the backbone. With five denoising steps and three camera inputs, producing a chunk takes 63 ms on one H100 Physical Intelligence 2025. π0.7 retains a Gemma3-based VLM, a MEM-style video-history encoder and a flow-matching expert. Its prompt adds language, episode metadata and subgoal images; it also uses FAST supervision with action-expert gradients stopped at the backbone Ai 2026.

How to read this module

The interactive’s eight drawn layers and 0 to 100 language meter are a deterministic illustration, not measured architecture depth or a published score curve. The Knowledge Insulation paper provides the qualitative spoon/trash example and the generalist table-bussing comparison: π0 takes 7.5 times as many training steps to reach similar performance as π0.5 + KI. Appendix B specifies a 2B language-model backbone and a 300M expert; the separate research-note diagram labels the complete VLM as 3B. The π0.6 card separately describes a Gemma3 4B backbone and an approximately 860M expert Driess 2025 Driess 2025 Physical Intelligence 2025.

See also

  • The Pi Line

    pi0 to pi0.7: flow-matching action experts, FAST tokenization, open-world generalization, and source-scoped checkpoint availability.

  • Vision-Language-Action Models

    RT-1, RT-2, RT-X, Octo, and OpenVLA: web-scale pretraining meets robot control, and the cost of discrete action tokens.

  • Hierarchical Approaches

    SayCan, code-as-policies, and keypoint affordances; why separate planners gave way to internalized hierarchy.

Linked from

  • The Pi Line

    pi0 to pi0.7: flow-matching action experts, FAST tokenization, open-world generalization, and source-scoped checkpoint availability.

  • Hierarchical Approaches

    SayCan, code-as-policies, and keypoint affordances; why separate planners gave way to internalized hierarchy.

References

  1. Danny Driess, Jost Tobias Springenberg, Brian Ichter, Lili Yu, Adrian Li-Bell, Karl Pertsch, Allen Z. Ren, Homer Walke, and 3 more, 2025.

    https://arxiv.org/abs/2505.23705

  2. Danny Driess, Jost Tobias Springenberg, Brian Ichter, Lili Yu, Adrian Li-Bell, Karl Pertsch, Allen Z. Ren, Homer Walke, and 3 more, 2025.

    https://www.pi.website/research/knowledge_insulation

  3. Karl Pertsch, Kyle Stachowicz, Brian Ichter, Danny Driess, Suraj Nair, Quan Vuong, Oier Mees, Chelsea Finn, and 1 more, 2025.

    https://arxiv.org/abs/2501.09747

  4. Kevin Black, Noah Brown, Danny Driess, Adnan Esmail, Michael Equi, Chelsea Finn, Niccolo Fusai, Lachy Groom, and 16 more, RSS 2025, 2024.

    https://arxiv.org/abs/2410.24164

  5. Physical Intelligence, 2025.

    https://website.pi-asset.com/pi06star/PI06_model_card.pdf

  6. Bo Ai, Ali Amin, Raichelle Aniceto, Ashwin Balakrishna, Greg Balke, Kevin Black, George Bokinsky, Shihao Cao, and 79 more, 2026.

    https://www.pi.website/download/pi07.pdf

  7. Anthony Brohan, Noah Brown, Justice Carbajal, Yevgen Chebotar, Xi Chen, Krzysztof Choromanski, Tianli Ding, Danny Driess, and 46 more, 2023.

    https://arxiv.org/abs/2307.15818

Spot a factual error or missing qualification? Report a content correction.