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

プロセス報酬誘導型適応ツリーロールアウトによる効率的なマルチターン強化学習

記事の要約

PATRを提案。タスクに適したプロセスフィードバックを用いて部分軌跡をスコアリングし、有望な状態から選択的に分岐、共有プレフィックスを再利用、劣化パスを停止することでサンプリングの無駄を削減。FrozenLakeとSWE-Benchでそれぞれ9.3ポイント、5.0ポイントの向上。

ソースarXiv Computational Linguistics著者: Xintong Li, Sha Li, Yuwei Zhang, Changlong Yu, Rongmei Lin, Hongye Jin, Shuyi Guan, Xin Liu, Linwei Li, Qingyu Yin, Jingbo Shang
プロセス報酬誘導型適応ツリーロールアウトによる効率的なマルチターン強化学習
誤りを報告

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

訂正案内
本文へ

強化学習(RL)は大規模言語モデル(LLM)エージェントの訓練における主要な手法となっているが、GRPO/RLOOなどの一般的な手法は、複数の独立してサンプリングされた完全な軌跡を用いてアドバンテージ推定を行う。長期にわたるエージェントタスクでは、このような一様なロールアウト戦略は、情報のない行き止まりの試行に予算を浪費し、有望な中間状態が十分に探索されない可能性がある。エージェントの軌跡は多ターン構造(アクションと観察が交互に現れる)を持つため、軌跡グループをツリーとして整理し、各ターンを探索の意思決定点と見なすことが自然である。この視点は、効果的な探索を「どこで分岐するか」という問題に再構成する。

本研究では、プロセススコアラー誘導型適応ツリーロールアウト(PATR)を提案する。これはマルチターンエージェントRLのための品質認識ロールアウトフレームワークである。PATRはタスク適切なプロセスフィードバックを用いて部分軌跡をスコアリングし、有望な状態から選択的に分岐し、共有プレフィックスを再利用し、劣化パスを保守的に停止することで無駄なサンプリングを削減する。具体的には、PATRはプロセススコアラー(ルールベースまたは学習済みモデル)を使用して各部分軌跡にスコアを割り当て、そのスコアに基づいてその状態から展開を続けるか停止するかを決定する。共有プレフィックスにより、複数の分岐が既に実行された同じアクションシーケンスを再利用でき、重複計算を避ける。劣化パスの停止メカニズムは、無効な方向へのリソース浪費を防ぐ。得られたロールアウトグループは標準的なポリシー最適化(PPOやGRPOなど)と互換性を保ちつつ、同じ訓練予算内でより効率的な探索を提供する。

評価はFrozenLakeと困難なSWE-Benchで行われた。FrozenLakeは古典的なグリッドワールドナビゲーションタスクであり、SWE-Benchは実際のGitHub問題に基づくソフトウェア工学ベンチマークで、これまでツリーロールアウトエージェントRL手法ではほとんど未探索だった。実験の結果、PATRはSWE-Benchで最大5.0ポイント、FrozenLakeで9.3ポイントの性能向上を示し、プロセス誘導型ツリーロールアウトがスケーラブルなマルチターンRLの効果的な戦略であることを明らかにした。この研究はエージェント訓練のサンプル効率向上に新たな方向性を開き、より複雑な実世界タスクへの応用が期待される。

要点と分析を開く

記事インテリジェンス

エンジニア中級

要点

  • GRPO/RLOOなどの既存手法は一様なロールアウト戦略を採用し、無情報な行き止まり試行に予算を浪費。
  • PATRはプロセスフィードバックで部分軌跡を評価し、有望な状態から分岐、共有プレフィックスを再利用、劣化パスを停止。
  • 実験でFrozenLakeで9.3ポイント、SWE-Benchで5.0ポイントの性能向上を達成。

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