通过动态奖励机制让模型快慢结合,既省token又提解题准度。
Learn to Reason Efficiently with Adaptive Length-based Reward Shaping
- 用可调长度的阶梯奖励引导模型生成更精炼推理链
- 在AIME2024上提升6.1分,同时减少63%的令牌使用
- 适配不同难度问题,自动调节奖励强度,适合高效推理研究者
大型推理模型(LRMs)在强化学习(RL)驱动下能生成长推理链解决复杂问题,但冗余内容严重降低效率。本文提出统一框架,将各类高效推理方法归因于基于长度的奖励塑造。在此基础上,提出新方法LASER,采用阶梯函数作为奖励,由目标长度控制,实现了性能与效率的更好权衡。进一步引入两个关键思想:一是奖励应随训练动态调整以适应模型行为演化;二是奖励需具备难度感知能力,对简单问题更严厉惩罚长推理链。由此提出的LASER-D方法实现快慢思维融合,在DeepSeek-R1-Distill-Qwen-1.5B、7B和32B模型上验证有效,显著提升推理表现并压缩输出长度。例如,在AIME2024上取得+6.1分提升,同时减少63%的令牌消耗。分析显示,该方法生成的推理链更简洁,冗余自省内容大幅减少。
原文摘要 · Abstract (English)
Large Reasoning Models (LRMs) have shown remarkable capabilities in solving complex problems through reinforcement learning (RL), particularly by generating long reasoning traces. However, these extended outputs often exhibit substantial redundancy, which limits the efficiency of LRMs. In this paper, we investigate RL-based approaches to promote reasoning efficiency. Specifically, we first present a unified framework that formulates various efficient reasoning methods through the lens of length-based reward shaping. Building on this perspective, we propose a novel Length-bAsed StEp Reward shaping method (LASER), which employs a step function as the reward, controlled by a target length. LASER surpasses previous methods, achieving a superior Pareto-optimal balance between performance and efficiency. Next, we further extend LASER based on two key intuitions: (1) The reasoning behavior of the model evolves during training, necessitating reward specifications that are also adaptive and dynamic; (2) Rather than uniformly encouraging shorter or longer chains of thought (CoT), we posit that length-based reward shaping should be difficulty-aware i.e., it should penalize lengthy CoTs more for easy queries. This approach is expected to facilitate a combination of fast and slow thinking, leading to a better overall tradeoff. The resulting method is termed LASER-D (Dynamic and Difficulty-aware). Experiments on DeepSeek-R1-Distill-Qwen-1.5B, DeepSeek-R1-Distill-Qwen-7B, and DeepSeek-R1-Distill-Qwen-32B show that our approach significantly enhances both reasoning performance and response length efficiency. For instance, LASER-D and its variant achieve a +6.1 improvement on AIME2024 while reducing token usage by 63%. Further analysis reveals our RL-based compression produces more concise reasoning patterns with less redundant "self-reflections". Resources are at https://github.com/hkust-nlp/Laser.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。