arXiv:2603.00963cs.LGcs.CL2026-03

通过让模型输出更稳定,提升强化学习训练效果。

Stabilizing Policy Optimization via Logits Convexity

  • 在模型输出层引入凸性优化,改善梯度方向性。
  • 实验证明新方法比传统强化学习更稳定,性能更好。
  • 适合需要稳定训练的大模型微调场景。

尽管强化学习(RL)是大型语言模型(LLMs)近期成功的关键,但其优化过程以不稳定著称,尤其相较于监督微调(SFT)。本文从梯度视角分析了SFT与RL之间的稳定性差距,发现SFT损失在模型logits上的凸性是实现稳定训练的关键。理论分析表明,该性质能引导更优的梯度方向。相比之下,广泛使用的近端策略优化(PPO)采用裁剪代理目标,缺乏这种稳定机制。受此启发,我们提出对数凸优化(LCO),一种简单而有效的策略优化框架,使学习策略逼近原始RL目标下的最优解,从而模拟logits层面凸性的稳定效应。在多个模型家族上的大量实验表明,所提LCO框架显著提升了训练稳定性,并在广泛基准测试中超越传统强化学习方法。

原文摘要 · Abstract (English)

While reinforcement learning (RL) has been central to the recent success of large language models (LLMs), RL optimization is notoriously unstable, especially when compared to supervised fine-tuning (SFT). In this work, we investigate the stability gap between SFT and RL from a gradient-based perspective, and show that the convexity of the SFT loss with respect to model logits plays a key role in enabling stable training. Our theoretical analysis demonstrates that this property induces favorable gradient directionality during optimization. In contrast, Proximal Policy Optimization (PPO), a widely adopted policy gradient algorithm utilizing a clipped surrogate objective, lacks this stabilizing property. Motivated by this observation, we propose Logits Convex Optimization (LCO), a simple yet effective policy optimization framework that aligns the learned policy with an optimal target derived from the original RL objective, thereby emulating the stabilizing effects of logits-level convexity. Extensive experiments across multiple model families show that our LCO framework consistently improves training stability and outperforms conventional RL methods on a broad range of benchmarks.

强化学习大模型微调训练稳定

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。