PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
September 30, 20250 citationsOpen Access

FLAME-MoE: A Transparent End-to-End Research Platform for Mixture-of-Experts Language Models

View Full Paper
HKHao KangZYZichun YuCXChenyan Xiong

Key Points

  • FLAME-MoE shows an average accuracy improvement of up to 3.4 points over dense baselines that utilize identical FLOPs.
  • The platform leverages full training trace transparency to investigate expert specialization and co-activation sparsity.
  • Analysis reveals that experts become increasingly specialized on distinct token subsets during training.
  • Training materials and logs are publicly available, promoting reproducibility in the usage of the FLAME-MoE platform.

Abstract

Recent large language models such as Gemini-1.5, DeepSeek-V3, and Llama-4 increasingly adopt Mixture-of-Experts (MoE) architectures, which offer strong efficiency-performance trade-offs by activating only a fraction of the model per token. Yet academic researchers still lack a fully open, end-to-end MoE platform for investigating scaling, routing, and expert behavior. We release FLAME-MoE, a completely open-source research suite composed of seven decoder-only models, ranging from 38M to 1.7B active parameters, whose architecture--64 experts with top-8 gating and 2 shared experts--closely reflects modern production LLMs. All training data pipelines, scripts, logs, and checkpoints are publicly available to enable reproducible experimentation. Across six evaluation tasks, FLAME-MoE improves average accuracy by up to 3.4 points over dense baselines trained with identical FLOPs. Leveraging full training trace transparency, we present initial analyses showing that (i) experts increasingly specialize on distinct token subsets, (ii) co-activation matrices remain sparse, reflecting diverse expert usage, and (iii) routing behavior stabilizes early in training. All code, training logs, and model checkpoints are available at https://github.com/cmu-flame/FLAME-MoE.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Kang et al. (2025) studied this question.

synapsesocial.com/papers/68dc12cc8a7d58c25ebb0b02https://doi.org/10.48550/arxiv.2505.20225
Ask AI
Helpful
Bookmark
Share
View Full Paper