解释推理模型为何循环,提出温度仅是临时缓解。
Wait, Wait, Wait... Why Do Reasoning Models Loop?
- 发现低温度下模型易循环,源于学习偏差而非单纯采样问题。
- 大模型循环少,蒸馏模型循环严重,显示训练分布与模型不匹配。
- 高温可减少循环但延长生成,根本解法需改进训练过程。
推理模型(如 DeepSeek-R1)通过长思维链解决复杂问题,但常在低温度或贪婪解码时出现重复循环。我们研究发现,低温度下循环普遍,大模型循环较少,而蒸馏后的学生模型即使教师极少循环,自身仍显著循环。这表明训练分布与模型学习之间存在偏差,称为学习错误,是循环主因。通过合成图推理任务,揭示两种机制:一是学习困难导致规避风险,模型更倾向选择易学的循环动作;二是变压器固有时间相关性偏见,使少数动作反复被选,形成循环。高温虽能通过促进探索减少循环,但无法根治学习错误,导致生成过长。因此,温度仅为权宜之计。论文建议从训练阶段干预,直接减少学习错误。
原文摘要 · Abstract (English)
Reasoning models (e.g., DeepSeek-R1) generate long chains of thought to solve harder problems, but they often loop, repeating the same text at low temperatures or with greedy decoding. We study why this happens and what role temperature plays. With open reasoning models, we find that looping is common at low temperature. Larger models tend to loop less, and distilled students loop significantly even when their teachers rarely do. This points to mismatches between the training distribution and the learned model, which we refer to as errors in learning, as a key cause. To understand how such errors cause loops, we introduce a synthetic graph reasoning task and demonstrate two mechanisms. First, risk aversion caused by hardness of learning: when the correct progress-making action is hard to learn but an easy cyclic action is available, the model puts relatively more probability on the cyclic action and gets stuck. Second, even when there is no hardness, Transformers show an inductive bias toward temporally correlated errors, so the same few actions keep being chosen and loops appear. Higher temperature reduces looping by promoting exploration, but it does not fix the errors in learning, so generations remain much longer than necessary at high temperature; in this sense, temperature is a stopgap rather than a holistic solution. We end with a discussion of training-time interventions aimed at directly reducing errors in learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。