对比监督与强化学习微调,发现后者主要修复前者引起的模型方向偏移。
RL Is Neither a Panacea Nor a Mirage: Understanding Supervised vs. Reinforcement Learning Fine-Tuning for LLMs
- 通过奇异向量方向分析,揭示微调中模型表征变化的关键机制。
- 恢复70%-80%的分布外性能仅需重置顶层25%层或前20%奇异向量方向。
- 强监督微调检查点更易被强化学习修复,过拟合则难恢复。
训练大语言模型从头开始日益不切实际,因此监督微调(SFT)和强化学习微调(RL-FT,如PPO)成为现代实践的核心。利用分布外(OOD)的24点纸牌游戏变体及新的谱分析诊断工具,我们重新审视这两种阶段如何重塑模型表示与分布外性能。关键发现:(1)RL-FT可显著恢复SFT导致的分布外性能下降(如Llama-11B从8.97%提升至15.38%,Qwen-7B从17.09%升至19.66%),但若SFT引发严重过拟合与明显分布偏移,RL-FT无法完全恢复;(2)奇异向量的方向变化比奇异值大小更重要,且集中在最大与最小奇异值对应的方向上,整体谱结构保持稳定;(3)低秩与浅层恢复有效:仅恢复前20%奇异值对应的向量方向或前25%层的参数,即可恢复70%-80%的分布外性能;(4)更强的SFT检查点更利于被RL恢复,而过拟合版本则难以修复。这些结果解释了此前关于RL在分布外性能上优势的矛盾报告——其本质是修正SFT引入的方向漂移,而非发现新解。谱感知分析揭示了低成本的恢复手段,如低秩UV合并与浅层重置,可在昂贵的强化学习微调前使用。
原文摘要 · Abstract (English)
Training large language models (LLMs) from scratch is increasingly impractical, making post-training methods such as supervised fine-tuning (SFT) and reinforcement-learning fine-tuning (RL-FT, e.g., PPO) central to modern practice. Using an out-of-distribution (OOD) variant of the 24-point card game and new spectrum-based diagnostics, we revisit how these two stages reshape model representation and OOD performance. Our key findings are- (1) RL-FT can restore much of the OOD performance loss from SFT (e.g., Llama-11B 8.97% to 15.38%, Qwen-7B 17.09% to 19.66%). But when SFT induces severe overfitting and a clear distribution shift, RL-FT cannot fully recover OOD performance. (2) Direction shifts of singular vectors matter more than singular value magnitudes. These shifts concentrate on directions linked to the largest and smallest singular values, leaving the bulk spectrum intact. (3) Low-rank and shallow recovery is effective: restoring singular vector directions for the top 20% of values or first 25% of layers recovers 70-80% of OOD performance. (4) Stronger SFT checkpoints enable better recovery by RL, while overfitted ones resist restoration. These results reconcile prior reports of RL superior OOD performance: RL primarily counteracts SFT-induced directional drift rather than finding new solutions. Our spectrum-aware analysis highlights inexpensive recovery knobs low-rank UV merging and shallow-layer resets that practitioners can use before costly RL fine-tuning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。