表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

モデルの重みを変えずに実行手順を改善するHarness Learning

Harness Learning

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

実行報告を読んで、AIを動かす周辺コードを修正するループです。図の例では回答の検証と再試行を追加し、その結果を次の修正へ戻しています。

原論文の図の説明を読む

Figure 1 : Harness learning. The proposer reads the task description, the current harness h t − 1 h_{t-1} , and its execution report, and writes a code edit y t y_{t} , here adding answer validation and a retry. Running the revised harness h t h_{t} yields a reward and feedback for the next round. Training updates only the proposer, and at test time both models are frozen. Fig.   2 details training.

出典:Harness Learning / arXiv ↗

概要

  • 実行結果からハーネスのコード修正を学ぶ提案モデルを訓練し、利用時は両モデルの重みを固定する。
  • 未学習の推論21課題群で平均得点0.32から0.62、MuSiQueでは反復修正後の完全一致率0.27を報告した。
  • 反復修正の効果と複数段階訓練の効果は別であり、後者の優位性は設定によって異なる。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、Harness Learningの仕組み

左から右へ読む
  1. 01課題・現在の実行コード
  2. 02実行結果を報告
  3. 03提案モデルがコードを修正
  4. 04再実行して改善を評価

たとえば回答モデル自体を変えるだけでなく、検索結果の渡し方など、周辺の実行手順を改善する。

もともとの課題・処理

決められた実行構成のまま、解答モデルを使う。

この研究のアプローチ

実行報告から周辺コードを修正するモデルを学習する。

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

言語モデルを使うエージェントの性能は、モデル自体だけでなく、何を入力し、どの道具をいつ呼ぶかにも左右される。ハーネスとは、この呼び出し順序、道具の利用、情報の受け渡しを管理する実行プログラムである。

同じモデルでも課題に適した構成は異なるため、手順の改善には試行錯誤が必要になる。本研究は実行結果を読んで有効な変更を提案する能力を学習し、別の課題へ持ち運べるかを問う。

手法

提案モデルは、課題説明、現在のハーネス、実行報告を受け取り、既存コードの一部を置き換える修正を出力する。修正版を問題群で動かし、その得点と実行の成否を報酬として強化学習する。これは、良い実行結果を生む修正の出力確率を高める訓練で、必要に応じて教師の成功例による初期学習も行う。

利用時は提案モデルと解答モデルの重みを固定し、実行、報告、コード修正を繰り返す。例えば複数文書をたどる質問では、次の検索には要約を使いつつ、最終回答には取得した原文を直接渡すよう変更する。出力は回答だけでなく改良された実行手順であり、修正選択用の問題と最終評価用の問題は分離する。

新規性

提案の中心は、実行可能なプログラムを適応対象とするメタ学習、すなわち課題経験から改善の仕方そのものを学ぶ枠組みである。提案モデルが、実行結果を次のコード修正へ変換する役割を担う。

研究では、未学習課題への転移と修正の反復を検証する。単発の修正だけで訓練したモデルも、更新済みの実行手順をさらに改善できる点を報告している。

従来手法との違い

比較には未修正の初期構成、訓練前の提案モデル、教師の修正例で学習したモデル、強化学習後のモデルを用いる。推論課題では、同じ修正条件で40億パラメータの提案モデルが350億の教師を平均点で上回った。

ただし、評価結果を見て8候補から最良を選ぶ参考値では教師が高い。先行するHarness-R1も類似の訓練方式を持ち、本研究は特に未知課題での反復適応を検証する位置付けである。

実験結果

著者らはReasoning Gymの未学習21課題群で、単発修正の平均得点が訓練前0.32から強化学習後0.62へ上がり、教師の0.56を上回ったと報告する。これは最良候補だけでなく、提案全体の平均である。

複数段階の検索を要するQAではHotpotQAで訓練し、MuSiQueと2WikiMultihopQAへ転移を評価した。MuSiQueの完全一致率は独立修正の平均0.15に対し、10回の反復修正後0.27だった。各反復実行は8候補を10回生成し、採点失敗の候補は0点とする。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:複数の社内文書をたどる問い合わせ対応の検索手順を改善する。入力は採点可能な開発用質問、現在の検索プログラム、取得文書や誤答を含む実行記録で、出力は修正済みプログラムとする。

例えば、検索を続けるための要約と、回答を作るための原文を別々に渡す構成を候補にする。開発問題で修正を選び、未使用の質問で確認してから業務へ反映する。これは論文の社内導入実績ではなく応用案である。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:正答と根拠文書を確認できる少数の質問を、改善用と評価用に分ける。既存の検索構成を基準に、同じ解答モデルで未修正、単発修正、反復修正を比較する。

回答正解率、根拠文書の取得率、実行失敗率、質問当たりの呼び出し回数と所要時間を記録する。候補数だけでなく総実行予算もそろえ、開発得点が上がっても未使用質問の得点が落ちる変更を見分ける。

限界と課題

著者らは、固定の解答モデルと所定の修正指示を使い、推論課題の教師も一つに限る点を挙げる。また、複数段階の修正状態で訓練しても、単発修正の訓練を一貫して上回らなかった。

編集上の確認事項は、採点可能な開発質問をどれだけ用意できるか、反復修正がその質問へ過適合しないか、追加実行の費用に見合うかである。QAの比較は候補数をそろえているが、総計算量とフィードバック量は一致しない。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

Harness Learning Enables Generalizable Test-Time Adaptation

A language-model agent is jointly defined by its model and its harness, the executable program that organizes model calls, tool use, and information flow. Because different tasks call for different ways of organizing these operations, the harness needs to be adapted using feedback from the task at hand. We introduce harness learning, which trains a proposer model to revise a solver's harness using execution feedback. We formulate this process as meta-learning over executable programs, with harness revisions playing the role of weight updates in gradient-based adaptation. We train the proposer with reinforcement learning, using the task performance of revised harnesses as the reward. At test time, the proposer uses feedback from successive executions on a new task to refine the harness, without performing any parameter-space update. Experiments on reasoning and multi-hop question answering show that harness learning improves revision quality and that the ability to adapt at test time transfers to unseen tasks. Policies trained on individual revisions can continue improving harnesses over multiple rounds, while the benefits of training on revision sequences vary across settings. These findings suggest a path towards continually learning agents that turn accumulated experience into generalizable improvements.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-09-29