X-Imitator: Spatial-Aware Imitation Learning via Bidirectional Action-Pose Interaction
For robotic manipulation, this work addresses the bottleneck of decoupled spatial perception and action generation, offering a modular framework that enhances existing visuomotor policies.
X-Imitator introduces a dual-path framework that models spatial perception and action execution as a bidirectional loop, enabling mutual refinement. It achieves significant improvements over baselines across 24 simulated and 3 real-world tasks.
Effectively handling the interplay between spatial perception and action generation remains a critical bottleneck in robotic manipulation. Existing methods typically treat spatial perception and action execution as decoupled or strictly unidirectional processes, fundamentally restricting a robot's ability to master complex manipulation tasks. To address this, we propose X-Imitator, a versatile dual-path framework that models spatial perception and action execution as a tightly coupled bidirectional loop. By reciprocally conditioning current pose predictions on past actions and vice versa, this framework enables continuous mutual refinement between spatial reasoning and action generation. This joint modeling exactly mimics human internal forward models. Designed as a modular architecture, the system can be seamlessly integrated into various visuomotor policies. Extensive experiments across 24 simulated and 3 real-world tasks demonstrate that our framework significantly outperforms both vanilla policies and prior methods utilizing explicit pose guidance. The code will be open sourced.