通过先决策再思考,减少无效推理,提升大模型推理效率
Fork-Think with Confidence

- 先用置信度识别关键分叉点,再生成多路径推理
- 最多降低57%运行时间,30%令牌消耗,性能相当或更优
- 无需训练即可集成现有优化技术,适合高效推理场景
并行思维在不需重训练的情况下显著提升了大模型在推理任务上的表现。然而,现有方法遵循‘先思考再决定’范式,即先生成多个推理路径,再剪枝或终止冗余路径,导致过度生成。相比之下,‘先决定再思考’的范式尚未得到充分探索。本文提出Fork-think with confidence,首先通过单条初始路径的模型置信度识别分叉点,再触发多路径生成与聚合以获得最终回答。在三个模型和三个推理基准上的实验表明,该方法最多可减少57%运行时间、30%令牌消耗,且性能相当或更优。分析显示,该方法能有效识别与下游任务相关的有意义分叉点,且在较晚位置采样可显著提升生成质量。此外,结合早期停止和加权投票等机制后,性能可媲美现有最先进方法,且无需预热或离线训练。结果表明,预设分叉是一种有前景的高效大模型推理方向。
原文摘要 · Abstract (English)
Parallel thinking has enjoyed great success for boosting LLM performance on reasoning tasks without the need for any re-training. However, existing methods follow a think-first-then-decide paradigm, i.e., they first sample multiple reasoning paths, which inevitably leads to overgeneration, then prune or stop unnecessary paths to compensate. In contrast, decide-first-then-think, i.e., first identifying points that are likely to lead to desirable generations, has been underexplored so far. Following this paradigm, we propose Fork-think with confidence, that first identifies forking points using model confidence in a single seeding path, then triggers thinking, sampling multiple continuations and aggregating them for the final response. Our experiments across three models and three reasoning benchmarks show that Fork-think reduces the token consumption by up to 30% and run-time by up to 57%, while performing comparable to or better than parallel thinking. Our analysis reveals that Fork-think is able to identify forking points that are meaningful with respect to the downstream task and that sampling at later positions can lead to substantially better generations. Finally, we demonstrate how combining Fork-think with existing mechanisms such as early stopping and weighted voting can further boost the performance and perform comparably to existing state-of-the-art methods, without requiring any warm-up or offline training. Our results establish pre-determined forking as a promising research direction for efficient LLM reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。