ベクトル量子化ワールドモデル:ロボットマニピュレーションのための潜在フィードバック運動計画
Vector-Quantized World Model: Latent Feedback Motion Planning for Robot Manipulation
ベクトル量子化した潜在状態間の遷移を離散探索で計画し、連続的な視覚フィードバック制御で追従する運動計画フレームワークを提案。ハノイの塔の実機操作で短いデモデータのみから100%の成功率を達成した。
著者: Xuezhou Xu, Chung-Pang Wang, Zhenzhen Li, Benjamin Riviere
分類: cs.RO
原文アブストラクト
Visual world models have shown strong potential for robot planning, yet long-horizon manipulation remains challenging due to compounding prediction errors, large search complexity, and latent plans that the low-level controller cannot reliably reach. In this work, we introduce Vector-Quantized World Model (VQ-WM), a latent feedback motion planning framework that integrates discrete long-horizon search with continuous visual feedback control: a discrete latent predictor models transitions between vector-quantized latent states, allowing the planner to look far ahead and find a shortest path to the goal through discrete search and a continuous learned controller sequentially reaches the planned states through visual feedback, correcting deviations to mitigate compounding errors. Whereas most vector quantization uses Euclidean distance, we use a novel estimator of the first-passage-time as a quasi-metric, which aligns the planning abstraction with the controller's capabilities. We evaluate on the physical robot manipulation of the Tower of Hanoi puzzle: whereas existing state-of-the-art foundation models (VLAs, WAMs) only succeed when the solution trace in their training dataset, our method achieves 100\% success rate with only short-play data, demonstrate its ability to compose long-horizon solutions beyond the short demonstrated trajectories and execute them reliably. Our solution also outperforms strongest world-models by 13.3--85.3 percentage points on simulation benchmarks spanning manipulation and locomotion, demonstrating consistent progress and new capabilities.