arXiv:2502.11284cs.LG2025-02ACL被引 6

在有限数据预算下,先小规模SFT再做PFT能显著提升模型表现。

Balancing the Budget: Understanding Trade-offs Between Supervised and Preference-Based Finetuning

  • 先用少量数据做SFT,再分配更多预算给PFT,效果更优
  • 完全跳过SFT会使数学等任务表现下降15-20%
  • 适合预算紧张但需高质量输出的研究者和工程师

大语言模型的后训练通常包含监督微调(SFT)和偏好微调(PFT)两个阶段,两者所需标注数据结构与成本差异巨大。本文通过覆盖四个不同任务、多种模型规模和数据标注成本的实验,研究如何在固定数据预算下最优分配资源。结果表明,在低数据量场景(<1,000条)时,仅使用SFT即可取得最佳性能;随着预算增加,结合SFT与PFT并逐步提高偏好数据比例能实现最优表现。然而,直接在基础模型上仅运行PFT会因分布偏移导致性能下降,尤其在数学推理任务中出现‘冷启动’问题。通过仅投入<10%预算进行初步SFT,可在GSM8k等分析性基准上提升15-20%表现。这些发现为预算受限下的模型优化提供了可操作建议。

原文摘要 · Abstract (English)

Post-training of Large Language Models often involves a pipeline of Supervised Finetuning (SFT) followed by Preference Finetuning (PFT) using methods like Direct Preference Optimization. Both stages require annotated data that are very different in structure and costs. We study how to optimally allocate a fixed training data budget between the two stages, through extensive experiments spanning four diverse tasks, multiple model sizes and various data annotation costs. Our findings reveal that just SFT on the base model dominates performance in low-data regimes ($<1,000$ annotated examples). With larger data-budgets, we observe that a combination of SFT and PFT, often with increasing portions allocated towards preference data yields optimal performance. However, completely eliminating SFT and running PFT directly on the base model yields suboptimal performance, described as the cold start problem on tasks like mathematics. We observe that this is due to the distribution shift arising from using DPO directly on the base model to elicit step-by-step reasoning. This limitation can be effectively addressed by allocating even a small portion ($<10$%) of the budget to SFT first, resulting in performance improvements of $15-20$% on analytical benchmarks like GSM8k. These results provide actionable insights for researchers and practitioners optimizing model development under budget constraints, where high-quality data curation often represents a significant portion of the total costs of model development.

微调策略数据预算偏好优化SFT/PFT

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。