arXiv:2504.12329cs.CLcs.AI2025-04被引 54

小模型推理时借大模型指导,提升准确率并缩短输出

Speculative Thinking: Enhancing Small-Model Reasoning with Large Model Guidance at Inference Time

  • 用大模型在推理阶段指导小模型的思考过程,无需训练
  • 1.5B模型在MATH500上准确率提升6.2%,输出减少15.7%
  • 适用于需要高效推理的小模型,尤其适合资源受限场景

近期进展通过后训练提升模型推理能力,但通常需昂贵训练流程,且输出冗长。我们提出Speculative Thinking,一种无需训练的框架,在推理阶段让大推理模型指导小模型的思考过程,与在词元层面操作的推测解码不同。方法基于两点观察:(1) 如"wait"等支持推理的词常出现在结构分隔符如"\n\n"之后,作为反思或延续信号;(2) 大模型对反思行为有更强控制力,减少无谓回溯并提升推理质量。通过将反思步骤委托给更强大的模型,该方法显著提升小模型推理准确性并缩短输出。借助32B推理模型,1.5B模型在MATH500上的准确率从83.2%提升至89.4%,提高6.2%;平均输出长度由5439词元降至4583词元,减少15.7%。此外,应用于非推理模型Qwen-2.5-7B-Instruct时,其在相同基准上准确率从74.0%升至81.8%,相对提升7.8%。

原文摘要 · Abstract (English)

Recent advances leverage post-training to enhance model reasoning performance, which typically requires costly training pipelines and still suffers from inefficient, overly lengthy outputs. We introduce Speculative Thinking, a training-free framework that enables large reasoning models to guide smaller ones during inference at the reasoning level, distinct from speculative decoding, which operates at the token level. Our approach is based on two observations: (1) reasoning-supportive tokens such as "wait" frequently appear after structural delimiters like "\n\n", serving as signals for reflection or continuation; and (2) larger models exhibit stronger control over reflective behavior, reducing unnecessary backtracking while improving reasoning quality. By strategically delegating reflective steps to a more capable model, our method significantly boosts the reasoning accuracy of reasoning models while shortening their output. With the assistance of the 32B reasoning model, the 1.5B model's accuracy on MATH500 increases from 83.2% to 89.4%, marking a substantial improvement of 6.2%. Simultaneously, the average output length is reduced from 5439 tokens to 4583 tokens, representing a 15.7% decrease. Moreover, when applied to a non-reasoning model (Qwen-2.5-7B-Instruct), our framework boosts its accuracy from 74.0% to 81.8% on the same benchmark, achieving a relative improvement of 7.8%.

推理增强小模型优化大模型指导推理效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。