世界モデルを固定したまま小さな補正で新環境に適応
Sandwich-Residuals: Parameter-Efficient Test-time Adaptation of World Models
この論文をやさしく読む
ひとことで言うと
環境が変わったとき、AIの未来予測モデル全体を書き換えず、入力側と出力側の小さな補正を学んで対応する方法です。
何に役立つ?
新しい条件で動く計画システムの適応に、更新パラメータを少なく抑える方法として役立ちます。ラベルや報酬ではなく、予測のずれから補正を学びます。
この研究の面白いところ
21条件で固定モデルの1.3倍、複合的な変化では1.9倍の成功率を報告しながら、更新するパラメータを大きく削減しています。事前学習済みの内部表現を保つ設計です。
どこまで分かった?
97〜99%減は更新パラメータ数についてであり、同率の実行時間や消費電力削減を意味しません。95%は最良の比較変種に対する性能の割合で、課題の成功率が95%という意味でもありません。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
潜在世界モデルは、学習した表現空間で行動の影響を予測することで計画を可能にするが、テスト時の条件が学習時と異なると、予測の信頼性が低下することがある。既存のテスト時適応法は、事前学習済みモデルの一部を更新してこれに対処するが、多くの場合、数百万のパラメータを変更し、内部のどの構成要素を適応させるか選ぶ必要がある。 本研究では、事前学習済み世界モデルを固定し、予測器の前後に置いた小さな残差補正だけを学習する軽量な代替法Sandwich-Residualsを導入する。残差は、モデル自身の自己教師あり予測誤差を用いてオンラインで最適化され、報酬、ラベル、元の領域のデータを必要としない。 AdaJEPAベンチマークの21条件で、提案法は固定モデルの1.3倍の成功率を達成し、最も強力なAdaJEPA変種の性能の95%を維持しながら、適応させるパラメータを97〜99%減らした。複合的な分布変化のもとでは、固定モデルに対する成功率の優位は1.9倍へ拡大し、内部ブロックを適応させる方法と同程度の性能を保った。さらに、3次元操作用のDINO-WMモデルでも、同じ適応原理を実証する。これらの結果は、世界モデルをテスト時に有効に適応させるために、事前学習済みの内部重みを必ずしも変更する必要はないことを示唆する。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-18(UTC)
- 最新改訂
- 2026-09-18 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-18 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Latent world models enable planning by predicting the effects of actions in a learned representation space, but their predictions can become unreliable when test-time conditions differ from training. Existing test-time adaptation methods address this by updating parts of the pretrained model, often modifying millions of parameters and requiring a choice of which internal components to adapt. We introduce Sandwich-Residuals, a lightweight alternative that keeps the pretrained world model frozen and learns only small residual corrections around the predictor. The residuals are optimized online using the model's self-supervised prediction error and require no rewards, labels, or source-domain data. Across 21 conditions on the AdaJEPA benchmark, our method achieves $1.3\times$ the success rate of the frozen model while retaining 95% of the performance of the strongest AdaJEPA variant and adapting 97-99% fewer parameters. Under compound shifts, this advantage increases to $1.9\times$ the success rate of the frozen model, while remaining comparable to internal block adaptation. We further demonstrate the same adaptation principle on a DINO-WM model for 3-D manipulation. These results suggest that effective test-time adaptation of world models does not necessarily require modifying their pretrained internal weights.
arXiv ID: 2609.21740 / 要約の誤りについて