北京大学杨仝教授团队近期发布了其在高效大型语言模型研究方向的一项新成果—— FairyR1-32B 模型。该模型基于 DeepSeek-R1-Distill-Qwen-32B 基座,通过结合微调与模型合并技术构建。研究探索了在参数量大幅减少的情况下,模型在特定任务上实现与更大模型相当甚至更优性能的可能性。该研究得到了国家自然科学基金委项目(624B2005, 62372009)的资助。FairyR1-32B 模型已在 huggingface 开源:https://huggingface.co/PKU-DS-LAB/FairyR1-32B。
FairyR1-32B 模型是在团队前期 TinyR1 工作基础上进行的进一步探索,沿用了 " 分合蒸馏 " 的研究思路,提出了多种改进方法,包括自我合并、多教师交叉蒸馏、轻蒸馏等方法,并在数据处理进行了优化,模型精度有了显著提升。
本次工作重点改进了蒸馏数据的构建流程,对来源于 AI-MO/NuminaMath-1.5(数学)和 open-thoughts/OpenThoughts-114k(代码)等数据集的原始数据,通过多个 " 教师模型 " 生成答案,随后对问答数据进行精心筛选、结构调整与思维链优化,并进行多阶段筛选。筛选过程包括基于答案的正确性验证(针对数学数据),以及基于长度的筛选(数学数据保留 2k-8k tokens 范围,代码数据保留 4k-8k tokens 范围),最终构建了更具针对性的约 6.6k 条数学数据和约 3.8k 条代码数据用于训练。
在模型结构方面,研究团队尝试训练两个领域(数学和代码)的专业模型进行合并,旨在进一步优化流程和资源消耗。这两个专业模型在一致的训练参数下(例如相同的学习率和批次大小)独立训练约 5 个周期后,利用 AcreeFusion 工具进行了合并。在多个公开基准测试中,FairyR1 展现出了在低参数量下的竞争力表现。以下为 FairyR1 与 DeepSeek-R1-671B 及 DeepSeek-R1-Distill-Qwen-32B 在部分基准上的得分对比:
从测试结果可以看出,FairyR1-32B 在 AIME 2025 和 LiveCodeBench 基准上得分略高于 DeepSeek-R1-671B,在 AIME 2024 上表现接近。在 GPQA-Diamond 科学基准上,FairyR1 的得分低于 DeepSeek-R1-671B。这些结果表明,FairyR1 在采用 DeepSeek-R1-Distill-Qwen-32B 基座并经过特定技术处理后,能够在约 5% 参数量的情况下,在数理和编程等领域实现与大型模型相当或略优的性能水平,但在科学等其他领域可能存在差距。这项工作探索了通过优化的数据处理和模型融合技术,在保证特定任务性能的前提下,大幅降低模型规模和潜在推理成本的可能性。
北京大学杨仝教授团队表示:"FairyR1-32B 模型是我们探索高效大型语言模型技术路线的阶段性成果。通过对蒸馏和合并方法的改进,我们初步验证了在有限资源下实现高性能模型的可行性。"
团队成员:李旺、周俊廷、刘文睿、姚一伦、王融乐、杨仝
登录后才可以发布评论哦
打开小程序可以发布评论哦