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

Evolutionary Optimization of Model Merging Recipes

About

Large language models (LLMs) have become increasingly capable, but their development often requires substantial computational resources. While model merging has emerged as a cost-effective promising approach for creating new models by combining existing ones, it currently relies on human intuition and domain knowledge, limiting its potential. Here, we propose an evolutionary approach that overcomes this limitation by automatically discovering effective combinations of diverse open-source models, harnessing their collective intelligence without requiring extensive additional training data or compute. Our approach operates in both parameter space and data flow space, allowing for optimization beyond just the weights of the individual models. This approach even facilitates cross-domain merging, generating models like a Japanese LLM with Math reasoning capabilities. Surprisingly, our Japanese Math LLM achieved state-of-the-art performance on a variety of established Japanese LLM benchmarks, even surpassing models with significantly more parameters, despite not being explicitly trained for such tasks. Furthermore, a culturally-aware Japanese VLM generated through our approach demonstrates its effectiveness in describing Japanese culture-specific content, outperforming previous Japanese VLMs. This work not only contributes new state-of-the-art models back to the open-source community, but also introduces a new paradigm for automated model composition, paving the way for exploring alternative, efficient approaches to foundation model development.

Takuya Akiba, Makoto Shing, Yujin Tang, Qi Sun, David Ha• 2024

Related benchmarks

TaskDatasetResultRank
General KnowledgeMMLU
MMLU General Knowledge Accuracy53.4
373
MathematicsMATH
MATH Accuracy12.3
172
Language UnderstandingMMLU (test)--
167
Reading ComprehensionDROP
DROP Accuracy25.9
138
Language UnderstandingMMLU-Pro
Accuracy27.84
130
Mathematical ReasoningGSM8K (val)
Accuracy43
115
Multi-task Language UnderstandingMMLU (test)
Normalized Accuracy56.9
107
Multitask Language UnderstandingMMLU (val)
Accuracy61.5
94
Common Sense ReasoningHELLASWAG (test)
Accuracy59.2
86
Natural Language UnderstandingGLUE
Average Score (GLUE)79.4
76
Showing 10 of 42 rows

Other info

Follow for update