本文にスキップ
AI News HubLIVE
サイト内リライト1 分で読了

ピアの成功と失敗を活用したマルチロールアウト・オン・ポリシー蒸留

記事の要約

MOPDフレームワークを提案。学生モデルの複数の生成軌跡(成功と失敗の両方)を利用して教師信号を強化し、標準的なオン・ポリシー蒸留手法を凌駕する。実験では、成功と失敗を混合したコンテキストが検証報酬により正確に一致する教師スコアを生成することを示した。

ソースarXiv Machine Learning著者: Weichen Yu, Xiaomin Li, Yizhou Zhao, Xiaoze Liu, Ruowang Zhang, Haixin Wang, Yinyi Luo, Chen Henry Wu, Gaurav Mittal, Matt Fredrikson, Yu Hu
ピアの成功と失敗を活用したマルチロールアウト・オン・ポリシー蒸留
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

大規模言語モデルの事後訓練では、スパースな検証報酬がよく用いられる。これはサンプリングされた軌跡の成功可否のみを示し、推論のどの部分が成功または失敗したかについての限定的な指針しか提供しない。オン・ポリシー蒸留(OPD)は学生自身が生成した軌跡で学習することで、より密なトークンレベルの教師信号を提供するが、既存手法は各軌跡を独立に蒸留し、同一プロンプトに対してサンプリングされた他の試行を無視する。

本研究では、マルチロールアウト・オン・ポリシー蒸留(MOPD)を提案する。MOPDは学生のローカルなロールアウトグループを活用して、より情報豊富な教師信号を構築するピア条件付き蒸留フレームワークである。成功したピアの軌跡は有効な推論パターンの肯定的証拠を提供し、失敗した軌跡は構造化された否定的証拠を提供することで、モデルが一般的な誤りを回避するのに役立つ。

研究では2種類のピアコンテキスト構築法を検討した:正の模倣(positive peer imitation)と対照的成功-失敗条件付け(contrastive success-failure conditioning)。競技プログラミング、数学的推論、科学質問応答、ツール使用のベンチマーク実験において、MOPDは標準的なオン・ポリシーベースラインを一貫して上回った。さらに教師信号の分析から、成功と失敗を混合したコンテキストが教師スコアと検証報酬をより良く整合させることが分かり、性能向上がより忠実でインスタンス適応的な監督に起因することを示している。

これらの結果は、効果的なオン・ポリシー蒸留には、ロールアウトを孤立したサンプルとして扱うのではなく、学生のマルチロールアウト試行錯誤行動を活用すべきであることを示唆している。本論文はWeichen Yuら10名の著者によるもので、23ページにわたり詳細な実験と分析が報告されている。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • MOPDは同一プロンプトに対する複数の試行(成功と失敗)を活用して蒸留を改善する。
  • 2つのピアコンテキスト構築法を研究:正の模倣と対照的成功-失敗条件付け。
  • プログラミング、数学的推論、科学質問応答、ツール使用のベンチマークで標準ベースラインを上回る。
  • 成功と失敗の混合コンテキストはより忠実でインスタンス適応的な教師信号を生成する。

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。