当推理收敛时停止:保留语义的推理模型提前退出
摘要
本文介绍 PUMA,一个即插即用框架,通过检测思维链推理中的语义冗余实现提前退出,在多个模型和基准测试中平均减少 26.2% 的 Token,同时保持准确性和推理质量。
查看缓存全文
缓存时间: 2026/05/19 06:31
论文页面 - 当推理收敛时停止:面向推理模型的语义保持型早退机制
来源:https://huggingface.co/papers/2605.17672
摘要
研究人员提出了 PUMA,这是一种利用语义冗余检测来提升大型模型推理效率的框架,能够识别出继续思考不会带来新见解的时刻,从而在保持答案准确性和推理质量的同时减少计算浪费。
Large Reasoning Models (https://huggingface.co/papers?q=Large%20Reasoning%20Models)(LRMs)通过生成长链思维 (https://huggingface.co/papers?q=chains%20of%20thought)(CoT)取得了强劲性能,但常常过度思考——在解决方案已经稳定后仍在继续推理,从而浪费 token 并增加延迟。现有的推理时早期退出方法 (https://huggingface.co/papers?q=early-exit%20methods) 主要依赖答案级信号 (https://huggingface.co/papers?q=answer-level%20signals)(如置信度或尝试答案的一致性)来决定何时停止。然而,这些信号主要反映答案的准备状态,而非推理收敛 (https://huggingface.co/papers?q=reasoning%20convergence):它们可能在模型完成探索或自我纠正之前触发,导致过早退出,从而降低最终答案的准确性,并使保留的推理链在语义上不完整。我们识别出推理层面的语义冗余 (https://huggingface.co/papers?q=semantic%20redundancy) 作为语义保持型早期退出的补充信号:当连续步骤不再增加新颖进展,而是回到已建立的结论时,推理轨迹很可能已经收敛。基于这一洞察,我们提出了 PUMA,一个即插即用框架,它将轻量级冗余检测器 (https://huggingface.co/papers?q=Redundancy%20Detector) 与答案级验证 (https://huggingface.co/papers?q=answer-level%20verification) 相结合。检测器标记语义冗余的候选退出点,而验证则确认停止是否安全,使 PUMA 能够移除冗余的后续推理,同时保留答案准确性以及连贯的推理前缀。在五个 LRM 和五个具有挑战性的推理基准上,PUMA 实现了平均 26.2% 的 token 减少 (https://huggingface.co/papers?q=token%20reduction),同时保持了准确性和保留的 CoT 质量。关于代码生成、零样本视觉语言推理以及学习停止策略内化方面的额外实验进一步表明,推理层面的冗余是一种鲁棒、可迁移且可学习的信号,可用于高效推理。我们的代码可在 https://github.com/giovanni-vaccarino/PUMA 获取。
查看 arXiv 页面 (https://arxiv.org/abs/2605.17672)
查看 PDF (https://arxiv.org/pdf/2605.17672)
GitHub2 (https://github.com/giovanni-vaccarino/PUMA)
添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2605.17672)
在你的代理中获取此论文:
hf papers read 2605.17672
没有最新的 CLI? curl -LsSf https://hf.co/cli/install.sh | bash
引用此论文的模型
尚无模型链接到此论文
请在模型 README.md 中引用 arxiv.org/abs/2605.17672 以从该页面链接它。
引用此论文的数据集
尚无数据集链接到此论文
请在数据集 README.md 中引用 arxiv.org/abs/2605.17672 以从该页面链接它。
引用此论文的 Spaces
尚无 Space 链接到此论文
请在 Space README.md 中引用 arxiv.org/abs/2605.17672 以从该页面链接它。
包含此论文的收藏
尚无收藏包含此论文
请将此论文添加到收藏 (https://huggingface.co/new-collection) 以从该页面链接它。
相似文章
当进一步推理无益时停止:推理模型中的注意力状态自适应生成
本文提出ASAG,一种无需训练的方法,基于注意力分布自适应地停止大型推理模型的推理,在使用DeepSeek-R1-Distill和Qwen3模型的基准测试中,将token使用量减少约40%,同时准确率提升3.2%。
早期剪枝学习!高效并行推理的路径剪枝方法
本文提出了 STOP(SuperTOken for Pruning),一个系统框架,用于在大型推理模型的并行推理中早期剪枝低效推理路径。该方法在 1.5B 到 20B 参数的模型中实现了优异的效率和效果,在固定计算预算下将 GPT-OSS-20B 在 AIME25 上的准确率从 84% 提升到 90%。
@HuggingPapers:并行推理及时止损——STOP 通过读取 KV-cache 状态提前剪枝注定失败的轨迹,…
STOP 方法利用 KV-cache 状态提前剪枝注定失败的推理轨迹,token 用量降 70%,在 1.5B–20B 模型上提升 AIME 与 GPQA 准确率。
学习何时停止有用?推理模型早期退出的成本感知研究
本文介绍了LearnStop,一种用于推理模型的轻量级检查点停止器,它从在线特征中预测前缀正确性,并发现学习式停止仅在多个问题早期正确且没有单一可靠的标量信号时,才比标量规则更有价值。
及时止损!学习早期剪枝路径以实现高效并行推理
本文介绍了STOP(用于剪枝的超令牌),一种轻量级方法,通过在并行解码中附加可学习令牌并读取KV缓存状态,学会早期剪枝不优的推理路径,在AIME和GPQA基准测试中实现70%的令牌减少,同时提高性能。