AI News JAPAN

世界のAIニュースを最速で把握できるメディア

Advertisement

AIエージェントのRL訓練を33%高速化・性能15%向上、新手法「SAPO」をarXivで公開

30秒サマリー

  • LLMのエージェント強化学習に特化した新フレームワーク「SAPO」が論文として公開された
  • 別途クリティックモデルが不要で、PPO比で実行時間を33.2%削減しながら性能は平均15.1ポイント向上
  • 小規模モデル(1.5B/7B)での検証であり、大規模実装への汎化は今後の課題とみられる

何が起きたか

Dayang Liangら4名の研究者が2026年8月20日、arXiv(cs.AI)に論文「SAPO: Single-Rollout Autoregressive Policy Optimization for Agentic Reinforcement Learning」を投稿した。

SAPOは、LLMのポスト訓練において注目されるエージェント強化学習(Agentic RL)向けのフレームワーク。既存の代表的手法であるGRPO(グループ相対ポリシー最適化)は複数のロールアウトを必要とし、長期タスクでの「アドバンテージ崩壊」やサンプリングコストと性能のトレードオフという課題を抱えていた。SAPOはLLMの自己回帰構造を活用し、ポリシー関数と価値関数を単一のモデルバックボーンで共有することで、別途クリティックモデルを不要とする。

論文によると、ALFWorldおよびWebShopの2タスクにおいてQwen2.5-1.5Bと7Bモデルを用いた実験を実施。PPO比で平均+15.1ポイント、GRPO比で平均+12.1ポイントの性能向上を達成した。またPPOと比較して反復あたりの実行時間を33.2%削減できるとしている。さらに、各ターンの貢献を安定的に推定するため、ラムダリターンとバッチ正規化を組み合わせた「軌跡レベル汎化アドバンテージ推定器」を導入している。

原典ハイライト

論文アブストラクトは「SAPOはLLMの自己回帰構造を活用し、ポリシーと価値予測を共有パラメータで行いながら、PPO目標とSARSA目標を独立最適化する」と説明。ALFWorld・WebShopでのPPO比+15.1ポイント・実行時間33.2%削減という数値を明記している。

出典: arXiv cs.AI(論文)

So What?(なぜ重要か)

エージェントAIの強化学習訓練は計算コストと性能のトレードオフが長年の課題だった。SAPOがクリティックモデルを廃止しながら性能と速度を同時に改善できることを示した点は、訓練インフラのコスト構造を変える可能性を持つ。ただし本論文はarXivへの投稿段階であり、査読を経た検証はこれからとみられる。

日本企業への示唆

AIエージェント開発に取り組む日本企業にとって、訓練コストの削減は導入障壁を下げる直接的な要因となる。SAPOのアーキテクチャはクリティックモデル分のGPUメモリを節約できるため、限られたオンプレミスGPUリソースでの自社モデル訓練に適用可能かどうか、技術チームが検討する価値がある。一方、実験は1.5B・7Bの比較的小規模なモデルに限られており、数十〜数百億パラメータ規模への適用可能性は原文では言及がない点に留意が必要だ。

背景・経緯

LLMのポスト訓練においてRLHFやGRPOなどの強化学習手法が広く活用されており、特に複数ステップのタスクをこなす「エージェント」用途への応用が活発化している。クリティックモデルを別途用意するPPOはメモリ負荷が高く、クリティック不要なGRPOは長期タスクでの安定性に課題があるとされてきた。SAPOはこの両者の弱点を補う位置づけとして提案されている。