powered by TechFeed
表示モード
Microsoft

強化学習環境の標準プロトコル「OpenEnv」をMicrosoftが提唱 — 捨てていたトークン89%を再利用するECHO手法も公開

6月20日、Microsoftが「OpenEnv is the right protocol for reinforcement learning environments (RLEs)」と題した記事を公開した。この記事では、強化学習環境の標準プロトコルとして提唱される「OpenEnv」と、Microsoftが実装した世界モデル学習手法「ECHO」について詳しく紹介されている。

6月20日、Microsoftが「OpenEnv is the right protocol for reinforcement learning environments (RLEs)」と題した記事を公開した。この記事では、強化学習環境の標準プロトコルとして提唱される「OpenEnv」と、Microsoftが実装した世界モデル学習手法「ECHO」について詳しく紹介されている。

「学習ループを自社資産にする」という発想

昨年、エージェントがツールを呼び出す標準規格としてMCP(Model Context Protocol)が普及した。MCPが解決したのは「ツールごとに異なる方言」という統合コストの問題だ。しかしエージェントが「実際の業務で練習して上達する」ための共通規格は、いまだ存在しない。OpenEnvはその空白を埋めようとするプロトコルである。

MicrosoftのEVP・Jay Parikhは「勝者はデモの数ではなく、AIを継続的に改善し続けるシステムを持つ企業だ」と述べ、Satya Nadellaも「保有すべき資産はレンタルしたモデルではなく、自社が所有する学習ループだ」と指摘している。

この「学習ループ」の構造を具体的に示すと:

  • エージェントが実業務を行う環境(Environment)
  • 実際に意味のある成果を採点するルーブリック(評価基準)
  • 繰り返し実行できるロールアウト(試行記録)
  • そのスコアをもとにエージェントを改善するフィードバック機構

この4つが組み合わさったループが「複利的に」積み上がる。モデル自体は交換可能だが、ループは交換できない——それがこの考え方の核心だ。

OpenEnvの設計:テストハーネスではなく学習システム

多くのチームが「環境」をテストとして扱っている。エージェントを走らせ、スコアを読んで終わり、という使い方だ。OpenEnvはその認識を変えようとしている。

OpenEnvのAPIはresetstepstateという3つのメソッドのみで構成されるシンプルな契約だ。これによって以下の3つの性質を実現する:

  • オープン:標準はコミュニティが策定
  • 相互運用可能:どのモデル・トレーナー・ランタイムとも組み合わせ可能
  • モジュール型:環境を再構築せずに任意のコンポーネントを交換可能

現在の参加組織はHugging Face、MetaのPyTorchチーム、NVIDIA、Prime Intellect、Unsloth、Modalなど。MicrosoftもここにJoinしている。OpenEnvはフレームワークではなく、プロトコルだ。

ECHOが面白い:捨てていたトークンで世界モデルを学ぶ

ここが記事の一番のポイントだ。MicrosoftはOpenEnvにRFC 010としてECHOをマージした。元の研究は「Terminal Agents Learn World Models for Free」(arXiv 2605.24517)、実装はmicrosoft/echo-rlで公開されている。

通常のエージェントRLが捨てているもの

エージェントのトレース(記録)は、モデルが書いたアクショントークンと、環境が返したオブザベーショントークンの2種類で構成される。従来の強化学習はアクショントークンだけを学習に使い、オブザベーショントークンを捨てていた。

これがどれだけの損失かというと、実際のエージェントエピソードでは学習可能なトークンの89%(5,247トークン中4,659トークン)がオブザベーションだった。アクショントークンの実に7.9倍だ。

ECHOの仕組み

ECHOはこのオブザベーショントークンを使って、環境の次の状態を予測する「世界モデル」をポリシーに同時に学習させる。損失関数は以下の1行で表現できる:

L = L_GRPO(action tokens) + λ · CrossEntropy(observation tokens)

追加のロールアウトも、教師データも、別の損失関数も不要。同一の順伝播パスで計算済みのロジットを使い、1つのオプティマイザーステップで完結する。λ = 0にすれば通常のRLと同一になるため、段階的に導入できる。

実験結果

ECHOの効果は「学習速度」と「汎化精度」の2軸で報告されている:

  • 保留テストセットでのpass@1(未見タスクへの汎化精度)が約2倍(TerminalBench-2.0)
  • RLが目標精度に達するまでの学習速度が約2.3倍
  • 教師なしで専門家SFT(教師ありファインチューニング)の50〜104%を回収

注意点として、訓練時の報酬はほぼ変わらない。ECHOが効くのは汎化(未見タスクへの転移)の部分だ。λは小さく保ち、スイープで調整することが推奨されている。過剰に大きくすると過学習する。

再現もしやすい。以下のコマンドでCPU上、約40秒で動作を確認できる:

python train_echo.py --steps 60 --seed 0

コードはOpenEnvのexamples/echo_world_modelに公開されている。

同様のアプローチはMeta/Prime Intellectの「True Agents Model the World」でも独立して報告されており、ツールレスポンストークンへの教師あり学習を「一定の正の優位性を持つRL」として再定式化している。2つのグループが独立して同じ方向に収束したという事実は、この手法の妥当性を裏付けている。

ロードマップ:再帰的自己改善

ワークフロー・ツール・ルーブリックがOpenEnv環境に収まると、ポストトレーニングに使った同じトレースデータが環境自体の改善にも使えるようになる。

  • エージェントが上達するにつれて難易度を上げるカリキュラム生成
  • 本番環境のトレースから新しい環境を自動構築
  • ハーネスオプティマイザー

これらは「再帰的自己改善」と呼ばれ、現在ロードマップに載っている段階だ。学習が重みの中だけに蓄積するのではなく、環境(ジム)側にも蓄積される——その環境は自社が所有し続けられる、というのがこの設計の主張だ。

詳細はOpenEnv is the right protocol for reinforcement learning environments (RLEs)を参照していただきたい。