Axiom-0: Hierarchical Self-Refinement for Vision-Language-Action Policies
cs.RO, cs.CV
Submitted: 2026-10-09
Updated: 2026-10-09
Code: https://github.com/XiaomiRobotics/Xiaomi-Robotics-1
License: http://creativecommons.org/publicdomain/zero/1.0/
The gist: Reliable manipulation requires policies that generalize across tasks, layouts, and object interactions.
Abstract
Reliable manipulation requires policies that generalize across tasks, layouts, and object interactions. Vision-language-action (VLA) models combine pretrained multimodal understanding with learned control, but their performance remains limited on unfamiliar task compositions. We present Axiom-0, a VLA policy that addresses both the quality of its representations and their use in action generation through hierarchical self-refinement. Intra-VLM Representation Refinement (IVRR) strengthens selected intermediate features through consistency supervision with learned feature projections during training. VLM-Action Interface Refinement (VAIR) recursively updates the resulting conditioning before action generation. Joint optimization of the supervised VLM layers, VAIR, and the action generator connects the two components. On RoboCasa365, Axiom-0 achieves 62.32% overall success in our evaluation, exceeding the highest published leaderboard score (Paimon-0, 58.1%) by 4.22 percentage points and the Xiaomi-Robotics-1 reference by 4.92 points. Under the evaluated profiling setup, mean action-chunk generation latency increases by 1.7% over Xiaomi-Robotics-1. These results demonstrate improved task completion with modest inference overhead.
Sources
- $\pi_0$: A Vision-Language-Action Flow Model for General Robot Control
- $\pi_{0.5}$: a Vision-Language-Action Model with Open-World Generalization
- Xiaomi-Robotics-1: Scaling Vision-Language-Action Models with over 100K Hours of Real-World Trajectories
- RoboCasa365: A Large-Scale Simulation Framework for Training and Benchmarking Generalist Robots
- LIBERO-Plus: In-depth Robustness Analysis of Vision-Language-Action Models
- LIBERO-Safety: A Comprehensive Benchmark for Physical and Semantic Safety in Vision-Language-Action Models
- SpatialVLA: Exploring Spatial Representations for Visual-Language-Action Model
- From Spatial to Actions: Grounding Vision-Language-Action Model in Spatial Foundation Priors
- Spatial Forcing: Implicit Spatial Representation Alignment for Vision-language-action Model
- VEGA: Visual Encoder Grounding Alignment for Spatially-Aware Vision-Language-Action Models
- Looped Transformers are Better at Learning Learning Algorithms
Related papers
- FMT x: An Efficient and Asymptotically Optimal Extension of the Fast Marching Tree for Dynamic Replanning
- MPCFormer: A physics-informed data-driven approach for explainable socially-aware autonomous driving
- RoboLab: A High-Fidelity Simulation Benchmark for Analysis of Task Generalist Policies
- ARC: A Reasoning Recipe for Robot Foundation Models
- HRDexDB: A 4D Dexterous Grasping Dataset Across Human and Multiple Robot Embodiments
- APT: Action Expert Pretraining Improves Instruction Generalization of Vision-Language-Action Policies