LIBMoE: A Library for comprehensive benchmarking Mixture of Experts in Large Language Models

Nam V. Nguyen, Thong T. Doan, Luong Tran, Van Nguyen, Quang Pham
11/1/2024
cs.CLcs.AIcs.LG

No official code found. Best implementation: Fsoft-AIC/LibMoE (50 stars, Apache-2.0, last commit 2026-04-23).

Abstract

Mixture of experts (MoE) architectures have become a cornerstone for scaling up and are a key component in most large language models such as GPT-OSS, DeepSeek-V3, Llama-4, and Gemini-2.5. However, systematic research on MoE remains severely constrained by the prohibitive computational costs of training and evaluation, restricting large-scale studies accessible to most researchers. We introduce LibMoE, a unified framework for reproducible, efficient, and extensible MoE research that supports both pretraining and sparse-upcycling regimes. Beyond unified implementations, the framework provides transparent analytical tools for probing routing and expert dynamics. Leveraging this foundation, we conduct a comprehensive analysis along three dimensions: (i) routing dynamics, covering expert selection patterns, routing stability and optimality, and how routing entropy reveals task specialization and expert diversity; (ii) the effect of lightweight initialization on load balancing, demonstrating how subtle changes in router initialization shape early expert utilization; and (iii) training regime differences, revealing how sparse upcycling and full pretraining exhibit distinct routing patterns and stability profiles. By lowering the barrier to entry and standardizing evaluation, along with our comprehensive analysis, LibMoE broadens access to MoE research and establishes a reliable benchmark to guide future innovations. Project page: https://fsoft-aic.github.io/fsoft-LibMoE.github.io.

View on arXivView PDF

Code Implementations(1)

Fsoft-AIC/LibMoECommunity100%
Apache-2.0

LibMoE: A LIBRARY FOR COMPREHENSIVE BENCHMARKING MIXTURE OF EXPERTS IN LARGE LANGUAGE MODELS

500Shell, PythonOct 21, 20246 months agoApache-2.0
frameworkllm-trainingmoe

Similar Papers

Cite this paper

@article{nguyen2024libmoe,
  title  = {LIBMoE: A Library for comprehensive benchmarking Mixture of Experts in Large Language Models},
  author = {Nam V. Nguyen and Thong T. Doan and Luong Tran and Van Nguyen and Quang Pham},
  year   = {2024},
  eprint = {2411.00918},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.00918v2}
}

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/2411.00918.svg)](https://codeofpaper.com/paper/2411.00918)

Discussion