用动态权重融合专家数据与强化学习,避免模型行为突变
On-Policy RL Meets Off-Policy Experts: Harmonizing Supervised Fine-Tuning and Reinforcement Learning via Dynamic Weighting
- 将监督微调视为强化学习中的动态辅助目标
- 在多个任务上显著提升训练稳定性与性能表现
- 适合需要安全微调大模型的实践者使用
监督微调(SFT)和强化学习(RL)是优化大语言模型能力与对齐行为的两种主流后训练范式。现有融合SFT与RL的方法常导致已建立的响应模式被破坏,并过度拟合专家数据。为此,本文从“离策略”与“在线策略”的视角出发,提出CHORD框架——一种通过动态加权实现在线与离策略强化学习协同的可控谐调方法。该框架将SFT重新定义为在线策略强化学习过程中的动态加权辅助目标。基于对离策略专家数据在整体与细粒度层面影响的分析,CHORD引入双控制机制:首先使用全局系数引导从离策略模仿向在线探索的平稳过渡;其次采用词元级加权函数,实现对专家数据的精细学习,促进在线探索并减少离策略数据带来的干扰。我们在多种实际任务上进行了广泛实验,实证表明CHORD实现了稳定高效的训练过程,显著优于基线方法。代码已开源,以推动后续研究。
原文摘要 · Abstract (English)
Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL) are two prominent post-training paradigms for refining the capabilities and aligning the behavior of Large Language Models (LLMs). Existing approaches that integrate SFT and RL often face the risk of disrupting established response patterns and inducing overfitting to expert data. To address this, we present a novel investigation into the unified view of SFT and RL through an off-policy versus on-policy lens. We propose CHORD, a framework for Controllable Harmonization of On- and Off-Policy Reinforcement Learning via Dynamic Weighting, which reframes SFT not as a separate stage but as a dynamically weighted auxiliary objective within the on-policy RL process. Based on an analysis of off-policy expert data's influence at both holistic and granular levels, we incorporate a dual-control mechanism in CHORD. Specifically, the framework first employs a global coefficient to holistically guide the transition from off-policy imitation to on-policy exploration, and then applies a token-wise weighting function that enables granular learning from the expert, which promotes on-policy exploration and mitigates disruption from off-policy data. We conduct extensive experiments across various practical tasks, providing empirical evidence that CHORD achieves a stable and efficient learning process. By effectively harmonizing off-policy expert data with on-policy exploration, CHORD demonstrates significant improvements over baselines. We release the implementation at https://github.com/modelscope/Trinity-RFT/tree/main/examples/mix_chord to inspire further research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。