OLMoE: Open Mixture-of-Experts Language Models
Muennighoff, Niklas, Soldaini, Luca, Groeneveld, Dirk, Lo, Kyle, Morrison, Jacob, Min, Sewon, Shi, Weijia, Walsh, Pete, Tafjord, Oyvind, Lambert, Nathan, Gu, Yuling, Arora, Shane, Bhagia, Akshita, Schwenk, Dustin, Wadden, David, Wettig, Alexander, Hui, Binyuan, Dettmers, Tim, Kiela, Douwe, Farhadi, Ali, Smith, Noah A., Koh, Pang Wei, Singh, Amanpreet, Hajishirzi, Hannaneh
–arXiv.org Artificial Intelligence
We introduce OLMoE, a fully open, state-of-the-art language model leveraging sparse Mixture-of-Experts (MoE). OLMoE-1B-7B has 7 billion (B) parameters but uses only 1B per input token. We pretrain it on 5 trillion tokens and further adapt it to create OLMoE-1B-7B-Instruct. Our models outperform all available models with similar active parameters, even surpassing larger ones like Llama2-13B-Chat and DeepSeekMoE-16B. We present various experiments on MoE training, analyze routing in our model showing high specialization, and open-source all aspects of our work: model weights, training data, code, and logs.
arXiv.org Artificial Intelligence
Sep-3-2024
- Country:
- North America > United States
- Arizona (0.04)
- Europe
- Ireland (0.04)
- Italy > Calabria
- Catanzaro Province > Catanzaro (0.04)
- Asia
- Singapore (0.04)
- Middle East
- Jordan (0.04)
- Republic of Türkiye (0.04)
- Lebanon (0.04)
- Iraq (0.04)
- Iran (0.04)
- Africa > Middle East
- Egypt (0.04)
- North America > United States
- Genre:
- Research Report (1.00)
- Industry:
- Health & Medicine (0.66)
- Technology: