提出层间剪枝任务向量,提升模型融合在域外数据的表现。
Merging Smarter, Generalizing Better: Enhancing Model Merging on OOD Data
- 基于任务向量冗余度构建显著性评分,实现分层剪枝
- 在域外数据上性能显著提升,域内表现不下降
- 可无缝集成到多数现有融合方法中,适用性强
多任务学习(MTL)通过同时训练模型处理多种任务数据,挖掘共性特征以提升整体性能。近期研究致力于将多个独立模型参数合并为统一模型,避免依赖训练数据,扩展了MTL的应用场景。然而,现有方法主要关注域内(ID)数据上的表现,忽视了在域外(OOD)数据上的有效性。本文提出LwPTV(层间剪枝任务向量),通过构建显著性评分来衡量任务向量中参数的冗余程度,生成每项任务对应的掩码向量,实现对任务向量的分层剪枝,仅保留合并模型对应层的预训练参数。该方法具有高度灵活性,可无缝集成至多数现有模型融合方法中,显著提升其在OOD任务上的性能。大量实验表明,应用本方法后,模型在域外数据上的表现得到明显改善,同时保持了原有的域内性能。
原文摘要 · Abstract (English)
Multi-task learning (MTL) concurrently trains a model on diverse task datasets to exploit common features, thereby improving overall performance across the tasks. Recent studies have dedicated efforts to merging multiple independent model parameters into a unified model for MTL, thus circumventing the need for training data and expanding the scope of applicable scenarios of MTL. However, current approaches to model merging predominantly concentrate on enhancing performance within in-domain (ID) datasets, often overlooking their efficacy on out-of-domain (OOD) datasets. In this work, we proposed LwPTV (Layer-wise Pruning Task Vector) by building a saliency score, measuring the redundancy of parameters in task vectors. Designed in this way ours can achieve mask vector for each task and thus perform layer-wise pruning on the task vectors, only keeping the pre-trained model parameters at the corresponding layer in merged model. Owing to its flexibility, our method can be seamlessly integrated with most of existing model merging methods to improve their performance on OOD tasks. Extensive experiments demonstrate that the application of our method results in substantial enhancements in OOD performance while preserving the ability on ID tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。