AI Agents 相关度: 8/10

Rethinking the Trust Region in LLM Reinforcement Learning

Penghui Qi, Xiangxin Zhou, Zichen Liu, Tianyu Pang, Chao Du, Min Lin, Wee Sun Lee
arXiv: 2602.04879v1 发布: 2026-02-04 更新: 2026-02-04

AI 摘要

论文提出DPPO算法,通过直接估计策略差异来改进LLM强化学习中的PPO算法,提升训练稳定性和效率。

主要贡献

  • 提出 Divergence Proximal Policy Optimization (DPPO)算法
  • 使用策略差异的直接估计替代启发式裁剪
  • 引入 Binary 和 Top-K 近似以减少内存占用

方法论

提出基于策略差异估计的DPPO算法,使用策略差异(例如 Total Variation 或 KL 散度)作为约束,并通过近似方法减少内存占用。

原文摘要

Reinforcement learning (RL) has become a cornerstone for fine-tuning Large Language Models (LLMs), with Proximal Policy Optimization (PPO) serving as the de facto standard algorithm. Despite its ubiquity, we argue that the core ratio clipping mechanism in PPO is structurally ill-suited for the large vocabularies inherent to LLMs. PPO constrains policy updates based on the probability ratio of sampled tokens, which serves as a noisy single-sample Monte Carlo estimate of the true policy divergence. This creates a sub-optimal learning dynamic: updates to low-probability tokens are aggressively over-penalized, while potentially catastrophic shifts in high-probability tokens are under-constrained, leading to training inefficiency and instability. To address this, we propose Divergence Proximal Policy Optimization (DPPO), which substitutes heuristic clipping with a more principled constraint based on a direct estimate of policy divergence (e.g., Total Variation or KL). To avoid huge memory footprint, we introduce the efficient Binary and Top-K approximations to capture the essential divergence with negligible overhead. Extensive empirical evaluations demonstrate that DPPO achieves superior training stability and efficiency compared to existing methods, offering a more robust foundation for RL-based LLM fine-tuning.

标签

Reinforcement Learning Large Language Models Policy Optimization Proximal Policy Optimization

arXiv 分类

cs.LG cs.AI cs.CL