让小模型像大模型一样高效推理,只需更少资源。
Enhancing Reasoning Abilities of Small LLMs with Cognitive Alignment
- 设计多智能体系统,分步优化小模型的思考过程
- 在多个推理基准上超越现有方法,性能显著提升
- 适合资源有限但需强推理能力的应用场景
大型推理模型(LRMs)如OpenAI的o1和DeepSeek-R1通过深度思考实现了显著进步,但其高昂的资源消耗凸显了训练高效小推理模型的需求。小模型与大模型在认知能力和思维路径上存在差异,直接从大模型蒸馏思维链(CoT)常效果不佳,且需大量标注数据。本文提出一种新型批判-重思-验证(CRV)系统,由多个专注于不同任务的LLM智能体组成:(i)根据小模型的认知能力评估CoT;(ii)基于反馈重构并优化思考过程;(iii)验证修正后结果的正确性。在此基础上,进一步提出认知偏好优化(CogPO)算法,持续对齐小模型的推理过程与其认知能力。在多个挑战性推理基准上的全面评估表明,CRV+CogPO框架显著优于其他方法。
原文摘要 · Abstract (English)
The reasoning capabilities of large reasoning models (LRMs), such as OpenAI's o1 and DeepSeek-R1, have seen substantial advancements through deep thinking. However, these enhancements come with significant resource demands, underscoring the need for training effective small reasoning models. A critical challenge is that small models possess different reasoning capacities and cognitive trajectories compared with their larger counterparts. Hence, directly distilling chain-of-thought (CoT) rationales from large LRMs to smaller ones can sometimes be ineffective and often requires a substantial amount of annotated data. In this paper, we first introduce a novel Critique-Rethink-Verify (CRV) system, designed for training smaller yet powerful LRMs. Our CRV system consists of multiple LLM agents, each specializing in unique tasks: (i) critiquing the CoT rationales according to the cognitive capabilities of smaller models, (ii) rethinking and refining these CoTs based on the critiques, and (iii) verifying the correctness of the refined results. Building on the CRV system, we further propose the Cognitive Preference Optimization (CogPO) algorithm to continuously enhance the reasoning abilities of smaller models by aligning their reasoning processes with their cognitive capacities. Comprehensive evaluations on challenging reasoning benchmarks demonstrate the efficacy of our CRV+CogPO framework, which outperforms other methods by a large margin.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。