用短推理作对比,让模型更爱长思考,提升推理能力
Thinking Preference Optimization
- 用短推理当负样本,长推理当正样本,做偏好优化
- 使数学推理准确率提升8.6%,输出长度增加25.9%
- 无需新数据,可持续提升已有的小模型推理能力
监督微调(SFT)是提升小型大语言模型长链思维(CoT)推理能力的有效方法,通过使用更大模型生成的长CoT响应进行微调。为持续改进推理能力,通常需收集新高质量长CoT SFT数据或重复训练现有数据集。然而,获取新数据成本高且受限,重复训练常导致性能停滞或下降。为此,我们提出思维偏好优化(ThinkPO),一种简单有效的后SFT方法,无需新长CoT响应即可提升长CoT推理能力。ThinkPO利用易得的短CoT响应作为拒绝答案,长CoT响应作为优选答案,对同一问题进行直接偏好优化,促使模型偏好更长的推理输出。实验表明,ThinkPO能显著提升SFT模型的推理性能,例如使SFT模型的数学推理准确率提高8.6%,输出长度增加25.9%。值得注意的是,ThinkPO可持续提升公开蒸馏的SFT模型性能,如将DeepSeek-R1-Distill-Qwen-7B在MATH500上的表现从87.4%提升至91.2%。
原文摘要 · Abstract (English)
Supervised Fine-Tuning (SFT) has been a go-to and effective method for enhancing long chain-of-thought (CoT) reasoning in relatively small LLMs by fine-tuning them with long CoT responses from larger LLMs. To continually improve reasoning abilities, we can either collect new high-quality long CoT reasoning SFT data or repeatedly train on existing SFT datasets. However, acquiring new long CoT SFT data is costly and limited, while repeated training often results in a performance plateau or decline. To further boost the performance with the SFT data, we propose Thinking Preference Optimization (ThinkPO), a simple yet effective post-SFT method that enhances long CoT reasoning without requiring new long CoT responses. Instead, ThinkPO utilizes readily available or easily obtainable short CoT reasoning responses as rejected answers and long CoT responses as chosen answers for the same question. It then applies direct preference optimization to encourage the model to favor longer reasoning outputs. Experiments show that ThinkPO further improves the reasoning performance of SFT-ed models, e.g. it increases math reasoning accuracy of SFT-ed models by 8.6% and output length by 25.9%. Notably, ThinkPO is capable of continually boosting the performance of the publicly distilled SFT model, e.g., increasing the official DeepSeek-R1-Distill-Qwen-7B's performance on MATH500 from 87.4% to 91.2%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。