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

ThinkBrake: 通过对数概率边际引导解码实现高效推理

ThinkBrake: Efficient Reasoning via Log-Probability Margin Guided Decoding

Graduate School of Data Science, Seoul National University(数据科学研究生院,首尔国立大学)

Sangjun Song, Minjae Oh, Seungkyu Lee, Sungmin Jo, Yohan Jo

AI总结:

ThinkBrake通过监控句子边界处最高延续词与终止词的对数概率边际,减少过度推理,提升推理效率和准确性。

摘要:

大型推理模型(LRMs)在链式思维(CoT)推理中消耗大量推理时间计算,提升了数学、科学问答和工具使用任务的性能。然而,这引入了过度推理:LRMs常在正确中间解后继续推理并覆盖为错误答案。我们首先证明,通过在每个句子边界注入终止符并选择历史最佳终止点(oracle stopping)可使平均准确率提高8%同时减少72%的思考token,暴露了大量过度推理。受此启发,我们提出了ThinkBrake,它在句子边界监控最高延续词与终止词的对数概率边际,当该边际变窄时停止推理。ThinkBrake无需训练,在数学、科学问答和工具使用基准上实现了良好的准确率-效率权衡,将思考token使用量减少高达30%。此外,我们提供理论分析,证明ThinkBrake等同于测试时重对齐并为终止符提供奖励。

英文摘要:

Large Reasoning Models (LRMs) allocate substantial inference-time compute to Chain-of-Thought (CoT) reasoning, improving performance on mathematics, scientific QA, and tool usage. However, this introduces overthinking: LRMs often reach a correct intermediate solution, continue reasoning, and overwrite it with an incorrect answer. We first demonstrate that oracle stopping--where we inject </think> at every sentence boundary and select the best stopping point in hindsight--improves average accuracy by 8% while reducing thinking tokens by 72%, exposing substantial overthinking. Motivated by this finding, we propose ThinkBrake, which monitors the log-probability margin between the top continuation token and </think> at sentence boundaries, stopping reasoning when this margin narrows. ThinkBrake requires no training and achieves favorable accuracy-efficiency trade-offs across math, scientific QA, and tool usage benchmarks, reducing thinking token usage by up to 30%. Furthermore, we provide theoretical analysis showing that ThinkBrake is equivalent to test-time realignment with a reward bonus for the </think> token.

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

↑