区分思考重要性,让大模型分工协作提升推理效率
Not All Thoughts are Generated Equal: Efficient LLM Reasoning via Multi-Turn Reinforcement Learning
- 按思考重要性分层生成,长思考模型专注关键步骤
- 在多个数学推理基准上减少80%以上token消耗
- 适合需要高效推理的部署场景或资源受限设备
压缩大语言模型(LLM)的长链式思考(CoT)是提升推理效率的新策略。现有方法对所有思考等同处理,限制了更简洁高效的推理表现。本文通过自动分块与蒙特卡洛推演,分析不同思考在推理中的有效性和效率,提出一个理论有界指标联合衡量两者。基于此,我们设计Long⊗Short框架:由长思考模型生成关键步骤,短思考模型高效生成其余部分。先用少量冷启动数据微调两类模型,再采用面向协同的多轮强化学习,促进模型自进化与协作。实验表明,在MATH500、AIME24/25、AMC23和GPQA Diamond基准上,Qwen2.5-7B与Llama3.1-8B达到与DeepSeek-R1-Distill-Qwen-7B和DeepSeek-R1-Distill-Llama-8B相当性能,同时token长度降低超80%。代码与数据已公开于https://github.com/usail-hkust/LongShort。
原文摘要 · Abstract (English)
Compressing long chain-of-thought (CoT) from large language models (LLMs) is an emerging strategy to improve the reasoning efficiency of LLMs. Despite its promising benefits, existing studies equally compress all thoughts within a long CoT, hindering more concise and effective reasoning. To this end, we first investigate the importance of different thoughts by examining their effectiveness and efficiency in contributing to reasoning through automatic long CoT chunking and Monte Carlo rollouts. Building upon the insights, we propose a theoretically bounded metric to jointly measure the effectiveness and efficiency of different thoughts. We then propose Long$\otimes$Short, an efficient reasoning framework that enables two LLMs to collaboratively solve the problem: a long-thought LLM for more effectively generating important thoughts, while a short-thought LLM for efficiently generating remaining thoughts. Specifically, we begin by synthesizing a small amount of cold-start data to fine-tune LLMs for long-thought and short-thought reasoning styles, respectively. Furthermore, we propose a synergizing-oriented multi-turn reinforcement learning, focusing on the model self-evolution and collaboration between long-thought and short-thought LLMs. Experimental results show that our method enables Qwen2.5-7B and Llama3.1-8B to achieve comparable performance compared to DeepSeek-R1-Distill-Qwen-7B and DeepSeek-R1-Distill-Llama-8B, while reducing token length by over 80% across the MATH500, AIME24/25, AMC23, and GPQA Diamond benchmarks. Our data and code are available at https://github.com/usail-hkust/LongShort.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。