日本フィジカルAI新聞

世界のフィジカルAIを、日本語で。

週刊ニュースレター購読
VLAarXiv:2608.09730v1

ワールドトークン:訓練時ワールドモデリングによる身体化ポリシーの強化

World Tokens: Enhancing Embodied Policies with Training-Time World Modeling

シェア:XThreadsFacebookLINEはてブBluesky

視覚言語行動モデルに、訓練時のみ未来映像のデノイザーを接続するワールドアダプタを導入し、推論時は映像モデルを除去して効率的な行動生成を実現する手法を提案した。

詳しい要約

1. どんなもの?

World Tokensは、embodied policyのためのアーキテクチャで、World Adapterを導入し、視覚言語理解、世界ダイナミクスモデリング、行動生成を橋渡しする。訓練中にworld modelingを用いて行動ポリシーを強化しつつ、展開時には効率的な推論を維持する。具体的には、VLM特徴を固定数のworld tokensに変換し、これが将来ビデオのdenoiserと行動エキスパートの両方に条件付けを与える。展開時にはworld-modelブランチを除去し、VLM、World Adapter、行動エキスパートのみを残すため、オンラインのビデオモデル推論は不要。

2. 先行研究と比べてどこがすごい?

従来のVLAモデルは閉ループ制御に優れるが、物理シーンの時間的進化を明示的にモデル化しない。一方、WAMsは事前学習済みビデオワールドモデルを利用して時空間進化を捉えるが、将来生成や大きなビデオバックボーンを制御ループに保持するため推論コストが増大する。World Tokensは、訓練時のみworld modelingを行い、展開時にはビデオモデル推論を排除することで、性能と効率の両立を図る点が新しい。

3. 技術・手法の肝は?

World AdapterがVLM特徴を固定数のworld tokensに変換し、これが共同で微調整されたfuture-video denoiserと行動エキスパートの唯一の視覚言語コンテキストとして機能する。この共有条件付けにより、将来ビデオのdenoisingからの勾配が行動予測に使われる表現を直接形成し、排他的ルーティングによりポリシーがその表現を迂回するのを防ぐ。展開時にはworld-modelブランチを除去し、VLM、World Adapter、行動エキスパートのみを残す。

4. どうやって有効だと検証した?

2Bバックボーンとembodied action pretrainingなしで、LIBEROで非常に競争力があり、SIMPLERで最高の平均を達成し、実世界のR1 Pro成功率を一致した行動のみのベースラインと比較して大幅に改善し、各行動チャンクをVLAレベルのレイテンシで生成することを検証した。

5. 議論はある?

要旨からは、World Tokensの限界や潜在的な欠点についての議論は不明。ただし、訓練時のみworld modelingを用いるため、展開時のダイナミクスモデリングの恩恵が限定的である可能性や、world tokensの固定数が表現力に影響する可能性などが考えられるが、要旨には明記されていない。

6. 次に読むべき論文は?

要旨で参照されている関連研究として、VLAモデル、WAMs(world-action models)、事前学習済みビデオワールドモデル、future-video denoiserなどが挙げられる。具体的な論文名は不明だが、これらの手法を詳述した論文を読むことが推奨される。

※ AIが要旨から生成した要約です。正確性は原文をご確認ください。

著者: Qu Tang, Benhui Zhuang, Bo Yuan, Xue Yu, Longteng Guo, Junlan Feng

分類: cs.CV, cs.RO

原文アブストラクト

Vision-language-action (VLA) models are a widely adopted paradigm for embodied policies. They excel at efficient closed-loop control but do not explicitly model how physical scenes evolve as a task unfolds. Recently emerging world-action models (WAMs) leverage pretrained video world models to capture spatiotemporal evolution, yet retaining future generation or a large video backbone in the control loop substantially increases inference cost. We introduce World Tokens, an embodied policy architecture built around a World Adapter that bridges visual-language understanding, world-dynamics modeling, and action generation. It uses world modeling during training to enhance the action policy while preserving efficient deployment. Specifically, the World Adapter transforms VLM features into a fixed set of world tokens, which condition a jointly fine-tuned future-video denoiser and simultaneously serve as the action expert's sole visual-language context. This shared conditioning allows gradients from future-video denoising to directly shape the representation used for action prediction, while exclusive routing prevents the policy from bypassing that representation. At deployment, the world-model branch is removed, leaving only the VLM, World Adapter, and action expert, with no online video-model inference. With a 2B backbone and no embodied action pretraining, World Tokens is highly competitive on LIBERO, attains the best reported averages on SIMPLER, substantially improves real-world R1 Pro success over a matched action-only baseline, and generates each action chunk at VLA-level latency.