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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multi-task Language Understanding | MMLU | Accuracy68.32 | 353 | |
| Bias Evaluation | BBQ | Accuracy87.3 | 175 | |
| Image Classification | Vision Multi-task Suite (SUN397, Cars, RESISC45, EuroSAT, SVHN, GTSRB, MNIST, DTD) | Average Accuracy94.24 | 171 | |
| Truthfulness | TruthfulQA | Truthfulness Accuracy54.12 | 98 | |
| Visual Classification | 8 Vision Tasks (SUN397, Stanford Cars, RESISC45, EuroSAT, SVHN, GTSRB, MNIST, DTD) | Average Accuracy90.4 | 90 | |
| Image Classification | SUN397, Cars, EuroSAT, GTSRB, MNIST, DTD Seen Tasks (test) | SUN397 Accuracy0.8182 | 34 | |
| Image Classification | RESISC45, SVHN Unseen Tasks (test) | RESISC45 Accuracy72.98 | 34 | |
| Multi-task Classification | 30 image classification tasks average (test) | Average Accuracy92.41 | 23 | |
| Question Answering | MMLU, TruthfulQA, and BBQ | MMLU Accuracy68.32 | 21 | |
| Natural Language Understanding | GLUE | CoLA76.98 | 16 |