AI・機械学習
上級

OpenAI PRM800K(オープンエーアイピーアールエムハッピャクマンケー)

OpenAIが2023年に公開した数学推論ステップの人手アノテーションデータセット。75万以上の解法ステップに「正しい」「中立」「誤り」の3クラスラベルを付与し、Process Reward Modelの研究基盤となった。

0 回閲覧
0 いいね
2026/6/6 更新
関連タグ
PRM800K
OpenAI
数学推論
データセット
ステップアノテーション
リワードモデル

OpenAI PRM800Kとは

PRM800Kは、OpenAIが2023年の論文「Let's Verify Step by Step」と共に公開した数学推論評価データセットです。75万以上の推論ステップに人手ラベルが付与されており、Process Reward Model(PRM)研究の出発点となりました。

データセットの構成

規模とラベル

  • 総ステップ数: 約800,000(約75万)
  • 問題元: MATHデータセット(競技数学問題集)
  • ラベルクラス: 3クラス(positive/neutral/negative)
ラベル意味割合(概算)
positiveこのステップは正しく有効~60%
neutralこのステップは判断保留~10%
negativeこのステップには誤りがある~30%

生成プロセス

  1. MATHデータセットの問題をGPT-4に解かせる
  2. GPT-4が生成したCoT(Chain-of-Thought)を各ステップに分割
  3. 人間アノテーターが各ステップを3クラスで評価
  4. 品質管理のため複数アノテーターでクロスチェック

論文の核心的発見

PRM vs ORM 比較実験

OpenAIはPRM800Kで訓練したPRMと、従来のORM(最終答えのみ評価)を比較しました:

MATH-500ベンチマーク(Best-of-N選択)

サンプル数NORM精度PRM精度PRM優位幅
N=149.9%49.9%±0
N=456.2%61.4%+5.2%
N=1660.1%68.9%+8.8%
N=25663.5%78.2%+14.7%

Nが増えるほどPRMの優位性が拡大する点が重要で、推論時計算スケーリングの有効性を示しています。

データ公開とライセンス

PRM800KはMITライセンスでGitHubに公開されており、研究者が自由に利用できます。ただし:

  • 商用利用前にOpenAIの利用規約確認が推奨
  • MATHデータセット自体のライセンスにも依存
  • GPT-4生成データを含むため、一部AI利用規約の考慮が必要

後続研究への影響

PRM800Kの公開は数学推論LLM研究に大きな影響を与えました:

  • Math-Shepherd(2024): 人手不要の自動アノテーション手法を開発
  • Skywork-o1-PRM(2024): PRM800Kで初期訓練→ドメイン適応
  • OmegaPRM(DeepMind, 2024): MCTSを用いた高品質版PRM
  • DeepSeek-R1(2025): 類似のPRM思想で数学推論を強化

限界と批判

  1. 規模の限界: 75万ステップはMathの多様性には十分でない部分もある
  2. GPT-4依存: 元のCoTがGPT-4生成のため、GPT-4の偏りを引き継ぐ
  3. MATHドメイン特化: 他の推論タスク(コーディング等)への汎化は不明
  4. コスト: 人手アノテーションの再現には膨大な費用が必要

これらの限界を克服するため、後続研究は自動アノテーションへと発展しました。

この記事について
カテゴリーAI・機械学習
難易度上級
作成日2026/6/6