bibtext: @inproceedings{
  sukhbaatar2024branchtrainmix,
  title={Branch-Train-MiX: Mixing Expert {LLM}s into a Mixture-of-Experts {LLM}},
  author={Sainbayar Sukhbaatar and Olga Golovneva and Vasu Sharma and Hu Xu and Xi Victoria Lin and Baptiste Roziere and Jacob Kahn and Shang-Wen Li and Wen-tau Yih and Jason E Weston and Xian Li},
  booktitle={First Conference on Language Modeling},
  year={2024},
  url={https://openreview.net/forum?id=nqLAuMOF6n}
  }
abstract: We investigate efficient methods for training Large Language Models (LLMs) to possess capabilities in multiple specialized domains, such as coding, math reasoning and world knowledge. Our method, named Branch-Train-MiX (BTX), starts from a seed model, which is branched to train experts in embarrassingly parallel fashion with high throughput and reduced communication cost. After individual experts are asynchronously trained, BTX brings together their feedforward parameters as experts in  Mixture-of-Expert (MoE) layers and averages the remaining parameters, followed by an MoE-finetuning stage to learn token-level routing. BTX generalizes two special cases, the Branch-Train-Merge method, which does not have the MoE finetuning stage to learn routing, and sparse upcycling, which omits the stage of training experts asynchronously. Compared to alternative approaches, BTX achieves the best accuracy-efficiency tradeoff.
title: Branch-Train-MiX: Mixing Expert LLMs into a Mixture-of-Experts LLM
authors: Sainbayar Sukhbaatar Olga Golovneva Vasu Sharma Hu Xu Xi Victoria Lin Baptiste Roziere Jacob Kahn Shang-Wen Li Wen-tau Yih Jason E Weston Xian Li