大规模模型合并效果受基底模型质量与规模影响显著。
What Matters for Model Merging at Scale?
- 用4种方法合并1B-64B参数模型,研究规模对融合效果的影响。
- 合并8个大模型后,泛化能力优于多任务训练模型。
- 大模型更易合并,且不同方法在大尺度下表现趋同。
模型合并旨在将多个专家模型整合为单一更强大模型,可降低存储与部署成本、提升泛化能力并支持分布式开发。然而以往研究多聚焦小模型合并,缺乏对模型规模扩大时关键因素(如基底模型质量、专家数量)影响的系统评估。本研究系统考察了大规模模型合并的效用,采用4种主流方法(Averaging、Task-Arithmetic、Dare、TIES)合并1B至64B参数模型,最多融合8个专家模型。评估涵盖原任务表现及零样本泛化能力。结果表明:合并效果随基底模型零样本性能提升而增强;大模型更易合并;合并始终提升泛化能力;合并更多专家模型在大模型上更有效;不同方法在大尺度下表现趋于一致。研究揭示了大规模合并的关键规律,也为未来工作提供参考。
原文摘要 · Abstract (English)
Model merging aims to combine multiple expert models into a more capable single model, offering benefits such as reduced storage and serving costs, improved generalization, and support for decentralized model development. Despite its promise, previous studies have primarily focused on merging a few small models. This leaves many unanswered questions about the effect of scaling model size and how it interplays with other key factors -- like the base model quality and number of expert models -- , to affect the merged model's performance. This work systematically evaluates the utility of model merging at scale, examining the impact of these different factors. We experiment with merging fully fine-tuned models using 4 popular merging methods -- Averaging, Task~Arithmetic, Dare, and TIES -- across model sizes ranging from 1B-64B parameters and merging up to 8 different expert models. We evaluate the merged models on both held-in tasks, i.e., the expert's training tasks, and zero-shot generalization to unseen held-out tasks. Our experiments provide several new insights about model merging at scale and the interplay between different factors. First, we find that merging is more effective when experts are created from strong base models, i.e., models with good zero-shot performance. Second, larger models facilitate easier merging. Third merging consistently improves generalization capabilities. Notably, when merging 8 large expert models, the merged models often generalize better compared to the multitask trained models. Fourth, we can better merge more expert models when working with larger models. Fifth, different merging methods behave very similarly at larger scales. Overall, our findings shed light on some interesting properties of model merging while also highlighting some limitations. We hope that this study will serve as a reference point on large-scale merging for upcoming research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。