Preprint
Computer Vision

τ_0-VLA: a Hierarchical Robot Foundation Model with World-Model-Guided Test-Time Computation

Xiaowei Cai, Yunuo Cai, Bingao Chen, Jingxiao Chen, Zhi Chen, Siyuan Feng, Tengyu Hou, Jingshun Huang, Han Jiang, Runkun Ju, Dong Li, Mingxiang Li, Shaowei Li, Xinchen Li, Yifan Li, Yi Liu, Zhongyuan Liu, Jianlan Luo, Junwen Miao, Ruiqi Ni, Buqing Nie, Mingjie Pan, Xinlin Ren, Jianheng Song, Jiaxu Wang, Peiqi Wang, Sen Wang, Xiaoyan Wang, Dafeng Wei, Dongming Wu, Pengwei Xie, Pu Yang, Hangjian Ye, Xiangyu Yue, Jinyu Zhang, Qinglin Zhang, Xueyong Zhao, Pengfei Zhou, Yue Zhou
August 17, 2026

0

Citations

0

Influential Citations

Venue

2026

Year

Abstract

Long-horizon robot manipulation requires a robot to both execute individual skills reliably and sequence them coherently over extended tasks. Most hierarchical vision-language-action (VLA) models make each such decision with a single forward pass, leaving no mechanism to allocate additional computation to difficult or consequential choices. We introduce τ_0-VLA, a hierarchical robot foundation model that formulates high-level subtask generation as a compute-scalable inference problem through world-model-guided test-time computation. At each inference step, the high-level policy uses execution memory to generate a subtask and, when needed, searches over alternatives before committing to its output. A low-level policy then executes the generated subtask across multiple robot embodiments. The policy is trained on 40,115 hours of heterogeneous real-world data with multimodal co-training. Across in-domain and distribution-shifted settings, allocating additional test-time computation substantially improves next-subtask prediction accuracy, and these gains translate into higher closed-loop success on long-horizon robot manipulation tasks.

Analysis

Why This Paper Matters

Long-horizon robot manipulation remains a central challenge in embodied AI, requiring both reliable low-level control and coherent high-level task sequencing. Most hierarchical VLA models make decisions with a single forward pass, which limits their ability to handle ambiguous or high-stakes choices. τ_0-VLA addresses this by introducing test-time computation at the high-level policy, allowing the model to deliberate more when needed. This is a significant step toward making robot policies more robust and adaptable, especially in real-world settings where tasks vary and mistakes are costly.

The paper also emphasizes training on 40,115 hours of heterogeneous real-world data, which is a substantial scale that enables cross-embodiment generalization. By combining world-model-guided search with execution memory, τ_0-VLA offers a new paradigm for hierarchical decision-making that could influence future designs of robot foundation models.

Technical Contributions

  • Hierarchical VLA with test-time computation: The high-level policy generates subtasks and can search over alternatives before committing, guided by a world model. This is a novel formulation of subtask generation as a compute-scalable inference problem.
  • Execution memory: The high-level policy uses memory of past executions to inform subtask generation, enabling context-aware decision-making.
  • World-model-guided search: A world model guides the search over alternative subtasks, likely by predicting future states or outcomes, improving the quality of decisions.
  • Multimodal co-training: The policy is trained on 40,115 hours of real-world data with multimodal co-training, which likely includes vision, language, and action data, enabling cross-embodiment execution.
  • Closed-loop integration: The low-level policy executes generated subtasks across multiple embodiments, demonstrating a complete system for long-horizon manipulation.

Results

The paper reports that allocating additional test-time computation substantially improves next-subtask prediction accuracy, both in-domain and under distribution shift. These gains translate into higher closed-loop success on long-horizon manipulation tasks. While specific numerical metrics are not provided in the abstract, the qualitative claims suggest a clear positive correlation between compute allocation and task performance. The use of heterogeneous real-world data likely contributes to robustness across embodiments.

Significance

τ_0-VLA introduces a practical method for scaling inference compute in robot foundation models, which is a relatively underexplored area compared to scaling training compute. By enabling the model to deliberate on high-level decisions, it improves reliability and generalization in long-horizon tasks. This could inspire further research on compute-adaptive policies, where the model decides how much computation to use based on task difficulty or uncertainty. The hierarchical design also offers a modular approach that could be integrated with other low-level controllers or world models. Overall, this work pushes toward more capable and trustworthy embodied AI systems.