小型の検索エージェントに適した報酬設計
Reinforcement Learning with Verifiable Rewards for Small Search Agents
この論文をやさしく読む
ひとことで言うと
検索しながら答える小型モデルで、強化学習の報酬の形が結果にどう影響するか調べた。
何に役立つ?
小型の質問応答エージェントを学習させる際、報酬設計を選ぶ参考になる。
この研究の面白いところ
0.8Bモデルで蒸留なしに平均完全一致率を0.092から最良0.352へ高めたが、完全一致だけの報酬は三方式中最下位だった。
どこまで分かった?
結果はMuSiQueでの訓練と七つの未使用ベンチマーク、各報酬三乱数種での比較に基づく。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
検証可能な報酬を用いる強化学習(RLVR)は数学やコード作成のように報酬が明確な課題で良い結果を示すが、報酬がそれほど明確でない課題にも有効かは未解決だった。検索しながら推論する方法は、検索で回答の根拠を得て、参照回答との一致を報酬にすることで、RLVRを幅広い分野の質問応答に適用する。これまで大規模モデルでは実証されていたが、10億パラメータ未満では大きな教師モデルからの蒸留を使った例だけだった。本研究は小型モデルでこの方法を検証する。 Qwen3.5-0.8Bを、Wikipedia検索ツールを途中で使えるようにして、MuSiQueデータセット上でGroup Relative Policy Optimization(GRPO)により訓練した。変えた条件は報酬の形だけで、三種類についてそれぞれ三つの乱数種を用い、全てのチェックポイントを、学習には使わない七つの質問応答ベンチマークで評価した。最良の実行では平均完全一致率0.352に達し、未訓練時の0.092に対して3.8倍となった。学習の途中に蒸留は使っていない。報酬の形も重要だった。Search-R1に忠実な、完全一致だけを報酬にする方式は、訓練期間をそろえると全ての乱数種で三方式中最も悪く、直接最適化している完全一致率でも最下位だった。数学やコード作成のRLVRで標準的な疎な完全一致報酬は、この規模のモデルの出発点として不適切だと結論付ける。検索しながら推論する設定なら小型モデル向けの適切な報酬を与えられるが、大型モデル用の方法を単に縮小するのではなく、小型モデル固有の報酬設計を調べる必要がある。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-23(UTC)
- 最新改訂
- 2026-09-23 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-23 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Reinforcement Learning with Verifiable Rewards (RLVR) performs well on problems with clear rewards, such as mathematics and coding, but whether it also works where the reward is less clear remains open. The reason-over-search recipe applies RLVR to open-domain question answering, where retrieval grounds the answer and a match against the reference supplies the reward. So far it has been demonstrated on large models, and below one billion parameters only with distillation from a larger teacher. We test the recipe on a small model. We train Qwen3.5-0.8B with Group Relative Policy Optimization (GRPO) and an interleaved Wikipedia-search tool on MuSiQue, varying only the reward across three shapes over three seeds each, and we evaluate every checkpoint held-out on a seven-benchmark question-answering suite. The recipe works: the best run reaches 0.352 average exact match against a 0.092 untrained floor, a 3.8-fold gain, with no distillation step in the training loop. The reward shape also matters. The Search-R1-faithful exact-match-only reward is the worst of the three at every seed at the matched training horizon, and it is worst even on exact match, the metric it directly optimises. We conclude that the sparse exact-match reward, RLVR's default in mathematics and code, is the wrong starting point for models of this size. The reason-over-search setting can supply a suitable reward for RLVR on small models, but small-model RLVR needs its own reward-design study rather than a scaled-down copy of a large-model recipe.
arXiv ID: 2609.28765 / 要約の誤りについて