不训练大模型,用小模型引导就能激发长链条推理能力
Logit Arithmetic Elicits Long Reasoning Capabilities Without Training
- 用小模型作为引导,通过解码时的逻辑运算激活大模型推理潜力
- 在四个数学数据集上使准确率提升26%~29%,仅用21倍小的引导模型
- 无需额外训练即可迁移强化学习获得的推理技能,适合资源有限场景
大型推理模型(LRMs)可通过包含回溯与自修正等认知策略的长链思维(CoT)完成复杂推理。近期研究认为部分模型天生具备此类能力,但需额外训练才能释放。本文首次探究是否可在不训练的情况下激发该能力。提出一种解码时方法ThinkLogit,利用日志算术(Liu et al., 2024)技术,以一个显著更小的模型作为引导,调整目标大语言模型的推理行为。进一步通过偏好优化训练引导模型,使用来自目标和引导模型的正确/错误推理对进行优化,称为ThinkLogit-DPO。实验表明,ThinkLogit和ThinkLogit-DPO在使用Qwen2.5-32B作为目标模型、R1-Distill-Qwen-1.5B(小21倍)作为引导模型时,在四个数学数据集上相对基线提升pass@1达26%和29%。最后证明,ThinkLogit可迁移强化学习中习得的推理技能,使pass@1相对原始模型提升13%。本工作提供了一种计算高效、几乎零训练成本激发大模型长推理能力的方法。
原文摘要 · Abstract (English)
Large reasoning models (LRMs) can do complex reasoning via long chain-of-thought (CoT) involving cognitive strategies such as backtracking and self-correction. Recent studies suggest that some models inherently possess these long reasoning abilities, which may be unlocked via extra training. Our work first investigates whether we can elicit such behavior without any training. To this end, we propose a decoding-time approach, ThinkLogit, which utilizes logits arithmetic (Liu et al., 2024) to tune a target large LM for long reasoning using a substantially smaller model as guider. We then show that we can further boost performance by training the guider model with preference optimization over correct/incorrect reasoning pairs sampled from both the target and guider model -- a setup we refer to as ThinkLogit-DPO. Our experiments demonstrate that ThinkLogit and ThinkLogit-DPO achieve a relative improvement in pass@1 by 26% and 29%, respectively, over four mathematical datasets using the Qwen2.5-32B when guided by R1-Distill-Qwen-1.5B -- a model 21x smaller. Lastly, we show that ThinkLogit can transfer long reasoning skills acquired through reinforcement learning, improving pass@1 by 13% relative compared to the Qwen2.5-32B base model. Our work presents a computationally-efficient method to elicit long reasoning in large models with minimal or no additional training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。