arXiv論文メモ
新着一覧
cs.LG / cs.AI · 査読状況未確認

トークンごとの寄与を定義して批評器を整えるPACT

PACT: From Credit Assignment to Critic Alignment

Jiayan Fu, Hang Xu, Yong Zhang, Zhaokai Luo, Yao Hu, Dongyan Zhao, Mu Chuan

この論文をやさしく読む

ひとことで言うと

LLMの強化学習で各トークンの寄与を数学的に定め、更新後の方策に合う批評器の学習法を提案した。

何に役立つ?

LLMの強化学習で報酬を各トークンへ割り当て、批評器の誤差を扱う学習法を考える材料になる。

この研究の面白いところ

三条件による一意な定義から既存のOPDやRLOOの信号を説明し、その分析をPACTの更新順序に結びつける。

どこまで分かった?

報告された正解率・通過率は示された数学課題とSWE-bench Verifiedでの比較であり、すべての課題での優位性を示すものではない。

v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。

アブストラクトの日本語訳

強化学習は大規模言語モデル(LLM)の学習後調整の中心になっているが、トークン単位の寄与には広く認められた数学的定義がなく、よく使われる学習信号との関係も不明確である。本研究は、完全性、接頭辞の整合性、中立性という三つの正則性条件を定め、それらがトークン単位の寄与を一意に決めることを証明する。この特徴づけは既存アルゴリズムで見られる現象を統一的に説明する基盤となり、改良したアクター・クリティック学習手順の開発を導く。この観点から、オンポリシー蒸留(OPD)の理想的な教師は暗黙の批評器として働き、トークン単位の寄与が誘導するものに比例した期待方策勾配を生む。また、応答単位のREINFORCE Leave-One-Out(RLOO)の信号は、粒度は粗いものの、トークン単位の寄与の期待方策勾配への貢献と一致する。さらに、結果の報酬に上限があるときの寄与の近似的な疎性を確立し、一般化優位性推定(GAE)における中間の批評器の誤差が、基礎となる寄与と同程度になり得ることを示す。これらに基づきPolicy Aligned Critic Training(PACT)を提案する。PACTは、まずアクターを、次に批評器を更新する順序を採り、批評器の学習に重要度サンプリングによる補正を適用して、更新後の方策との整合性を高める。エージェントによる数学的推論では、四つのベンチマークで平均正解率72.87%に達し、GRPOとPPOをそれぞれ8.80、13.16ポイント上回った。SWE-bench Verifiedでは通過率67.4%に達し、PPO、GRPO、SAOをそれぞれ2.4、2.0、3.8ポイント上回った。

v1の要旨から自動生成。本文の精読・人による確認は未実施。

初稿
2026-09-22(UTC)
最新改訂
2026-09-22 · v1
査読・掲載
査読状況未確認
arXivで読むPDF

更新履歴

取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。

原文の要旨

Reinforcement learning has become a central component of large language model (LLM) post-training, yet token-level credit lacks a generally accepted mathematical definition, leaving its relationship to commonly used training signals unclear. We formulate three regularity conditions, namely Completeness, Prefix Consistency, and Neutrality, and prove that they uniquely determine token-level credit. This characterization provides a unified basis for explaining phenomena across existing algorithms and guides the development of an improved actor-critic training procedure. Through this lens, an ideal teacher in On-Policy Distillation (OPD) acts as an implicit critic, yielding an expected policy gradient proportional to that induced by token-level credit. Response-level REINFORCE Leave-One-Out (RLOO) signals match the expected policy-gradient contribution of token-level credit despite their coarser granularity. We further establish approximate credit sparsity under bounded outcome rewards and show how intermediate critic errors in Generalized Advantage Estimation (GAE) can become comparable to the underlying credit. These motivate Policy Aligned Critic Training (PACT), which adopts an Actor-then-Critic update order to apply importance sampling correction to critic training and better align the critic with the updated policy. In agentic mathematical reasoning, PACT achieves 72.87% average accuracy across four benchmarks, outperforming GRPO and PPO by 8.80 and 13.16 percentage points, respectively. On SWE-bench Verified, PACT achieves a pass rate of 67.4%, outperforming PPO, GRPO, and SAO by 2.4, 2.0, and 3.8 percentage points, respectively.

arXiv ID: 2609.26355 / 要約の誤りについて