arXiv:2607.25091cs.AIcs.CL2026-07

让小模型强化学习更稳定,解决训练崩溃、精度溢出等关键问题。

Towards Robust Reinforcement Learning for Small-Scale Language Model Agents

论文配图:Towards Robust Reinforcement Learning for Small-Scale Language Model Agents
图 1 · 摘自论文原文
  • 用合并重初始化适配器修复LoRA冻结问题
  • 浮点32精度+三重安全机制避免策略崩溃
  • 小模型也能高效对齐,适合资源有限的场景

在70-500M参数范围内的小语言模型(SLMs)使用强化学习对齐常不稳定,但其失败机制尚未系统研究。本文在15个(模型, 数据集)组合上使用PPO训练Pythia-70M、160M、410M及SmolLM2-135M、360M,覆盖TinyStories、CNN/DailyMail和Wikitext-103数据集。发现三种可复现的失败模式:标准PEFT/TRL流程中LoRA参数静默冻结、bfloat16下重要性比率数值溢出、奖励模型误差导致策略灾难性崩溃。通过合并重初始化适配器、PPO更新时采用float32精度,以及引入包含奖励归一化、重要性比率保护和权重回滚的三层安全机制加以解决。提出容量余量假说:PPO在小模型上的表现取决于流畅的监督模型(PPL<20)和判别性奖励信号,而非参数数量。所提系统在所有实验中稳定收敛,在具备流畅先验和信息丰富奖励信号的配置中,偏好胜率优于SFT基线,且优于指令微调基线,同时所需训练数据显著更少。所有检查点、偏好数据集和训练脚本均已公开。

原文摘要 · Abstract (English)

The alignment of Small Language Models (SLMs) in the 70--500M parameter range using reinforcement learning is often considered unstable, though the underlying failure mechanisms have not been systematically investigated. In the State-of-the-Art (SOTA) research, fifteen (model, corpus) configurations were trained using Proximal Policy Optimization (PPO). The experiments included Pythia-70M, 160M, 410M and SmolLM2-135M, 360M on the TinyStories, CNN/DailyMail, and Wikitext-103 corpora. Three reproducible failure modes were identified in small-scale language models: silent LoRA parameter freezing in standard PEFT/TRL pipelines, numerical overflow in importance ratios when using bfloat16, and catastrophic policy collapse due to reward-model error. These issues were addressed using a merge-and-reinitialize adapter technique, float32 precision during PPO updates, and a three-layer safety mechanism comprising reward whitening, importance-ratio guarding, and weight rollback. In this paper, a capacity-headroom hypothesis is proposed, which states that PPO performance at the SLM scale depends on both a fluent supervised model ($\text{PPL}<20$) and a discriminative reward signal, rather than on the number of model parameters. The proposed system converged stably in all experiments and improved preference win rate over the SFT baseline in configurations with a fluent prior and an informative reward signal. Furthermore, it outperformed instruction-tuned baselines while requiring significantly less training data. All checkpoints, preference datasets, and training scripts are publicly released$^§$.

强化学习小模型稳定性PPO

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