日本フィジカルAI新聞

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

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

DreamWAM: RGB未来予測を超えた世界行動モデル

DreamWAM: Beyond RGB Future Prediction for World Action Models

シェア:XThreadsFacebookLINEはてブBluesky

世界行動モデル(WAM)の未来予測をRGBだけでなく、外観・動き・幾何・意味の複数視点で行う構造化世界モデルを提案し、ロボット操作タスクの成功率とロバスト性を向上させた。

詳しい要約

1. どんなもの?

DreamWAMは、World Action Models (WAMs) のための新しいフレームワークであり、将来予測をRGB空間に限定せず、外観、動き、幾何、意味論という補完的な視点で構造化された世界モデリングとして再構成する。トレーニング時にはRGBと動きの潜在拡散に、幾何と意味論の軽量なゲート付き残差ブランチを組み合わせ、推論時にはRGBのみで動作する。

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

既存のWAMは将来予測をRGB空間で行い、タスク関連の状態遷移がテクスチャや照明などの変動と絡み合う。DreamWAMは、アクション関連の将来状態を明示的に予測することで、RGB予測だけに頼る従来手法より堅牢性を向上させる。

3. 技術・手法の肝は?

DreamWAMは、VideoDiTとActionDiTの間で共有アテンションを用い、アクションブランチが将来状態予測から学習する。RGBと動きの結合潜在拡散に、幾何と意味論のための軽量なゲート付き残差ブランチを追加する。推論時には、RGB以外の監視ブランチは無効化され、デプロイはRGBのみで行われる。

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

LIBEROベンチマークで、no-rolloutとjoint video-action推論の両方でRGBのみのベースラインを上回り、成功率が97.30%から98.40%、98.00%から98.90%に向上。未見のLIBERO-Plus摂動では51.36%から63.44%、69.16%から75.47%に向上。実世界操作では、照明・背景・物体配置の未見変化に対して平均成功率74.4%を達成し、Fast-WAM-Jointの55.6%を上回った。

5. 議論はある?

要旨からは、RGB以外の予測がなぜ堅牢性を向上させるのかの理論的解析や、計算コストの詳細、他のタスクへの一般化可能性などは不明。また、幾何と意味論のブランチが推論時に無効化されることの影響についても議論の余地がある。

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

要旨で参照されているFast-WAM-Joint、および関連するWorld Action Modelsの研究。また、VideoDiTやActionDiTの基盤となる拡散モデルやビデオ予測の研究。

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

著者: Shanglin Yuan, Weiheng Zhao, Xin Shi, Haoyi Jiang, Xianda Guo, Liu Liu, Wenyu Liu, Wei Sui, Xinggang Wang

分類: cs.RO

原文アブストラクト

World Action Models (WAMs) learn action-relevant representations by predicting how the observed world will evolve. Most existing WAMs define this future in RGB space, where task-relevant state transitions are entangled with nuisance variations in texture, illumination, background, and viewpoint. We argue that WAMs should explicitly predict action-relevant future state rather than relying on RGB prediction alone. We introduce DreamWAM, which reformulates future prediction as structured world modeling beyond RGB, representing future states through complementary views of appearance, motion, geometry, and semantics. During training, DreamWAM combines joint latent denoising of RGB and motion with lightweight gated residual branches for geometry and semantics. Shared attention between VideoDiT and ActionDiT allows the action branch to learn from these future-state predictions, while all beyond-RGB supervision branches are disabled at inference and deployment remains RGB-only. Across both no-rollout and joint video-action inference, DreamWAM consistently improves the matched RGB-only baselines on LIBERO, from 97.30\% to 98.40\% and from 98.00\% to 98.90\%, respectively. The gains become larger under unseen LIBERO-Plus perturbations, from 51.36\% to 63.44\% and from 69.16\% to 75.47\%. The same robustness extends to real-world manipulation, where DreamWAM attains an average success rate of 74.4\% across unseen changes in lighting, background, and object layout, compared with 55.6\% for Fast-WAM-Joint. These results show that robust world-action learning depends not only on predicting the future, but on representing it in a form that matters for action. The code and models are publicly released at https://github.com/hustvl/DreamWAM.