arXiv:2512.16248cs.CLcs.AI2025-12

超稀疏专家模型,激活参数仅0.5B却达顶尖性能。

Sigma-MoE-Tiny Technical Report

  • 每层96专家,每令牌仅激活1个,实现极致稀疏。
  • 激活0.5B参数仍超越同规模模型,20B总参数。
  • 提出渐进稀疏训练策略,解决低层负载失衡问题。

混合专家(MoE)因其高效可扩展性成为基础模型的有力范式。本文提出Sigma-MoE-Tiny,一个开源语言模型,在现有模型中实现了最高稀疏度。该模型每层采用最多96个专家,每个令牌仅激活1个专家,总参数量为200亿,但仅激活0.5亿参数。极端稀疏带来的主要挑战是专家负载均衡。我们发现,传统负载均衡损失在低层失效。为此,提出渐进稀疏化训练策略,平衡专家利用与训练稳定性。模型在多样化高质量语料上预训练,并通过后训练进一步释放能力,整个训练过程稳定,未出现不可恢复的损失突增。全面评估显示,尽管仅激活0.5亿参数,其性能仍位居同类模型前列,甚至优于显著更大的模型。此外,本文深入探讨高度稀疏MoE中的负载均衡问题,为未来高稀疏架构发展提供洞见。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) has emerged as a promising paradigm for foundation models due to its efficient and powerful scalability. In this work, we present Sigma-MoE-Tiny, an MoE language model that achieves the highest sparsity compared to existing open-source models. Sigma-MoE-Tiny employs fine-grained expert segmentation with up to 96 experts per layer, while activating only one expert for each token, resulting in 20B total parameters with just 0.5B activated. The major challenge introduced by such extreme sparsity lies in expert load balancing. We find that the widely-used load balancing loss tends to become ineffective in the lower layers under this setting. To address this issue, we propose a progressive sparsification schedule aiming to balance expert utilization and training stability. Sigma-MoE-Tiny is pre-trained on a diverse and high-quality corpus, followed by post-training to further unlock its capabilities. The entire training process remains remarkably stable, with no occurrence of irrecoverable loss spikes. Comprehensive evaluations reveal that, despite activating only 0.5B parameters, Sigma-MoE-Tiny achieves top-tier performance among counterparts of comparable or significantly larger scale. In addition, we provide an in-depth discussion of load balancing in highly sparse MoE models, offering insights for advancing sparsity in future MoE architectures. Project page: https://qghuxmu.github.io/Sigma-MoE-Tiny Code: https://github.com/microsoft/ltp-megatron-lm

MoE稀疏模型语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。