arXiv:2503.10460cs.CLcs.LG2025-03ACL被引 191

用公开数据训练出性能媲美大模型的长链条推理模型

Light-R1: Curriculum SFT, DPO and RL for Long COT from Scratch and Beyond

论文配图:Light-R1: Curriculum SFT, DPO and RL for Long COT from Scratch and Beyond
图 1 · 摘自论文原文
  • 分阶段渐进式训练,结合多阶段微调提升模型能力
  • 32B模型数学推理超越深求32B蒸馏版,14B模型达74.0/60.2高分
  • 方法开源可复现,适合研究与实际部署使用

本文提出Light-R1,一个基于公开数据和模型的开源长链条推理模型训练框架。针对DeepSeek-R1系列依赖私有数据的问题,我们采用渐进式课程训练,结合多阶段后训练策略。以Qwen2.5-32B-Instruct为起点,训练出的Light-R1-32B在数学推理上优于DeepSeek-R1-Distill-Qwen-32B。在不同阶段使用多样数据集时,仅用3,000个挑战性样本微调已预训练的深求模型,即生成了顶尖的7B和14B模型;而32B版本的Light-R1-32B-DS表现接近QwQ-32B与DeepSeek-R1。进一步应用GRPO优化后,最终的Light-R1-14B-DS在数学任务上达到14B模型中的领先水平,AIME24与AIME25得分分别为74.0与60.2,超越多数32B模型及DeepSeek-R1-Distill-Llama-70B。尽管专注数学训练,该模型仍展现良好跨领域泛化能力。所有模型、数据与代码均已开源。

原文摘要 · Abstract (English)

This paper introduces Light-R1, an open-source suite for training long reasoning models using reproducible and cost-effective methodology. Given the proprietary nature of data used in the DeepSeek-R1 series, we develop an alternative approach leveraging exclusively public data and models. Our curriculum training progressively increases data difficulty, combined with multi-staged post-training. Our Light-R1-32B model, trained from Qwen2.5-32B-Instruct, outperforms DeepSeek-R1-Distill-Qwen-32B in math reasoning. Experimental results show that this curriculum approach becomes more effective when distinct, diverse datasets are available for different training stages: fine-tuning DeepSeek-R1-Distilled models (pre-tuned by DeepSeek team on proprietary data) with 3,000 challenging examples from our curriculum dataset yielded state-of-the-art 7B and 14B models, while the 32B model, Light-R1-32B-DS performed comparably to QwQ-32B and DeepSeek-R1. Furthermore, we extend our work by applying GRPO on long reasoning models. Our final Light-R1-14B-DS achieves SOTA performance among 14B models in math, with AIME24 & 25 scores of 74.0 and 60.2 respectively, surpassing many 32B models and DeepSeek-R1-Distill-Llama-70B. Despite math-focused training, Light-R1-14B-DS demonstrates strong cross-domain generalization. Light-R1 represents a significant advancement in making sophisticated reasoning models more accessible and implementable in real-world applications. Our models, training data and code have been made available at https://github.com/Qihoo360/Light-R1.

长链推理课程训练数学推理开源模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。