动态调整奖励权重,让模型更高效学习复杂推理任务。
ADORA: Training Reasoning Models with Dynamic Advantage Estimation on Reinforcement Learning
- 根据在线推演中样本的实时价值动态划分优劣数据
- 在几何与数学推理任务中显著提升长序列推理性能
- 无需调参即可适配现有算法,适合强化学习研究者
强化学习已成为复杂任务中构建推理模型的核心技术,涵盖数学问题求解到想象推理。此类模型的优化通常依赖于策略梯度方法,其效果取决于优势函数的准确估计。然而,现有方法普遍采用静态优势估计,忽视了训练样本随时间演变的动态价值,导致信用分配效率低下。这引发策略更新不理想,表现为收敛速度慢和学习不稳定。为此,我们提出ADORA(基于在线推演的优势动态调整),通过在线模型推演中自适应地将训练数据划分为暂时有利与不利样本,动态调整优势函数权重。该差异化策略可无缝集成至现有策略优化算法中,无需大幅架构改动,使模型优先学习更具信息量的经验,实现更高效的策略更新。在多种模型家族和不同数据规模下的广泛评估表明,ADORA具备鲁棒性与高效性,在几何与数学推理任务中持续实现显著性能提升,且无需敏感超参数调优。
原文摘要 · Abstract (English)
Reinforcement learning has become a cornerstone technique for developing reasoning models in complex tasks, ranging from mathematical problem-solving to imaginary reasoning. The optimization of these models typically relies on policy gradient methods, whose efficacy hinges on the accurate estimation of an advantage function. However, prevailing methods typically employ static advantage estimation, a practice that leads to inefficient credit assignment by neglecting the dynamic utility of training samples over time. This limitation results in suboptimal policy updates, which in turn manifest as slower convergence rates and increased learning instability, as models fail to adapt to evolving sample utilities effectively. To address this problem, we introduce \textbf{ADORA} (\textbf{A}dvantage \textbf{D}ynamics via \textbf{O}nline \textbf{R}ollout \textbf{A}daptation), a novel framework for policy optimization. ADORA dynamically adjusts the advantage function's weighting by adaptively categorizing training data into temporarily advantageous and disadvantageous samples, based on their evolving utility during online model rollouts. This tailored data differentiation strategy allows ADORA to be seamlessly integrated into existing policy optimization algorithms without significant architectural modifications, enabling the policy to prioritize learning from more informative experiences and thereby achieve more efficient policy updates. Extensive evaluations across diverse model families and varying data scales demonstrate that ADORA is a robust and efficient framework. It significantly enhances long reasoning in both geometric and mathematical tasks, consistently achieving notable performance gains without requiring sensitive hyperparameter tuning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。