用强化学习优化专家模型,仅用三分之一参数达到顶尖推理性能
Ring-lite: Scalable Reasoning via C3PO-Stabilized Reinforcement Learning for LLMs
- 通过联合蒸馏与强化学习训练,提升模型稳定性
- 在多个难题上表现媲美顶级小模型,激活参数减少67%
- 适合追求高效推理的AI研发者与系统优化人员
我们提出Ring-lite,一种基于混合专家(MoE)架构的大语言模型,通过强化学习(RL)优化以实现高效且鲁棒的推理能力。基于公开的Ling-lite模型(168亿参数,激活27.5亿参数),该方法在AIME、LiveCodeBench、GPQA-Diamond等挑战性基准测试中表现媲美当前最优小型推理模型,但仅需同类模型三分之一的激活参数。为此,我们设计了融合蒸馏与强化学习的联合训练流程,揭示了MoE强化学习训练中的未被记录的挑战。首先,发现强化学习训练中存在优化不稳定性,提出约束上下文计算策略优化(C3PO),通过算法-系统协同设计提升训练稳定性并提高计算吞吐量。其次,实证表明基于熵损失选择蒸馏检查点进行强化学习训练,相比使用验证指标,能获得更优的性能-效率权衡。最后,构建两阶段训练范式以协调多领域数据融合,解决混合数据训练中的领域冲突问题。模型、数据集和代码将开源。
原文摘要 · Abstract (English)
We present Ring-lite, a Mixture-of-Experts (MoE)-based large language model optimized via reinforcement learning (RL) to achieve efficient and robust reasoning capabilities. Built upon the publicly available Ling-lite model, a 16.8 billion parameter model with 2.75 billion activated parameters, our approach matches the performance of state-of-the-art (SOTA) small-scale reasoning models on challenging benchmarks (e.g., AIME, LiveCodeBench, GPQA-Diamond) while activating only one-third of the parameters required by comparable models. To accomplish this, we introduce a joint training pipeline integrating distillation with RL, revealing undocumented challenges in MoE RL training. First, we identify optimization instability during RL training, and we propose Constrained Contextual Computation Policy Optimization(C3PO), a novel approach that enhances training stability and improves computational throughput via algorithm-system co-design methodology. Second, we empirically demonstrate that selecting distillation checkpoints based on entropy loss for RL training, rather than validation metrics, yields superior performance-efficiency trade-offs in subsequent RL training. Finally, we develop a two-stage training paradigm to harmonize multi-domain data integration, addressing domain conflicts that arise in training with mixed dataset. We will release the model, dataset, and code.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。