arXiv:2505.16066cs.LGcs.AI2025-05被引 7

用模型合并快速找到最优数据集组合,省去多次调参

Merge to Mix: Mixing Datasets via Model Merging

  • 用模型合并代替逐个训练混合数据集
  • 在多个任务上优于现有方法,准确率提升显著
  • 适合需要高效调优大模型的科研与工程人员

为大型语言模型(LMs)微调而混合数据集已成为提升下游任务性能的关键。然而,设计有效的数据集组合通常依赖经验法则和反复试验,常需多次微调才能达到理想效果。本文提出新方法「Merge to Mix」,通过模型合并技术加速数据集组合的构建。模型合并是一种将多个独立微调过的语言模型通过简单算术运算融合为单一模型的技术。核心洞察是:对每个数据集单独微调后的模型进行合并,可有效替代在完整混合数据集上微调的模型。因此,Merge to Mix 可在无需对每个候选组合进行完整微调的情况下,快速筛选出最佳数据集组合。实验表明,该方法在大型语言模型微调的数据集选择任务中显著超越现有最先进方法。

原文摘要 · Abstract (English)

Mixing datasets for fine-tuning large models (LMs) has become critical for maximizing performance on downstream tasks. However, composing effective dataset mixtures typically relies on heuristics and trial-and-error, often requiring multiple fine-tuning runs to achieve the desired outcome. We propose a novel method, $\textit{Merge to Mix}$, that accelerates composing dataset mixtures through model merging. Model merging is a recent technique that combines the abilities of multiple individually fine-tuned LMs into a single LM by using a few simple arithmetic operations. Our key insight is that merging models individually fine-tuned on each dataset in a mixture can effectively serve as a surrogate for a model fine-tuned on the entire mixture. Merge to Mix leverages this insight to accelerate selecting dataset mixtures without requiring full fine-tuning on each candidate mixture. Our experiments demonstrate that Merge to Mix surpasses state-of-the-art methods in dataset selection for fine-tuning LMs.

模型合并数据集混合微调优化

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