arXiv:2505.17746cs.CL2025-05EMNLP被引 8

让大模型高效思考:不生成显式思维令牌也能提升推理准确率

Fast Quiet-STaR: Thinking Without Thought Tokens

  • 通过渐进式训练让模型内化简洁的抽象推理过程
  • 在相同推理时间内,准确率比原方法提升9%至5.7%
  • 适合追求高效推理的AI研究者与工程应用

大型语言模型在自然语言处理任务中表现优异,但复杂推理任务的进一步提升需超越单纯扩大模型规模或训练数据。近期工作Quiet STaR通过生成逐标记的思维轨迹显著改善推理,但带来显著推理开销。本文提出Fast Quiet STaR,一种更高效的推理框架,在保留逐标记推理优势的同时降低计算成本。该方法采用基于课程学习的训练策略,逐步减少思维标记数量,使模型内化更抽象、更简洁的推理过程。进一步通过强化学习微调,将其扩展至标准的下一标记预测(NTP)设置,形成Fast Quiet-STaR NTP,推理时无需显式生成思维标记。在四个基准数据集上,使用Mistral 7B和Qwen2.5 7B进行实验表明,Fast Quiet-STaR在相同推理时间预算下平均准确率持续优于Quiet-STaR。特别地,Fast Quiet-STaR NTP在Mistral 7B上实现9%的平均准确率提升,在Qwen2.5 7B上提升5.7%,且保持相同推理延迟。代码将公开于https://github.com/huangwei200012/Fast-Quiet-STaR。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have achieved impressive performance across a range of natural language processing tasks. However, recent advances demonstrate that further gains particularly in complex reasoning tasks require more than merely scaling up model sizes or training data. One promising direction is to enable models to think during the reasoning process. Recently, Quiet STaR significantly improves reasoning by generating token-level thought traces, but incurs substantial inference overhead. In this work, we propose Fast Quiet STaR, a more efficient reasoning framework that preserves the benefits of token-level reasoning while reducing computational cost. Our method introduces a curriculum learning based training strategy that gradually reduces the number of thought tokens, enabling the model to internalize more abstract and concise reasoning processes. We further extend this approach to the standard Next Token Prediction (NTP) setting through reinforcement learning-based fine-tuning, resulting in Fast Quiet-STaR NTP, which eliminates the need for explicit thought token generation during inference. Experiments on four benchmark datasets with Mistral 7B and Qwen2.5 7B demonstrate that Fast Quiet-STaR consistently outperforms Quiet-STaR in terms of average accuracy under the same inference time budget. Notably, Fast Quiet-STaR NTP achieves an average accuracy improvement of 9\% on Mistral 7B and 5.7\% on Qwen2.5 7B, while maintaining the same inference latency. Our code will be available at https://github.com/huangwei200012/Fast-Quiet-STaR.

推理优化大模型思维链高效推理

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