动态融合微调与强化学习,提升大模型推理能力
Step-wise Adaptive Integration of Supervised Fine-tuning and Reinforcement Learning for Task-Specific LLMs
- 分步自适应融合SFT与RL,根据训练状态自动调整
- 在数学推理任务上超越单一方法和静态混合方案
- 适合需要稳定推理性能的定制化大模型应用
大语言模型在数学推理和逻辑问题求解方面表现优异。当前主流训练范式主要依赖监督微调(SFT)和强化学习(RL)来增强模型推理能力。然而,单独使用SFT易过拟合,而RL则存在模式崩溃风险。现有混合训练方法多采用静态切换,存在任务泛化能力差、对数据质量依赖高等问题。为此,受人类推理培养中“课程学习-测验”机制启发,我们提出SASR——一种分步自适应混合训练框架,理论上统一SFT与RL,并在优化过程中动态平衡二者。SASR先用SFT进行初始热身以建立基础推理能力,随后基于梯度范数和分布偏离度设计自适应调整算法,无缝集成在线强化学习方法GRPO。通过监控模型训练状态并按序调整训练流程,SASR实现训练方案的平滑过渡,既保持核心推理能力又探索多样化路径。实验表明,SASR在数学推理任务上优于SFT、RL及静态混合方法。
原文摘要 · Abstract (English)
Large language models (LLMs) excel at mathematical reasoning and logical problem-solving. The current popular training paradigms primarily use supervised fine-tuning (SFT) and reinforcement learning (RL) to enhance the models' reasoning abilities. However, when using SFT or RL alone, there are respective challenges: SFT may suffer from overfitting, while RL is prone to mode collapse. The state-of-the-art methods have proposed hybrid training schemes. However, static switching faces challenges such as poor generalization across different tasks and high dependence on data quality. In response to these challenges, inspired by the curriculum learning-quiz mechanism in human reasoning cultivation, We propose SASR, a step-wise adaptive hybrid training framework that theoretically unifies SFT and RL and dynamically balances the two throughout optimization. SASR uses SFT for initial warm-up to establish basic reasoning skills, and then uses an adaptive dynamic adjustment algorithm based on gradient norm and divergence relative to the original distribution to seamlessly integrate SFT with the online RL method GRPO. By monitoring the training status of LLMs and adjusting the training process in sequence, SASR ensures a smooth transition between training schemes, maintaining core reasoning abilities while exploring different paths. Experimental results demonstrate that SASR outperforms SFT, RL, and static hybrid training methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。