arXiv:2508.19839cs.LGcs.AI2025-08被引 1

用粒子群优化合并模型,提升多任务模型性能与效率

PSO-Merging: Merging Models Based on Particle Swarm Optimization

  • 基于粒子群算法初始化并优化模型融合过程
  • 在多个语言模型上优于基线方法,收敛快且效果好
  • 适合需要高效合并大型专家模型的研究者使用

模型合并已成为通过整合多个可用专家模型的优势来构建多任务模型的有效策略,从而减少从头微调预训练模型以适应所有任务的需求。现有无数据依赖方法因缺乏数据驱动指导而性能受限;数据驱动方法也面临挑战:基于梯度的方法计算成本高,难以应用于大型专家模型的合并;而现有的无梯度方法在有限优化步数内往往无法取得满意结果。为此,本文提出PSO-Merging,一种基于粒子群优化(PSO)的新颖数据驱动合并方法。该方法以预训练模型、专家模型及稀疏化专家模型初始化粒子群,经过多次迭代后,将最终全局最优粒子作为合并模型。在不同语言模型上的实验表明,PSO-Merging普遍优于基线合并方法,为模型合并提供了更高效、可扩展的解决方案。

原文摘要 · Abstract (English)

Model merging has emerged as an efficient strategy for constructing multitask models by integrating the strengths of multiple available expert models, thereby reducing the need to fine-tune a pre-trained model for all the tasks from scratch. Existing data-independent methods struggle with performance limitations due to the lack of data-driven guidance. Data-driven approaches also face key challenges: gradient-based methods are computationally expensive, limiting their practicality for merging large expert models, whereas existing gradient-free methods often fail to achieve satisfactory results within a limited number of optimization steps. To address these limitations, this paper introduces PSO-Merging, a novel data-driven merging method based on the Particle Swarm Optimization (PSO). In this approach, we initialize the particle swarm with a pre-trained model, expert models, and sparsified expert models. We then perform multiple iterations, with the final global best particle serving as the merged model. Experimental results on different language models show that PSO-Merging generally outperforms baseline merging methods, offering a more efficient and scalable solution for model merging.

模型合并粒子群优化多任务学习

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