提前混合训练数据能显著提升模型后续微调的稳定性。
Early Data Exposure Improves Robustness to Subsequent Fine-Tuning

- 在预训练阶段提前引入目标数据,增强模型对后续微调的抗遗忘能力。
- 实验显示,早期暴露可使上游性能保留率提升20%以上(135M/1B模型)。
- 适合关注模型长期稳定性的研究者与工业应用开发者。
如何训练出在后续微调中仍能保持上游能力的模型?我们不依赖下游干预来缓解遗忘,而是研究上游训练方式——即能力获取方式——如何影响其鲁棒性。在三阶段语言模型流程(预训练、后训练获取目标能力、下游微调)中,我们在135M和1B模型、两个后训练领域、两个下游任务上发现:后训练后的即时表现无法可靠预测微调后的保留效果;看似相同的训练方案,其保留能力差异显著。特别地,早期暴露(将后训练数据混入预训练)能持续提升上游性能保留与下游性能之间的权衡边界。在计算量匹配实验中,最优分配并非极端偏向预训练或后训练。结合实证与理论分析,表明后训练促进即时专业化,而早期暴露提升对后续遗忘的鲁棒性。重放与丢弃法虽用于缓解微调中的遗忘,但与早期暴露具有互补作用。研究建议:应将对后续微调的鲁棒性作为上游训练的核心目标,通过早期暴露等预防性策略实现,而非事后应对。
原文摘要 · Abstract (English)
How can we train models whose post-trained capabilities survive subsequent fine-tuning? Rather than focusing on downstream interventions to mitigate forgetting of upstream capabilities, we study how upstream training choices - that is, the manner in which a capability is acquired - shape how robustly that capability is retained. We investigate this question in a controlled three-stage language-model pipeline: pretraining, post-training to acquire a target capability, and downstream fine-tuning on a new objective. Across 135M and 1B models, two post-training domains, and two downstream fine-tuning tasks, we find that immediate post-training performance does not reliably predict retention after subsequent fine-tuning: training recipes that look equivalent immediately after post-training can retain the target capability very differently after subsequent fine-tuning. In particular, early exposure - mixing post-training data into pretraining - consistently improves the frontier between retained upstream performance and downstream performance. In compute-matched experiments, where the target data must be allocated between pretraining and post-training, we find that the optimum lies at neither extreme. Together with our other empirical and theoretical findings, this supports the view that post-training drives immediate specialization while early exposure improves robustness to later forgetting. Replay and dropout, typically used to mitigate forgetting as it occurs during fine-tuning, provide complementary gains to early exposure when applied during post-training. Our findings suggest that robustness to subsequent fine-tuning should be treated as a first-class objective of upstream training, addressed preventatively through choices like early exposure rather than reactively during fine-tuning itself.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。