Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Stay Unique, Stay Efficient: Preserving Model Personality in Multi-Task Merging

About

Model merging has emerged as a promising paradigm for enabling multi-task capabilities without additional training. However, traditional basic merging methods often experience performance degradation due to parameter conflicts, even when applied to similar tasks. While recent personalized merging frameworks successfully preserve task-specific information to maintain performance, they typically incur storage overhead. In this paper, we propose Decomposition, Thresholding, and Scaling (DTS), an approximation-based personalized merging framework that pushes task-specific storage efficiency. DTS first applies singular value decomposition to the task-specific information and retains only a small subset of singular values and vectors. It then introduces a novel thresholding strategy that partitions singular vector elements into groups and assigns a scaling factor to each group. To enable generalization to unseen tasks, we further extend DTS with a variant that fuses task-specific information in a data-free manner based on the semantic similarity of task characteristics. Extensive experiments demonstrate that DTS consistently outperforms state-of-the-art baselines while requiring only 1\% extra storage per task. Furthermore, experiments on unseen tasks show that the DTS variant achieves significantly better generalization performance. Our code is available at https://github.com/krumpguo/DTS.

Kuangpu Guo, Aijing Yu, Jian Liang, Yuhe Ding, Zilei Wang, Ran He, Tieniu Tan• 2025

Related benchmarks

TaskDatasetResultRank
Multi-task Language UnderstandingMMLU
Accuracy68.32
353
Bias EvaluationBBQ
Accuracy87.3
175
Image ClassificationVision Multi-task Suite (SUN397, Cars, RESISC45, EuroSAT, SVHN, GTSRB, MNIST, DTD)
Average Accuracy94.24
171
TruthfulnessTruthfulQA
Truthfulness Accuracy54.12
98
Visual Classification8 Vision Tasks (SUN397, Stanford Cars, RESISC45, EuroSAT, SVHN, GTSRB, MNIST, DTD)
Average Accuracy90.4
90
Image ClassificationSUN397, Cars, EuroSAT, GTSRB, MNIST, DTD Seen Tasks (test)
SUN397 Accuracy0.8182
34
Image ClassificationRESISC45, SVHN Unseen Tasks (test)
RESISC45 Accuracy72.98
34
Multi-task Classification30 image classification tasks average (test)
Average Accuracy92.41
23
Question AnsweringMMLU, TruthfulQA, and BBQ
MMLU Accuracy68.32
21
Natural Language UnderstandingGLUE
CoLA76.98
16
Showing 10 of 12 rows

Other info

Follow for update