Training-Free Dynamic Upcycling of Expert Language Models

TL;DR

DUME is a training-free, scalable method that constructs a multi-domain expert language model by dynamically reusing dense experts through a closed-form ridge regression solution, avoiding additional training.

Contribution

It introduces DUME, a novel approach to build multi-domain language models without extra training, outperforming baselines and enabling dynamic expert addition.

Findings

DUME retains up to 97.6% of a dense expert model's performance in a specific domain.

DUME surpasses dense experts in reasoning tasks, achieving 102.1% performance.

The method is cost-efficient, scalable, and can be fine-tuned for further improvements.

Abstract

Large Language Models (LLMs) have achieved remarkable performance on a wide range of specialized tasks, exhibiting strong problem-solving capabilities. However, training these models is prohibitively expensive, and they often lack domain-specific expertise because they rely on general knowledge datasets. Expertise finetuning can address this issue; however, it often leads to overspecialization, and developing a single multi-domain expert remains difficult due to diverging objectives. Furthermore, multitask training is challenging due to interference and catastrophic forgetting. Existing work proposes combining the expertise of dense models within a Mixture of Experts (MoE) architecture, although this approach still requires multitask finetuning. To address these issues, we introduce Dynamic Upcycling MoE (DUME), a novel approach that reuses dense experts trained on different domains to construct a unified MoE model. Our method builds a single multitask model that preserves the capabilities of the original dense experts without requiring additional training. DUME is both cost-efficient and scalable: by leveraging the closed-form solution of ridge regression, it eliminates the need for further optimization and enables experts to be added dynamically while maintaining the model's original performance. We demonstrate that DUME consistently outperforms baseline approaches in both causal language modeling and reasoning settings. Finally, we also show that the DUME model can be fine-tuned to further improve performance. We show that, in the causal language modeling setting, DUME can retain up to 97.6% of a dense expert model specialized in one particular domain, and that it can also surpass it in the reasoning setting, where it can achieve 102.1% of the dense expert performance. Our code is available at: github.com/gensyn-ai/dume.