LLMの強化学習について

AI / LLM Machine Learning 論文解説: arXiv 2305.18290 2025.02.17
株式会社アプリズムのテック記事になります。

大規模言語モデル(LLM)の精度と安全性を向上させる鍵となる強化学習(Reinforcement Learning)。ChatGPTやClaudeが「人間らしい」応答を返せる理由はここにあります。本記事では、RLHFの基礎から最新手法まで、Stanford大学の論文(DPO / NeurIPS 2023)を交えてエンジニア目線で丁寧に解説します。

01

そもそも「強化学習」とは何か

機械学習の一分野である強化学習(Reinforcement Learning: RL)は、エージェントが環境と相互作用しながら試行錯誤を繰り返し、累積報酬を最大化するような行動方策を学習するアプローチです。ゲームAIや自律ロボット制御の分野で長年研究されてきましたが、近年はLLMのファインチューニングに応用され、その重要性が急速に高まっています。

強化学習の基本構造は「状態 → 行動 → 報酬」のループです。LLMに当てはめると、ユーザーの入力テキスト(状態)に対してトークンを生成し(行動)、その応答品質に対してスコアを受け取る(報酬)という形式になります。

💡 ポイント

LLMに強化学習を適用する目的は「人間の価値観や好みに沿った出力を生成させること」です。事前学習(Pretraining)だけでは、有害なコンテンツや不誠実な回答が生成されるリスクがあり、これを抑制するのが強化学習の役割です。

02

RLHFの仕組みを理解する

RLHF(Reinforcement Learning from Human Feedback)は、現在最も広く採用されているLLM調整手法です。OpenAIがInstructGPTで提唱し、ChatGPTでも採用されています。RLHFは主に3つのフェーズで構成されます。

01

教師あり微調整(SFT: Supervised Fine-Tuning)

人間が作成した高品質な(プロンプト, 応答)ペアを用いて、ベースモデルを微調整します。これにより、指示に従う基本的な能力を獲得させます。

02

報酬モデルの学習(Reward Model Training)

複数の応答候補に対して人間のランカーが優劣をつけ、そのデータを元に「どの応答が良いか」を判定する報酬モデル(RM)を学習させます。このRMがRL中の「審判」として機能します。

03

PPOによる強化学習(RL with PPO)

Proximal Policy Optimization(PPO)アルゴリズムを用いて、報酬モデルのスコアを最大化するようにSFTモデルを更新します。KLダイバージェンス制約により、元のモデルからの過度な逸脱を防ぎます。

PYTHON · RLHF 報酬計算(論文 Eq.3)
# RLHF の RL ファインチューニング目標(論文 Eq.3):
# max_πθ E[rϕ(x,y)] - β·KL[πθ(y|x) || πref(y|x)]
def compute_rlhf_reward(response, prompt, reward_model, policy, ref_policy, beta=0.1):
    rm_score   = reward_model(prompt, response)
    log_prob_new = policy.log_prob(response, prompt)
    log_prob_ref = ref_policy.log_prob(response, prompt)
    kl_penalty   = beta * (log_prob_new - log_prob_ref)
    return rm_score - kl_penalty

for step in range(training_steps):
    prompts   = sample_batch(dataset)
    responses = policy.generate(prompts)
    rewards   = [compute_rlhf_reward(r, p, rm, policy, ref, beta=0.1)
                 for r, p in zip(responses, prompts)]
    policy.ppo_update(prompts, responses, rewards)
03

RLHFの課題と代替アプローチ

RLHFは強力な手法ですが、実装の複雑さや計算コスト面での課題もあります。特にPPOの不安定性報酬モデルの過最適化(Reward Hacking)は実務上の大きな障壁です。PPOはファインチューニング中にLMからのサンプリングをループ内で行う必要があり、ポリシー・報酬モデル・参照モデル・クリティックの4モデルを同時に保持するため、計算コストが著しく膨大になります。

🎯

DPO NeurIPS 2023

報酬モデルを陽に学習せず、単純な二値クロスエントロピー損失だけでポリシーを直接最適化。PPOと同等以上の性能を達成。

🔄

RLHF (PPO) Classic

最も実績のある手法。複雑だが柔軟性が高く、大規模モデルでの性能が実証されている。報酬モデル学習とRL最適化の2段階パイプライン。

RLAIF

RL from AI Feedback。人間の代わりにAI(別LLM)が報酬シグナルを提供。スケーラブルで高速。Constitutional AIでも採用。

DPOの核心的アイデア

2023年にStanford大学がNeurIPS 2023で発表したDPO(論文:「Direct Preference Optimization: Your Language Model is Secretly a Reward Model」)は、LLMへの強化学習適用における根本的な発想の転換です。タイトルにある通り、「言語モデル自体が暗黙的な報酬モデルである」という洞察がその核心です。

RLHFでは「報酬モデルの学習 → PPOによるポリシー最適化」という2段階が必要でした。DPOはKL制約付き報酬最大化の最適解が閉形式で導出できるという数学的事実を活用し、この2段階を単一の分類損失に変換することに成功しています。

🔬 数学的直感

Bradley-Terry モデルは選好確率を報酬差分の関数として定義します。DPO はこの関係を逆用し、報酬関数を「ポリシーの対数確率比 β·log(π(y|x) / π_ref(y|x))」として表現します。この変数変換によりパーティション関数が消え、報酬モデルを陽に学習することなく、ポリシー自体を損失関数の引数として直接最適化できます。

PYTHON · DPO Loss — 論文 Eq.(7)
import torch.nn.functional as F

def dpo_loss(policy, ref_policy, chosen, rejected, prompt, beta=0.1):
    """
    L_DPO = -E[log σ(β·log(π_θ(yw|x)/π_ref(yw|x)) - β·log(π_θ(yl|x)/π_ref(yl|x)))]
    chosen(yw): 人間が好む応答 / rejected(yl): 人間が好まない応答
    """
    log_p_chosen   = policy.log_prob(chosen,   prompt)
    log_p_rejected = policy.log_prob(rejected, prompt)
    log_r_chosen   = ref_policy.log_prob(chosen,   prompt)
    log_r_rejected = ref_policy.log_prob(rejected, prompt)

    # 暗黙の報酬差分: r̂(x,yw) - r̂(x,yl)
    logits = beta * (
        (log_p_chosen   - log_r_chosen) -
        (log_p_rejected - log_r_rejected)
    )
    return -F.logsigmoid(logits).mean()

DPO の勾配が意味すること

論文ではDPOの損失関数の勾配を解析的に分解しており、その挙動が明快に理解できます。勾配は大きく2項から成ります。第1項は好まれる応答 y_w の尤度を上げる方向、第2項は好まれない応答 y_l の尤度を下げる方向です。重要なのは各サンプルに付与される重み σ(r̂(y_l) − r̂(y_w)) で、暗黙の報酬モデルが選好順序を誤って推定しているサンプルほど、大きな重みで更新される仕組みです。この動的な重みづけが、素朴な確率比最適化(Unlikelihood法)で起きるモデル劣化を防ぐ鍵です。

PPO との実験比較(論文より)

論文では感情生成・要約・対話の3タスクで評価が行われています。感情生成(IMDb)では、DPOはすべてのKL値においてPPOよりも高い期待報酬を達成し、報酬/KLトレードオフのフロンティアでPPOを上回りました。要約タスク(Reddit TL;DR)では温度0でのDPOの勝率は約61%でPPOの57%を超え、さらにDPOはサンプリング温度の変化に対してPPOより頑健でした。対話タスク(Anthropic HH)では、DPOのみが選好済み応答より高いスコアを達成した唯一の計算効率的な手法です。

手法 学習の安定性 計算コスト 実装難易度 主な採用例
RLHF (PPO) △ やや不安定 高い(4モデル同時) 高い ChatGPT, InstructGPT
DPO ◯ 安定 低〜中(SFTと同等) 低い Llama 3, Zephyr, Mistral
RLAIF ◯ 安定 Claude (Constitutional AI)
IPO ◎ 非常に安定 低い 低い 研究段階
📄 参照論文

Rafailov et al., “Direct Preference Optimization: Your Language Model is Secretly a Reward Model”, NeurIPS 2023.(Stanford University) arxiv: 2305.18290

04

Constitutional AI と RLAIF

Anthropicが提唱するConstitutional AI(CAI)は、「原則(Constitution)」と呼ばれるルールセットを使ってAI自身に自己批判・修正を行わせ、安全で有用な応答を引き出す手法です。RLAIFでは、人間のフィードバックの代わりにAIモデルが「どちらの応答が優れているか」を評価し、それをシグナルとして強化学習を行います。

🏛️ Constitutional AI の原則例

「有害・危険・非倫理的・違法なコンテンツを含む応答を選ばないでください」「人間の自律性を尊重し、ユーザーが自ら判断できるよう支援してください」「不確かな情報については、確信を持っているように見せかけないでください」

このアプローチの最大の利点はスケーラビリティです。人間のアノテーターのボトルネックを解消しつつ、数万〜数百万件の選好データを自動生成できます。ただし、AIによる評価がベースモデルのバイアスを引き継ぐリスクも内包しており、適切な設計と検証が不可欠です。

05

実装における主要な課題

⚠️

Reward Hacking(報酬ハッキング)

モデルが報酬モデルの「穴」を突き、実際には品質が低いにもかかわらず高スコアを得る応答を生成する現象。例:長い応答を生成するほど高スコアになる報酬モデルを過学習させると、冗長な出力が増加する。

📊

アノテーションの品質と一貫性

人間のフィードバックは主観的であり、アノテーター間でのばらつき(Inter-annotator Agreement)が低い場合、報酬モデルが矛盾したシグナルを学習してしまう。明確なガイドラインとダブルチェックが必要。

💾

計算資源とインフラコスト

PPOベースのRLHFは、ポリシーモデル・報酬モデル・参照モデル・クリティックを同時に保持・更新するため、メモリ使用量が膨大。70Bパラメータのモデルでは複数のA100/H100が必要。

🔄

Distribution Shift(分布シフト)

RLの学習が進むにつれて、ポリシーが生成する応答の分布が、報酬モデルの学習データの分布から乖離していく問題。定期的な報酬モデルの再学習やオンライン学習で対応する必要がある。

06

最新トレンド:2025年の動向

GRPO Group Relative Policy Optimization
クリティック不要で効率化
SimPO Simple Preference Optimization
参照モデル不要の軽量手法
RLVR RL with Verifiable Rewards
検証可能タスクへの応用

GRPO(Group Relative Policy Optimization)

DeepSeekが採用したことで注目を集めたGRPOは、PPOのクリティックネットワークを排除し、グループ内の相対的な報酬を使ってポリシーを更新します。同一プロンプトから複数の応答をサンプリングし、グループ内での相対的な優劣を学習シグナルとするため、メモリ効率が大幅に向上します。

RLVR(RL with Verifiable Rewards)

数学問題の正答、コードのテストパス、論理的推論など、客観的に検証可能なタスクにおける強化学習アプローチです。人間や別AIの主観的評価を必要とせず、自動的に報酬を計算できます。DeepSeek-R1やQwQ-32Bなどの推論特化型モデルで採用されており、Chain-of-Thought(CoT)能力の飛躍的な向上が報告されています。

07

アプリズムの取り組みと今後の展望

株式会社アプリズムでは、LLMを活用したプロダクト開発において、ドメイン特化の強化学習パイプライン構築を積極的に探求しています。特にDPOによる低コスト・高品質な特化モデルの構築と、RLAIF を活用したスケーラブルなデータ生成に注力しています。

強化学習は「完成した技術」ではなく、現在進行形で急速に発展しています。GRPOやRLVRのような新手法が次々と登場し、より少ない計算コストで、より高い品質のモデルを作れる時代が来ています。自社プロダクトにLLMを組み込む際は、こうした最新動向を常にウォッチしながら、最適な手法を選択することが重要です。