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

トークン数でなく情報密度に沿って言語モデルを強化学習

Information-Time Proximal Policy Optimization

Yongcheng Zeng, Xinyu Cui, Yan Song, Guoqing Liu, Hongsheng Xin, Kaike Zhang, Cheng Deng, Kun Zhan, Jian Ying, Jian Zhao, Haifeng Zhang, Jun Wang

この論文をやさしく読む

ひとことで言うと

文章中の各トークンを同じ一歩として数える代わりに、その場所が持つ情報量に合わせて学習時の時間と更新幅を調整します。長い数学推論の最後で得た報酬が、途中の学習に届きにくくなる問題に取り組んでいます。

何に役立つ?

検算できる答えを使って言語モデルを強化学習する際、長い応答でも割引を利用しやすくする方法として役立ちます。実験で確認された用途はQwen3による数学推論です。

この研究の面白いところ

報酬をさかのぼって伝える仕組みと、方策を一度にどれだけ変えるかという制約を、情報密度という共通の量から導いています。実装の工夫だけでなく方策改善の下界も解析しています。

どこまで分かった?

理論的な下界は情報時間MDPと所定の更新制約に基づきます。実験は5つの数学ベンチマークで、要旨には改善幅の数値や数学以外の課題への検証は示されていません。

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

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

検証可能な報酬による強化学習(RLVR)は、大規模言語モデル(LLM)の推論能力を大幅に向上させてきた。しかし既存の方法は、自己回帰的な生成軌跡に沿う情報の流れが著しく不均一であるにもかかわらず、通常、マルコフ決定過程(MDP)における時間の進行をトークンごとの生成でパラメーター化している。本論文では、生のトークン数ではなく情報密度を用いて時間の進行を再パラメーター化するInfoPPOを提案する。 この再パラメーター化は、時間方向の貢献度の伝播と方策更新の双方に共通する、状態に依存した構造をもたらす。InfoPPOは、長い推論過程において、自明でない割引を有効に機能させる。実効的な見通し期間を短縮する効果を保ちながら、長いトークン列にわたって終端の教師信号が過度に減衰することを避ける。さらに、情報時間に基づく方策改善の解析から状態依存の更新制約が自然に導かれ、これを適応的クリッピングで実装する。各トークン位置のクリッピング閾値を、その位置の状態に対応する情報密度へ適応させることで、方策の変化を近傍に制御しつつ、より的を絞った更新を可能にする。 理論面では、性能差と方策改善の解析を情報時間MDPへ拡張し、方策の変更を情報密度によって調整するときの方策改善の下界を導く。また、状態ごとの情報密度と局所的な方策の変動を関係づけることで、一般的な情報時間の解析を実際のLLMの方策最適化へ結びつけ、適応的更新機構に理論的根拠を与える。Qwen3モデルを用いた実験では、難易度の高い競技型数学推論の5つのベンチマークで、有力な比較手法に対する一貫した改善を示した。トークン時間によるPPOが劣化する自明でない割引設定の範囲でも、InfoPPOは安定した精度と応答長を維持する。

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

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

更新履歴

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

原文の要旨

RLVR has substantially improved the reasoning capabilities of LLMs. However, existing methods typically parameterize temporal progression in the Markov Decision Process by token-by-token generation, despite the highly non-uniform information flow along autoregressive trajectories. In this paper, we propose InfoPPO, which reparameterizes temporal progression using information density rather than raw token count. This reparameterization induces a common state-dependent structure for both temporal credit propagation and policy updates. InfoPPO restores the effectiveness of non-trivial discounting in long-horizon reasoning, retaining effective-horizon contraction while avoiding excessive attenuation of terminal supervision over long token sequences. Moreover, the information-time policy-improvement analysis naturally leads to a state-dependent update constraint, which we implement through adaptive clipping. By adapting the clipping threshold at each token position to the information density of its corresponding state, this mechanism enables more targeted policy updates while preserving proximal control. Theoretically, we extend performance-difference and policy-improvement analyses to the information-time MDP, deriving a policy-improvement lower bound when policy changes are regulated by information density. We further connect the general information-time analysis to practical LLM policy optimization by relating state-wise information density to local policy movement, while also providing theoretical grounding for the adaptive update mechanism. Experiments on Qwen3 models demonstrate consistent gains over competitive baselines across five challenging competition-style mathematical reasoning benchmarks. InfoPPO also maintains stable accuracy and response length across non-trivial discount settings under which token-time PPO deteriorates.

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