解决推理模型强化学习对齐中的训练不稳问题。
Balanced Actor Initialization: Stable RLHF Training of Distillation-Based Reasoning Models
- 通过两阶段加权合并,平衡指令微调与蒸馏模型
- 消除序列长度坍塌,缓解奖励曲线陡降现象
- 适合需要稳定对齐的高效推理模型开发
大型语言模型在对齐与推理能力的发展中,已通过指令微调和基于人类反馈的强化学习(RLHF)对齐范式,以及基于蒸馏的推理微调范式取得显著进展。尽管二者独立有效,但将RLHF应用于蒸馏训练模型时面临严峻挑战。我们发现该范式中存在两个关键现象:序列长度坍塌(生成长度在早期训练中急剧下降)和奖励冰球棒曲线(奖励分数严重下跌后缓慢回升)。这些不稳定性从根本上损害了模型的对齐与推理能力。为此,我们提出平衡演员初始化(BAI),一种两阶段加权模型合并方法:先融合指令遵循与蒸馏推理微调模型,再进一步结合预训练模型以保留基础知识。在多个基准上的全面实验与训练过程分析表明,BAI可有效解决序列长度坍塌,缓解奖励冰球棒曲线,并实现训练期间序列长度持续提升。分析还显示,平衡合并比例在训练稳定性与推理能力保留间达到最优权衡。本工作为该第三范式提供了稳定训练的有效方案,使兼具蒸馏效率与RLHF对齐能力的更强大推理模型成为可能。
原文摘要 · Abstract (English)
The development of alignment and reasoning capabilities in large language models has seen remarkable progress through two paradigms: instruction tuning and reinforcement learning from human feedback (RLHF) alignment paradigm, and distillation-based reasoning fine-tuning paradigm. While both approaches prove effective independently, the third paradigm of applying RLHF to distillation-trained models presents significant challenges. Our investigation reveals two critical phenomena that emerge in this paradigm: Sequence Length Collapse, where language generation dramatically reduces during early RLHF training, and the Reward Hockey Stick Curve, featuring severe reward score drops followed by gradual recovery. These instabilities fundamentally compromise the model's alignment and reasoning capabilities. To address these challenges, we propose Balanced Actor Initialization (BAI), a two-stage weighted model merging approach. BAI first merges instruction-following and distillation-based reasoning fine-tuned models, then further combines this intermediate model with the pretrained model to preserve foundational knowledge. Through comprehensive experiments across diverse benchmarks and detailed analysis of training experiments, we demonstrate that BAI resolves Sequence Length Collapse, mitigates the Reward Hockey Stick Curve, and enables continuous sequence length improvement during training. Additionally, our analysis reveals that balanced merging ratios achieve optimal trade-offs between training stability and reasoning capability preservation. Our work provides the effective solution for stable training in this third paradigm, enabling more capable reasoning models that combine distillation efficiency with RLHF alignment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。