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 ClassificationSUN397
Accuracy68.8
441
Image ClassificationMNIST
Accuracy98.11
398
ClassificationCars
Accuracy68.77
395
Image ClassificationSVHN
Accuracy89.16
395
Image ClassificationRESISC45
Accuracy85.27
349
Multi-task Language UnderstandingMMLU
Accuracy68.14
321
Image ClassificationEuroSAT
Accuracy96.65
207
Bias EvaluationBBQ
Accuracy86.46
113
Image Classification20 Vision Classification Tasks
Average Accuracy75.3
94
TruthfulnessTruthfulQA
Truthfulness Accuracy52.78
86
Showing 10 of 21 rows

Other info

Follow for update