解决大模型微调时因数据差异导致的训练不稳问题
Compatibility-Aware Dynamic Fine-Tuning for Large Language Models

- 根据模型预测概率动态计算样本兼容性,抑制不匹配数据的梯度
- 实验显示训练更稳定,泛化能力提升,冷启动强化学习效果更好
- 适合追求训练稳定性和泛化性能的LLM开发者
监督微调(SFT)是对齐大语言模型(LLMs)的主要方法,但存在优化不稳定和泛化能力有限的问题。现有研究将此归因于病态梯度缩放,并提出动态微调(DFT)在词元层面进行修正。然而,DFT假设所有示范数据均适合作为学习目标,这一假设在大规模指令数据中因示范与策略不匹配而失效,导致样本级高方差更新。我们提出兼容性感知动态微调(CADFT),从模型似然中推导出动态、依赖策略的兼容性信号,以调节监督更新,抑制不兼容示范带来的高方差梯度。进一步提出延迟、低频的兼容性引导重写策略,将持续不兼容的示范转化为可学习目标。实验证明,CADFT可视为一种方差控制估计器,将DFT的词元级稳定扩展至样本级。大量实验表明,该方法提升了稳定性、泛化能力及冷启动强化学习初始化效果,且保持全监督特性,无需显式奖励建模。
原文摘要 · Abstract (English)
Supervised Fine-Tuning (SFT) is the predominant paradigm for aligning large language models (LLMs), yet it suffers from optimization instability and limited generalization. Recent work attributes this issue to pathological gradient scaling and proposes Dynamic Fine-Tuning (DFT) to correct it at the token level. However, DFT assumes all demonstrations are equally suitable learning targets, an assumption violated by the strong heterogeneity of large-scale instruction data, where demonstration-policy mismatch induces high-variance updates at the sample level. We introduce Compatibility-Aware Dynamic Fine-Tuning (CADFT), a principled extension of DFT that controls sample-level optimization variance. CADFT derives a dynamic, policy-dependent compatibility signal from model likelihoods to modulate supervised updates, suppressing high-variance gradients from incompatible demonstrations. We further propose a delayed, low-frequency compatibility-guided rewriting strategy to transform persistently incompatible demonstrations into learnable targets. We show that CADFT can be interpreted as a variance-controlled estimator that generalizes token-level stabilization in DFT to the sample level. Extensive experiments demonstrate improved stability, generalization, and cold-start reinforcement learning initialization, while remaining fully supervised and independent of explicit reward modeling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。