arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2511.16331cs.CL
收藏

将自我修正纳入大语言模型推理强化

Incorporating Self-Rewriting into Large Language Model Reasoning Reinforcement

ByteDance China(字节跳动中国)

Jiashu Yao, Heyan Huang, Shuang Zeng, Chuwei Luo, WangJie You, Jie Tang, Qingsong Liu, Yuhang Guo, Yangyang Kang

AI总结:

本研究提出自我修正框架,通过重写推理文本提升大语言模型的内部推理质量,实验表明其在准确性与推理长度权衡上优于现有方法。

摘要:

通过基于结果正确性的强化学习(RL),具有扩展推理计算能力的大推理模型(LRMs)在复杂推理任务上已展现出显著的成功。然而,单一奖励机制仅关注最终正确性,限制了其对内部推理过程的详细监督能力。这种缺陷导致内部推理质量低下,表现为过度思考、不足思考、冗余思考和混乱思考等问题。受最近在LRM自我奖励方面的进展启发,我们引入了自我修正框架,其中模型会重写自己的推理文本,并随后从重写后的推理中学习以提高内部思维过程的质量。在算法设计方面,我们提出了一种选择性重写方法,其中只有“简单”的样本(由模型的一致正确性定义)被重写,从而保留了GRPO的所有原始奖励信号。在实际实现中,我们将在一个单一的批次中编译重写和普通生成,保持RL算法的可扩展性,并仅引入约10%的开销。在不同任务和不同模型大小上的广泛实验验证了自我重写的有效性。在准确性与长度的权衡方面,自我重写方法在不使用显式指令减少推理长度的情况下,实现了更高的准确性(+0.6)和显著更短的推理(-46%),优于现有强大的基线。在内部推理质量方面,自我重写在LLM-as-a-judge指标下实现了显著更高的分数(+7.2),成功缓解了内部推理缺陷。

英文摘要:

Through reinforcement learning (RL) with outcome correctness rewards, large reasoning models (LRMs) with scaled inference computation have demonstrated substantial success on complex reasoning tasks. However, the one-sided reward, focused solely on final correctness, limits its ability to provide detailed supervision over internal reasoning process. This deficiency leads to suboptimal internal reasoning quality, manifesting as issues like over-thinking, under-thinking, redundant-thinking, and disordered-thinking. Inspired by the recent progress in LRM self-rewarding, we introduce self-rewriting framework, where a model rewrites its own reasoning texts, and subsequently learns from the rewritten reasoning to improve the internal thought process quality. For algorithm design, we propose a selective rewriting approach wherein only "simple" samples, defined by the model's consistent correctness, are rewritten, thereby preserving all original reward signals of GRPO. For practical implementation, we compile rewriting and vanilla generation within one single batch, maintaining the scalability of the RL algorithm and introducing only ~10% overhead. Extensive experiments on diverse tasks with different model sizes validate the effectiveness of self-rewriting. In terms of the accuracy-length tradeoff, the self-rewriting approach achieves improved accuracy (+0.6) with substantially shorter reasoning (-46%) even without explicit instructions in rewriting prompts to reduce reasoning length, outperforming existing strong baselines. In terms of internal reasoning quality, self-rewriting achieves significantly higher scores (+7.2) under the LLM-as-a-judge metric, successfully mitigating internal reasoning flaws.

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

↑