用平滑度感知微调减少模型合并时的参数干扰,提升多任务性能。
Mitigating Parameter Interference in Model Merging via Sharpness-Aware Fine-Tuning
- 通过平滑度感知微调设计新训练目标,降低不同任务间参数冲突。
- 在多个合并方法上实现性能提升,优于传统简单加权平均。
- 适合需要高效融合多个下游任务模型的研究者使用。
大规模深度学习模型通常采用预训练-微调范式,产生大量从同一预训练模型微调出的任务专用模型。近期研究尝试将这些大模型通过简单的参数算术操作合并为单一多任务模型,但面临核心挑战:不同任务微调后参数间的相互干扰。少数工作虽提出新微调方案以减少干扰,却牺牲了各任务模型性能,从而限制了合并模型的整体表现。为改善合并模型性能,我们提出微调目标应同时兼顾(1)更小的参数干扰和(2)各任务微调模型的更好表现。本研究发现该目标函数与平滑度感知最小化(SAM)极为相似,后者旨在通过寻找平坦极小值来提升泛化能力。基于此观察,我们提出采用SAM进行预训练模型的微调。实验与理论结果表明,所提方法有效且与其他方法正交,在多种合并与微调策略中均取得性能提升。代码已开源于 https://github.com/baiklab/SAFT-Merge。
原文摘要 · Abstract (English)
Large-scale deep learning models with a pretraining-finetuning paradigm have led to a surge of numerous task-specific models fine-tuned from a common pre-trained model. Recently, several research efforts have been made on merging these large models into a single multi-task model, particularly with simple arithmetic on parameters. Such merging methodology faces a central challenge: interference between model parameters fine-tuned on different tasks. Few recent works have focused on designing a new fine-tuning scheme that can lead to small parameter interference, however at the cost of the performance of each task-specific fine-tuned model and thereby limiting that of a merged model. To improve the performance of a merged model, we note that a fine-tuning scheme should aim for (1) smaller parameter interference and (2) better performance of each fine-tuned model on the corresponding task. In this work, we aim to design a new fine-tuning objective function to work towards these two goals. In the course of this process, we find such objective function to be strikingly similar to sharpness-aware minimization (SAM) objective function, which aims to achieve generalization by finding flat minima. Drawing upon our observation, we propose to fine-tune pre-trained models via sharpness-aware minimization. The experimental and theoretical results showcase the effectiveness and orthogonality of our proposed approach, improving performance upon various merging and fine-tuning methods. Our code is available at https://github.com/baiklab/SAFT-Merge.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。