MobileMoE让手机端大模型更高效,用少一半参数实现更好性能。
MobileMoE: Scaling On-Device Mixture of Experts

- 在手机内存和算力限制下,设计了适合移动端的稀疏专家架构。
- 0.3-0.9B活跃参数,推理耗时比稠密模型快1.8-3.8倍。
- 适合追求低功耗、高效率的移动端大模型开发者与研究者。
混合专家(MoE)已成为百亿参数语言模型的主流架构,但在亿级以下规模用于设备端部署的优势尚未充分探索。为此,我们提出MobileMoE,一组具有亚十亿活跃参数(0.3-0.9B活跃,1.3-5.3B总参数)的设备端MoE语言模型,建立了设备端大模型的新帕累托前沿。我们首先制定了一套设备端MoE缩放定律,在移动内存与算力约束下联合优化架构,识别出中等稀疏度、细粒度且共享专家的最优配置,同时兼顾内存与计算效率。基于此架构,我们采用四阶段训练流程(预训练、中期训练、指令微调与量化感知训练),全部使用开源数据集完成。在14项基准测试中,MobileMoE以2-4×更少的推理浮点运算量达到或超越领先的稠密设备端模型,并在最多60%更少参数下媲美甚至超越当前最先进的MoE模型OLMoE-1B-7B。为打通部署最后一公里,我们首次实现了通用智能手机上的高效MoE推理并完成全面的设备端性能分析。在相近的INT4权重内存占用下,MobileMoE-S的预填充速度比稠密基线MobileLLM-Pro快1.8-3.8倍,解码速度快2.2-3.4倍。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) has become the de facto architecture for hundred-billion-parameter language models, yet its advantages at sub-billion scales for on-device deployment remain largely unexplored. To close this gap, we present MobileMoE, a family of on-device MoE language models with sub-billion active parameters (0.3-0.9B active and 1.3-5.3B total) that establish a new Pareto frontier for on-device LLMs. We first formulate an on-device MoE scaling law that jointly optimizes MoE architecture under mobile memory and compute constraints, identifying an on-device sweet spot - moderate sparsity with fine-grained and shared experts - that is simultaneously memory and compute-optimal. Building on the derived architectures, we train MobileMoE with a four-stage recipe covering pre-training, mid-training, instruction fine-tuning, and quantization-aware training, all on open-source datasets. Across 14 benchmarks, MobileMoE matches or exceeds leading on-device dense LLMs with 2-4$\times$ fewer inference FLOPs, and matches or surpasses the state-of-the-art MoE OLMoE-1B-7B with up to 60% fewer parameters. To bridge the last mile to mobile deployment, we provide the first efficient MoE inference on commodity smartphones with comprehensive on-device profiling. At comparable INT4 weight memory, MobileMoE-S delivers $1.8$-$3.8\times$ faster prefill and $2.2$-$3.4\times$ faster decode than the dense baseline MobileLLM-Pro.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。