AI・機械学習
中級

報酬モデル(Reward Model)(ホウシュウモデル)

RLHFにおいて人間の選好を数値スコアに変換する機械学習モデル。人間アノテータの比較判定データから訓練され、強化学習の報酬信号として機能する。

0 回閲覧
0 いいね

報酬モデルとは

報酬モデル(Reward Model、RM)は、RLHFパイプラインにおいて「人間が好む応答とはどのようなものか」を数値化するモデルである。人間アノテータが付けた応答間の選好ランキングを学習し、任意のプロンプト-応答ペアに対してスカラー報酬値を出力する。この報酬値がPPOやGRPOなどの強化学習アルゴリズムの最適化対象となる。

報酬モデルの訓練

データ形式

報酬モデルの訓練データは、同一プロンプトに対する2つ以上の応答を人間がランキングした選好データである。

データ形式説明主な使用先
対比較(Pairwise)AとBのどちらが良いかInstructGPT, Claude
K-wise rankingK個の応答を順位付けGPT-4, Gemini
スカラー評価1-5点の絶対評価一部の学術研究

対比較形式が最も広く使われる理由は、人間にとって「AとBどちらが良いか」という相対判断が、「この応答は5点中何点か」という絶対判断より一貫性が高いためである。

Bradley-Terryモデル

最も標準的な訓練手法はBradley-Terryモデルに基づく。応答 $y_w$(勝者)と $y_l$(敗者)に対し、以下の損失関数を最小化する:

L = -log(σ(r(x, y_w) - r(x, y_l)))

ここで r(x, y) は報酬モデルの出力、σ はシグモイド関数である。

報酬モデルの種類

結果ベース報酬モデル(Outcome RM)

応答全体に対して単一のスコアを出力する。最も一般的な形式。

プロセス報酬モデル(Process RM)

推論の各ステップに対して個別にスコアを付与する。OpenAIの「Let's Verify Step by Step」論文で提案され、数学的推論タスクで結果ベースRMを大幅に上回る性能を示した。

報酬モデル種別粒度精度コスト適用先
Outcome RM応答全体標準低一般的な対話
Process RMステップ毎高高(ステップ注釈必要)数学・推論

Generative RM(LLM-as-Judge)

専用の報酬モデルを訓練する代わりに、GPT-4やClaude等の高性能LLMに応答を評価させる手法。RLAIF(RL from AI Feedback)の基盤技術でもある。

報酬ハッキングの問題

報酬モデルには固有の弱点があり、方策モデルがその弱点を学習して「報酬は高いが実際には低品質な応答」を生成する現象が起きる。

代表的なハッキングパターン:

  • 冗長性バイアス: 長い応答ほど高スコアを得る傾向を悪用
  • 形式バイアス: 箇条書きや番号付きリストを過度に使用
  • 自信過剰バイアス: 不確実な回答でも断定的に述べる

対策として、KLペナルティ(参照モデルからの乖離制限)、報酬モデルのアンサンブル、定期的な再訓練が有効である。

報酬モデルの規模と性能

InstructGPTでは6Bパラメータの報酬モデルが使用された。一般にベースモデルと同等かやや小さいサイズの報酬モデルが使用されるが、最近の研究ではモデルサイズよりも訓練データの質と量が性能に支配的であることが示されている。

FAQ

Q: 報酬モデルの訓練データはどのくらい必要か?

A: InstructGPTでは約33,000件の比較データが使用された。実用的には10,000-50,000件の高品質な選好データがあれば安定した報酬モデルを訓練できる。データ品質(アノテータの一致率)が量より重要である。

Q: 報酬モデルの評価方法は?

A: 主な指標はアノテータとの一致率(accuracy on held-out comparisons)である。InstructGPTの6B報酬モデルは約72%の一致率を達成した。また、報酬モデルのスコアと人間評価の相関(Spearman ρ)も補助指標として使われる。

Q: 1つの報酬モデルで全タスクをカバーできるか?

A: 汎用報酬モデルは対話・要約・コーディング等の多タスクで合理的な性能を発揮するが、数学やコーディングなど正解が明確なドメインでは、タスク特化型RMが汎用RMを大幅に上回る。実務ではドメイン特化RMと汎用RMのアンサンブルが有効である。