テクニカルサマリー:Co-Harness: LLMエージェントのためのハーネスとモデル重みの共進化
1. 問題提起
自動化されたAI研究のためのポストトレーニングにおけるエージェント開発は、従来、モデルのパラメータを最適化する一方で、実行時システム(すなわち「ハーネス」)を固定された静的な足場として扱うものでした。ハーネスには、プロンプト、ツール、スキル、ミドルウェア、メモリポリシー、およびリトライロジックが含まれます。現在のパイプラインには根本的なミスマッチが存在します。すなわち、モデルは静的なハーネスによって生成された軌跡に基づいて教師あり微調整(SFT)や強化学習を通じて更新されますが、ハーネス自体は、トレーニング中に観察された失敗に基づいて最適化されることがありません。
この非対称性は、不適切なスキャフォールディング(例:曖昧なプロンプト、誤ったツールスキーマ、欠落したリトライフック、またはメモリオーバーフロー)が、高品質なトレーニング軌跡の生成を妨げるというボトルネックを生み出します。その結果、モデルはこれらの環境的な制限を克服する方法を学習できず、固定されたハーネスがエージェントの性能の天井となってしまいます。これは、特に単一のスキーマエラーがマルチターンのセッション全体を中断させかねない、複雑なツール統合推論(TIR)の設定において顕著です。
2. メソドロジー:Co-Harnessフレームワーク
著者らは、エージェントのハーネス(ϕ)とモデルパラメータ(θ)を交互に最適化するデュアルループ・フレームワークであるCo-Harnessを提案しています。コアとなる仮説は、より優れたハーネスはよりクリーンで情報量の多い軌跡を生み出し、それがより強力なモデルを訓練するというものです。そして、この強力なモデルは、以前は見えていなかった高次のハーネスのボトルネックを露呈させ、さらなるスキャフォールディングの改善を可能にするというものです。
本フレームワークは、ラウンド t において、以下の2つの交互ループで動作します。
A. Co-Harness ループ(ハーネスの最適化)
このフェーズでは、モデル θt を固定し、ハーネス ϕt を進化させます。
- 失敗の収集: エージェントは (θt,ϕt) の下でタスクを実行し、失敗した軌跡の集合 Ft− を収集します。
- HarnessCriticによる分析: LLMベースのコンポーネントである HarnessCritic (C) が、これらの失敗を分析します。これは、各失敗をハーネス構成内の構造化された根本原因(例:プロンプトの曖昧さ、ツールスキーマのエラー、スキルの欠如、ミドルウェアの不一致、またはメモリオーバーフロー)に帰属させます。
- 差分生成と検証: HarnessCriticは、ハーネスに対する局所的なパッチ(差分)を提案します。これらのパッチは、ターゲットとする失敗モードを改善しつつ、保持された挙動に退行(リグレッション)を引き起こさないことを確認するために、「ロールアウト」テストによって検証されます。
- レジストリの更新: 検証されたパッチはバージョン管理されたハーネスレジストリにコミットされ、有効な構成が ϕt∗ に更新されます。
B. モデルアライメント・ループ(モデルの最適化)
このフェーズでは、進化したハーネス ϕt∗ を固定し、モデルを更新します。
- 高品質な軌跡の収集: 現在のモデル θt を改良されたハーネス ϕt∗ の下で実行し、高品質な軌跡のデータセット Dt を生成します。
- 教師あり微調整 (SFT): モデルは Dt に基づいて微調整され、改良されたスキャフォールディングによって可能になった挙動を内面化し、更新されたモデル θt+1 を生成します。
- イテレーション: 新しいモデル θt+1 と進化したハーネス ϕt∗ は、次のラウンドの開始点となります。そこでは、より強力なモデルが、より複雑なハーネスの限界を明らかにする可能性があります。
失敗帰属のタクソノミー(分類学)
HarnessCriticは、失敗を5つの実行可能なハーネス次元と1つの棄却ラベルに分類します。
- プロンプトの曖昧さ (P): 指示の未指定、または矛盾する目標。
- ツールスキーマのエラー (T): 不正なツール呼び出し、引数スキーマの不備、またはバックエンドとの不一致。
- スキルの欠如 (S): 再利用可能なルーチンや分解プリミティブの不在。
- ミドルウェアの不一致 ($Mid$): ループプロトコル、フックの挙動、またはコンテキスト管理ロジックの欠陥。
- メモリオーバーフロー (M): 永続的な状態のオーバーフロー、古いメモリ、またはリトリーバルの失敗。
- エージェントのエラー: 失敗がハーネスではなく、モデル自身の推論に起因する場合。
3. 主な貢献
- 結合最適化の定式化: 本論文は、エージェントのポストトレーニングを、プロンプトやツールを固定されたインフラとして扱うパラダイムを超え、モデルの重みと実行時ハーネスの結合最適化問題として定式化しました。
- HarnessCritic: 失敗に基づいた手順を導入し、エラーを特定のハーネスコンポーネントに構造的に帰属させ、修復のための検証済み局所差分を提案することを実現しました。
- 累積的利得のメカニズム: ハーネスの進化とモデルのアライメントを交互に行うことで、スキャフォールディングの改善がより良いトレーニングデータをもたらし、それがさらに高度なスキャフォールディングを活用できる強力なモデルを生み出すという、正のフィードバックループが形成されることを実証しました。
4. 実験結果
本フレームワークは、数学的ベンチマーク(AIME 2024, AIME 2025, HMMT 2025)を用い、2つのモデルスケール(Qwen3-8BおよびQwen3-32B)を用いたツール統合推論(TIR)タスクで評価されました。
- 性能向上: Co-Harnessによる2ラウンドの共進化は、ベースライン(モデルのSFTなしでハーネスを一度だけ進化させたもの)と比較して、ベンチマークとモデルスケール全体で平均 +20.4 パーセントポイント (pp) の精度向上をもたらしました。
- 最大の利得は、最も困難なベンチマーク(HMMT 2025)におけるQwen3-32Bモデルで観察され、+27.2 pp の向上を記録しました。
- 本システムは、入念に設計された静的なハーネス構成を平均 +24.7 pp 上回りました。
- 自律的なケーススタディ: AIME 2024における200時間以上の自律実行において、システムは以下を実現しました。
- 初期システムクラッシュ(vllm KVキャッシュエラー)からの回復。
- グローバルバッチ推論を通じて、推論効率を 8.7倍 向上(実時間を3.78時間から1.11時間に短縮)。
- 人間の介入なしに、時間予算内で最高精度(63.3%)を達成するアンサンブル戦略(多様なシードの多数決)を発見。
- 堅牢性: システムは、バージョン管理されたレジストリを使用して、モデルの内部推論と競合するドメイン固有のプロンプトなどの退行を正常に検出し、ロールバックを行いました。
5. 重要性と主張
本論文は、Co-Harnessが現在のエージェント開発における重要な限界に対処していると主張しています。それは、モデルのトレーニングと、トレーニングデータを生成する環境との間のデカップリング(分離)です。ハーネスを静的な足場ではなく、学習可能な変数として扱うことで、Co-Harnessは累積的な自己改善を可能にします。
著者らは、以下の「正のフィードバックのスパイラル」を実現していると主張しています。
- より良いハーネスは、よりクリーンなツール使用と、回避可能なクラッシュの減少をもたらす。
- これらのクリーンな軌跡が、より強力なモデルを訓練する。
- より強力なモデルは、以前は使用不可能であった、より豊かで複雑なハーネス構成を活用できる。
これは、固定されたハーネスに対してモデルを最適化する(標準的なSFT/RL)、あるいは固定されたモデルに対してハーネスを最適化する(推論時の探索)といった、先行研究とは対照的なアプローチです。Co-Harnessは、両者を共進化させる最初のフレームワークとして提示されており、スキャフォールディングとモデルの重みの相互依存性が、固定ハーネスのポストトレーニングを超えてエージェントの能力を向上させるための主要な軸であることを示唆しています。
著者らが指摘する限界: このアプローチには、有能なクリティックLLM、コールドスタートのための十分な量の失敗軌跡、およびマルチラウンドSFTのための多大な計算資源が必要です。また、複雑な反事実的推論を必要とする失敗モードについては、属性付けの精度が低下する可能性があり、ハーネスの構造的な再設計は依然として人間の責任となります。