Merging in a Bottle: Differentiable Adaptive Merging (DAM) and the Path from Averaging to Automation

Thomas Gauthier-Caron, Shamane Siriwardhana, Elliot Stein, Malikeh Ehghaghi, Charles Goddard +3 more
10/10/2024
cs.CLcs.AIcs.LG

Official code: arcee-ai/DAM (57 stars, last commit 2024-11-06).

Abstract

By merging models, AI systems can combine the distinct strengths of separate language models, achieving a balance between multiple capabilities without requiring substantial retraining. However, the integration process can be intricate due to differences in training methods and fine-tuning, typically necessitating specialized knowledge and repeated refinement. This paper explores model merging techniques across a spectrum of complexity, examining where automated methods like evolutionary strategies stand compared to hyperparameter-driven approaches such as DARE, TIES-Merging and simpler methods like Model Soups. In addition, we introduce Differentiable Adaptive Merging (DAM), an efficient, adaptive merging approach as an alternative to evolutionary merging that optimizes model integration through scaling coefficients, minimizing computational demands. Our findings reveal that even simple averaging methods, like Model Soups, perform competitively when model similarity is high, underscoring each technique's unique strengths and limitations. We open-sourced DAM, including the implementation code and experiment pipeline, on GitHub: https://github.com/arcee-ai/DAM.

View on arXivView PDF

Code Implementations(1)

arcee-ai/DAMOfficial100%
577C++, ShellAug 13, 20241 years ago

Similar Papers

Cite this paper

@article{gauthiercaron2024merging,
  title  = {Merging in a Bottle: Differentiable Adaptive Merging (DAM) and the Path from Averaging to Automation},
  author = {Thomas Gauthier-Caron and Shamane Siriwardhana and Elliot Stein and Malikeh Ehghaghi and Charles Goddard and Mark McQuade and Jacob Solawetz and Maxime Labonne},
  year   = {2024},
  eprint = {2410.08371},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.08371v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.08371.svg)](https://codeofpaper.com/paper/2410.08371)

Discussion