arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.04697cs.CLcs.AIcs.LG
收藏

L1:通过强化学习控制推理模型的思考时长

L1: Controlling How Long A Reasoning Model Thinks With Reinforcement Learning

Carnegie Mellon University(卡内基梅隆大学)

Pranjal Aggarwal, Sean Welleck

AI总结:

本研究提出强化学习方法LCPO,训练出可精准控制思维链长度的推理模型L1,实现计算成本与准确率的平滑权衡,还发现其短推理能力,1.5B规模在等推理长度下超越GPT-4o。

摘要:

推理语言模型展现出一种惊人的能力:通过“更长时间思考”——即生成更长的思维链(chain-of-thought, CoT)序列、使用更多计算资源,就能在测试阶段提升性能。然而,这类模型的思维链推理长度无法控制,导致无法通过分配测试阶段计算资源来达到预期性能水平。\n我们提出了长度控制策略优化(Length Controlled Policy Optimization, LCPO),这是一种简单的强化学习方法,可同时优化准确率以及对用户指定长度约束的遵循度。我们使用LCPO训练了推理语言模型L1,该模型生成的输出能够满足提示中给出的长度约束。L1的长度控制能力可在广泛任务上实现计算成本与准确率的平滑权衡,并且在长度控制方面优于当前最优的S1方法。\n此外,我们还发现用LCPO训练的模型具备一种出人意料的短思维链能力。具体而言,我们通过LCPO得到了短推理模型(Short Reasoning Models, SRMs),它们展现出与全长度推理模型相似的推理模式,但生成的CoT长度可与非推理模型相当。这些模型取得了显著的性能提升,例如,我们的15亿参数L1模型在相同推理长度下表现超越了GPT-4o。总体而言,LCPO实现了对推理长度的精准控制,支持对测试阶段计算资源和准确率进行细粒度分配。我们在https://www.cmu-l3.github.io/l1 发布了代码和模型。

英文摘要:

Reasoning language models have shown an uncanny ability to improve performance at test-time by ``thinking longer''-that is, by generating longer chain-of-thought sequences and hence using more compute. However, the length of their chain-of-thought reasoning is not controllable, making it impossible to allocate test-time compute to achieve a desired level of performance. We introduce Length Controlled Policy Optimization (LCPO), a simple reinforcement learning method that optimizes for accuracy and adherence to user-specified length constraints. We use LCPO to train L1, a reasoning language model that produces outputs satisfying a length constraint given in its prompt. L1's length control allows for smoothly trading off computational cost and accuracy on a wide range of tasks, and outperforms the state-of-the-art S1 method for length control. Furthermore, we uncover an unexpected short chain-of-thought capability in models trained with LCPO. Specifically, using LCPO we derive Short Reasoning Models (SRMs), that exhibit similar reasoning patterns as full-length reasoning models, but can generate CoT lengths comparable to non-reasoning models. They demonstrate significant performance gains, for instance, our 1.5B L1 model surpasses GPT-4o at equal reasoning lengths. Overall, LCPO enables precise control over reasoning length, allowing for fine-grained allocation of test-time compute and accuracy. We release code and models at https://www.cmu-l3.github.io/l1

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑