arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2505.17266cs.CLcs.AI

Select2Reason: 为长链推理实现高效的指令微调数据选择

Select2Reason: Efficient Instruction-Tuning Data Selection for Long-CoT Reasoning

DataArc Tech Ltd.(DataArc科技有限公司);Hong Kong University of Science and Technology (Guangzhou)(香港科学与技术大学(广州));IDEA Research, International Digital Economy Academy(IDEA研究所,国际数字经济学院);Hithink RoyalFlush Information Network Co., Ltd(Hithink RoyalFlush信息网络有限公司)

Cehao Yang, Xueyuan Lin, Xiaojun Wu, Chengjin Xu, Xuhui Jiang, Honghao Liu, Hui Xiong, Jian Guo

AI总结:

Select2Reason通过高效数据选择提升长链推理性能,实验证明其在多个基准测试中表现优异。

摘要:

一种实用的方法是通过在由强大推理模型如DeepSeek-R1合成的指令数据集上进行监督微调,以激活预训练大语言模型的长链推理能力,这为强化学习提供了一个成本效益更高的替代方案。然而,具有超过10万样本的大型指令集会带来显著的训练开销,而有效的自动长链推理指令选择策略仍然未被探索。在本工作中,我们提出Select2Reason,一种新颖且高效的长链推理指令微调数据选择框架。从重新思考行为如自我纠正和回溯的出现角度出发,我们调查了可能决定长链推理指令质量的常见指标。Select2Reason利用一个量化器来估计问题的难度,并通过加权方案联合采用基于推理轨迹长度的启发式方法进行排名,以优先选择高实用性的示例。在OpenR1-Math-220k上的实验证明,仅在Select2Reason所选数据的10%上微调LLM,其性能可与或优于全数据微调和开源基线OpenR1-Qwen-7B在三个竞赛级别和六个综合数学基准测试中。进一步的实验突显了在不同数据规模下的可扩展性、推理过程中的效率以及其对其他指令池的适应性,成本极低。

英文摘要:

A practical approach to activate long chain-of-thoughts reasoning ability in pre-trained large language models is to perform supervised fine-tuning on instruction datasets synthesized by strong Large Reasoning Models such as DeepSeek-R1, offering a cost-effective alternative to reinforcement learning. However, large-scale instruction sets with more than 100k samples incur significant training overhead, while effective strategies for automatic long-CoT instruction selection still remain unexplored. In this work, we propose Select2Reason, a novel and efficient instruction-tuning data selection framework for long-CoT reasoning. From the perspective of emergence of rethinking behaviors like self-correction and backtracking, we investigate common metrics that may determine the quality of long-CoT reasoning instructions. Select2Reason leverages a quantifier to estimate difficulty of question and jointly incorporates a reasoning trace length-based heuristic through a weighted scheme for ranking to prioritize high-utility examples. Empirical results on OpenR1-Math-220k demonstrate that fine-tuning LLM on only 10% of the data selected by Select2Reason achieves performance competitive with or superior to full-data tuning and open-source baseline OpenR1-Qwen-7B across three competition-level and six comprehensive mathematical benchmarks. Further experiments highlight the scalability in varying data size, efficiency during inference, and its adaptability to other instruction pools with minimal cost.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑