arXiv:2607.15901：DSWorldワールドモデルがデータサイエンスエージェントの訓練と推論を最大14倍高速化

DSWorldとは何か、なぜデータサイエンスエージェントにワールドモデルが必要なのか？

ワールドモデルとは、行動が実際に実行される前に、その結果を内部でシミュレートするコンポーネントであり、エージェントが毎回実際の結果を待つ必要をなくします。Zherui Yang、Fan Liu、Hao Liuは、データサイエンスエージェント——データ分析、モデル訓練、結果評価のためのコードを自動的に書き実行するシステム——向けに特別に設計されたワールドモデル、DSWorldを導入しました。彼らが解決しようとしている問題は計算上の非効率性です。試されるすべての操作は通常、実際の実行を必要とし、それはしばしばモデル訓練や大規模データセット処理のようなリソース集約的なタスクにおいて顕著です。

DSWorldは構造化された状態、ルーティング、シミュレーションを組み合わせる

このアーキテクチャは3つの要素を組み合わせています。タスクの現在の文脈を簡潔に記述する構造化された状態、実際のシミュレーションを呼び出す価値があるときと迅速な見積もりの方が有利なときを判断するインテリジェントなルーティング、そして実際に実行することなくリソース集約的な操作の結果を予測するLLMベースのシミュレータです。このシステムはさらに、「Reflective World Model Optimization」と呼ばれる手法——ワールドモデル自身の予測誤差から学習する、誤差を意識した強化学習（RL）アプローチ——と、訓練の基盤となる8,000件の記録された軌跡のデータセットと組み合わされています。

DSWorld：訓練が14倍、推論が3〜6倍高速化

結果は顕著な高速化を示しています。DSWorldは、ワールドモデルを使わないアプローチと比較して、RLエージェントの訓練を約14倍高速化し、検索ベースの推論——エージェントが最良の行動を選ぶ前に複数の候補行動を探索するプロセス——を3〜6倍高速化します。さらに、このシステムは状態遷移予測——特定の操作の後にタスクの状態がどう変化するかを予測すること——のタスクにおいて、テストされた最も強力なLLMベースラインを35.6%上回りながら、データサイエンス全体のタスクにおいても競争力のある性能を維持しています。

高速化がエージェント作業の経済性を変える

14倍の訓練高速化と3〜6倍の推論高速化の組み合わせは、データサイエンスエージェントが同じ時間・計算予算の中でより多くの戦略を試せることを意味します。ハイパーパラメータ探索やモデル選択のような、通常はコストのかかる繰り返しのコード実行を必要とするタスクにとって、これは直接的により低いコストと、自動化されたデータサイエンスパイプライン開発におけるより速いイテレーションにつながります。