世界モデルの特徴を小型ロボット方策へ移す
Think Like a World Model, Act Like a VLA: Distilling World-Model Representations into Compact Robot Policies
この論文をやさしく読む
ひとことで言うと
世界モデルが学んだ物理場面の特徴を、制御時の追加計算なしで小型ロボット方策に移す方法。
何に役立つ?
ロボット方策の推論速度やメモリ使用量を保ちながら、操作性能を高める学習法の候補になる。実機への適用も報告されている。
この研究の面白いところ
世界モデルで未来を生成せず、保存した内部特徴との整合だけをVLA学習に加えている。8億パラメータの生徒モデルでLIBERO 97.9%、RoboCasa-GR1では48.2%から50.5%を報告する。
どこまで分かった?
速度とメモリの値はRTX 5090上での測定であり、各ベンチマークや片腕・両腕の実機での結果が、あらゆるロボットや環境に一般化するとは要旨からは言えない。
v2のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
視覚・言語・行動(VLA)モデルは観測から行動を直接求めるが、その行動に世界がどう反応するかを考慮する目的がないため、頑健性は主に学習データの範囲に左右される。世界モデルにはその目的があり、物理的な状況に根ざした表現を学べる一方、未来を生成するには判断ごとに数秒かかり、制御ループに組み込むことは難しい。本研究は、両者を切り離せることを示す。世界モデルが物理的な場面について獲得した知識は内部の特徴にあり、未来の生成はその特徴を生んだ学習目的にすぎない。そこで、生成機構を使わずに、物理的な表現だけを引き継ぐ。 通常のVLA学習に特徴整合の項を一つ加える。固定した世界モデルを学習用フレームに一度適用して特徴を保存し、生徒モデルがそれに一致するよう学習する。学習中に教師モデルを読み込む必要はなく、投影器は学習後に捨てる。実際に配置する方策は蒸留前の基準モデルと同一の構成で、一般向けのRTX 5090上で32ミリ秒、1.86GBで動作する。このため、性能向上は容量や推論時の追加計算ではなく表現に由来すると著者らは説明する。 8億パラメータの生徒モデルはLIBEROで97.9%に達し、RoboCasa-GR1の人型ロボット操作では48.2%から50.5%へ改善した。同じ学習目的は片腕と両腕の実機にも適用できた。改善は生徒モデルの規模、基盤モデル、整合層、教師モデルを変えても維持され、特定の二つのネットワーク間の壊れやすい整合ではなく、広く使える表現上の事前知識を示唆する。プロジェクトページ:https://thaw-vla.trung-dt.com/。
v2の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-21(UTC)
- 最新改訂
- 2026-09-22 · v2
- 査読・掲載
- 査読状況未確認
更新履歴
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Vision-Language-Action (VLA) models map observations to actions with no objective that accounts for how the world responds, so their robustness is bounded primarily by data coverage. World models carry precisely that missing objective and are better grounded for it, yet rolling the future forward costs seconds per decision and rules them out of the control loop. We show the two can be separated. What a world model knows about physical scenes lives in its internal features; generating the future is merely the objective that produced them, so the grounding can be inherited while the generative machinery is left behind. We add one feature-alignment term to ordinary VLA training: a frozen world model is run over the training frames once and cached, and the student learns to agree with that cache. No teacher is loaded during training, the projector is discarded after it, and the deployed policy is identical to the undistilled baseline, running in 32ms and 1.86GB on a consumer RTX5090, so every gain is attributable to the representation rather than to added capacity or test-time compute. A 0.8B student reaches 97.9% on LIBERO, improves from 48.2% to 50.5% on RoboCasa-GR1 humanoid manipulation, and the same objective carries over to real hardware, on both a single-arm and a bimanual platform. The gain survives changes of student scale, backbone, alignment layer, and teacher, indicating a broad representational prior rather than a fragile alignment between two particular networks. Project page: https://thaw-vla.trung-dt.com/.
arXiv ID: 2609.24682 / 要約の誤りについて