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

Twin-Merging: Dynamic Integration of Modular Expertise in Model Merging

About

In the era of large language models, model merging is a promising way to combine multiple task-specific models into a single multitask model without extra training. However, two challenges remain: (a) interference between different models and (b) heterogeneous data during testing. Traditional model merging methods often show significant performance gaps compared to fine-tuned models due to these issues. Additionally, a one-size-fits-all model lacks flexibility for diverse test data, leading to performance degradation. We show that both shared and exclusive task-specific knowledge are crucial for merging performance, but directly merging exclusive knowledge hinders overall performance. In view of this, we propose Twin-Merging, a method that encompasses two principal stages: (1) modularizing knowledge into shared and exclusive components, with compression to reduce redundancy and enhance efficiency; (2) dynamically merging shared and task-specific knowledge based on the input. This approach narrows the performance gap between merged and fine-tuned models and improves adaptability to heterogeneous data. Extensive experiments on $20$ datasets for both language and vision tasks demonstrate the effectiveness of our method, showing an average improvement of $28.34\%$ in absolute normalized score for discriminative tasks and even surpassing the fine-tuned upper bound on the generative tasks. Our implementation is available in \url{https://github.com/LZY-the-boys/Twin-Merging}

Zhenyi Lu, Chenghao Fan, Wei Wei, Xiaoye Qu, Dangyang Chen, Yu Cheng• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationDTD
Accuracy82.29
599
ClassificationCars
Accuracy86.81
492
Image ClassificationRESISC45
Accuracy95.13
472
Image ClassificationSVHN (test)
Accuracy95.3
470
Image ClassificationSUN397
Accuracy68.8
450
Image ClassificationMNIST
Accuracy98.11
398
Image ClassificationSVHN
Accuracy89.16
395
Multi-task Language UnderstandingMMLU
Accuracy68.14
353
Image ClassificationDTD (test)
Accuracy74
316
Instruction FollowingMT-Bench
MT-Bench Score8.2
287
Showing 10 of 62 rows

Other info

Follow for update