arXiv:2506.18631cs.LGcs.AI2025-06NeurIPS被引 14

给离散奖励加随机噪声,让大模型训练更快更稳。

ReDit: Reward Dithering for Improved LLM Policy Optimization

  • 用随机噪声扰动离散奖励,持续提供探索梯度。
  • 训练步数减少90%,性能仍比原方法高4%。
  • 适合需要高效优化的大模型策略训练场景。

DeepSeek-R1通过基于规则的奖励系统成功提升了大语言模型的推理能力,但此类奖励函数通常是离散的。实验发现,离散奖励会导致梯度异常、优化不稳和收敛缓慢。为此,我们提出ReDit(Reward Dithering),通过向离散奖励信号添加简单随机噪声,使学习过程中持续获得探索性梯度,实现更平滑的梯度更新并加速收敛。注入的噪声在平坦奖励区域引入随机性,促使模型探索新策略,跳出局部最优。跨多种任务的实验表明,ReDit平均仅需约10%的训练步数即可达到与原始GRPO相当的性能,且在相同训练时长下仍提升4%。可视化验证了梯度问题显著缓解。理论分析进一步支持其优势。

原文摘要 · Abstract (English)

DeepSeek-R1 has successfully enhanced Large Language Model (LLM) reasoning capabilities through its rule-based reward system. While it's a ''perfect'' reward system that effectively mitigates reward hacking, such reward functions are often discrete. Our experimental observations suggest that discrete rewards can lead to gradient anomaly, unstable optimization, and slow convergence. To address this issue, we propose ReDit (Reward Dithering), a method that dithers the discrete reward signal by adding simple random noise. With this perturbed reward, exploratory gradients are continuously provided throughout the learning process, enabling smoother gradient updates and accelerating convergence. The injected noise also introduces stochasticity into flat reward regions, encouraging the model to explore novel policies and escape local optima. Experiments across diverse tasks demonstrate the effectiveness and efficiency of ReDit. On average, ReDit achieves performance comparable to vanilla GRPO with only approximately 10% the training steps, and furthermore, still exhibits a 4% performance improvement over vanilla GRPO when trained for a similar duration. Visualizations confirm significant mitigation of gradient issues with ReDit. Moreover, theoretical analyses are provided to further validate these advantages.

大模型训练强化学习奖励设计

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