提出高效模型合并方法,提升多任务大模型的鲁棒性与泛化能力。
RobustMerge: Parameter-Efficient Model Merging for MLLMs with Direction Robustness
- 通过参数剪枝与系数调节,保持合并方向稳定
- 跨任务归一化提升未见任务的泛化性能
- 无需训练,适合轻量化多模态模型合并
使用定制数据微调预训练模型会产生大量特定任务的专家模型。将多个模型合并为一个通用模型以实现多任务能力,同时避免数据泄露,已成为流行趋势。随着数据和模型规模的扩大,参数高效微调已成为高效获取特定任务模型的常用方法。然而,针对高效合并的方法仍很少,且现有适用于全量微调合并的方法在参数高效场景下表现不佳。为此,我们从低秩分解角度分析,发现合并过程中的方向鲁棒性对高效模块合并至关重要,并进一步揭示:补偿显著不同的奇异值差距有助于提升方向鲁棒性。因此,我们提出 RobustMerge,一种无需训练的参数高效合并方法,通过互补参数适配维持方向鲁棒性。具体而言,(1) 通过参数剪枝与系数缩放,基于参数间关系调整奇异值,保持方向稳定性以规避任务干扰;(2) 进行跨任务归一化,增强对未见任务的泛化能力。我们构建了一个涵盖多种多模态任务的基准测试集,在其上验证了该方法在性能与泛化性上的优异表现。额外实验与深入分析进一步证明其有效性。代码已开源:https://github.com/AuroraZengfh/RobustMerge。
原文摘要 · Abstract (English)
Fine-tuning pre-trained models with custom data leads to numerous expert models on specific tasks. Merging models into one universal model to empower multi-task ability refraining from data leakage has gained popularity. With the expansion in data and model size, parameter-efficient tuning becomes the common practice for obtaining task-specific models efficiently. However, few methods are dedicated to efficient merging, and existing methods designed for full fine-tuning merging fail under efficient merging. To address the issue, we analyze from low-rank decomposition and reveal that direction robustness during merging is crucial for merging efficient modules. We furthermore uncover that compensating for the gap between stark singular values contributes to direction robustness. Therefore, we propose RobustMerge, a training-free parameter-efficient merging method with complementary parameter adaptation to maintain direction robustness. Specifically, we (1) prune parameters and scale coefficients from inter-parameter relation for singular values to maintain direction stability away from task interference, and (2) perform cross-task normalization to enhance unseen task generalization. We establish a benchmark consisting of diverse multimodal tasks, on which we conduct experiments to certify the outstanding performance and generalizability of our method. Additional studies and extensive analyses further showcase the effectiveness. Code is available at https://github.com/AuroraZengfh/RobustMerge.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。