arXiv:2504.18992cs.CL2025-04NAACL被引 14

用贝叶斯优化动态调整模型合并权重,提升多任务表现

Dynamic Fisher-weighted Model Merging via Bayesian Optimization

  • 通过贝叶斯优化动态调节各模型参数权重,融合时考虑参数重要性
  • 在多种模型和任务上超越基线,少量验证数据下即达近优性能
  • 适合需要高效集成多个微调模型的场景,尤其资源受限时

预训练语言模型的微调催生了大量任务专用模型。模型合并可在不依赖训练数据或联合训练的情况下,通过参数级融合生成多任务模型。现有方法通常采用模型级缩放或参数级重要性整合,但各有缺陷,导致性能显著低于多任务微调。本文将两类策略统一为更通用的合并框架,提出动态费雪加权合并(DF-Merge)。候选模型配备一组线性缩放其微调参数的系数,利用贝叶斯优化动态调整这些系数,以最大化验证集上的整体性能。每次迭代均基于系数条件下的费雪信息整合参数重要性。实验表明,DF-Merge 在不同规模模型与多种任务上均优于强基线。分析显示,其有效性源于合并的统一视角,且仅需少数迭代即可实现近最优性能,即使验证数据极少亦可。

原文摘要 · Abstract (English)

The fine-tuning of pre-trained language models has resulted in the widespread availability of task-specific models. Model merging offers an efficient way to create multi-task models by combining these fine-tuned models at the parameter level, without the need for training data or joint training on multiple datasets. Existing merging approaches typically involve scaling the parameters model-wise or integrating parameter importance parameter-wise. Both approaches exhibit their own weaknesses, leading to a notable performance gap compared to multi-task fine-tuning. In this paper, we unify these seemingly distinct strategies into a more general merging framework, and introduce Dynamic Fisher-weighted Merging (DF-Merge). Specifically, candidate models are associated with a set of coefficients that linearly scale their fine-tuned parameters. Bayesian optimization is applied to dynamically adjust these coefficients, aiming to maximize overall performance on validation sets. Each iteration of this process integrates parameter importance based on the Fisher information conditioned by the coefficients. Experimental results show that DF-Merge outperforms strong baselines across models of different sizes and a variety of tasks. Our analysis shows that the effectiveness of DF-Merge arises from the unified view of merging and that near-optimal performance is achievable in a few iterations, even with minimal validation data.

模型合并贝叶斯优化多任务学习

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