DECO让稀疏专家模型在端侧设备上达到稠密模型性能
DECO: Sparse Mixture-of-Experts with Dense-Comparable Performance on End-Side Devices

- 用可学习的缩放路由机制动态平衡专家贡献,实现高效稀疏计算
- 仅激活20%专家即达稠密模型效果,端侧推理速度提升2.93倍
- 适合资源受限场景下的高性能模型部署,如移动设备与边缘计算
虽然混合专家(MoE)能扩展模型容量而不成比例增加计算量,但其庞大的参数总量带来了显著的存储与内存访问瓶颈,阻碍了同时满足高性能、低计算成本和小存储开销的端侧部署。为此,我们提出DECO,一种稀疏MoE架构,在相同总参数预算和训练样本量下,性能可媲美稠密Transformer。DECO采用可微且灵活的基于ReLU的路由机制,并引入可学习的专家级缩放,自适应平衡路由专家与共享专家的贡献。此外,我们提出NormSiLU激活函数,在SiLU前对输入进行归一化,使路由专家激活率更稳定,内在稀疏性更高。实验表明,当仅激活20%的路由专家时,DECO性能与稠密模型相当,且优于现有主流MoE基线。专为端侧优化的加速内核在Jetson AGX Orin上实现2.93倍的推理加速。代码与模型权重已开源。
原文摘要 · Abstract (English)
While Mixture-of-Experts (MoE) scales model capacity without proportionally increasing computation, its massive total parameter footprint creates significant storage and memory-access bottlenecks, which hinder efficient end-side deployment that simultaneously requires high performance, low computational cost, and small storage overhead. To achieve these properties, we present DECO, a sparse MoE architecture designed to match the performance of dense Transformers under identical total parameter budgets and training tokens. DECO utilizes the differentiable and flexible ReLU-based routing enhanced by learnable expert-wise scaling, which adaptively balances the contributions of routed and shared experts. Furthermore, we introduce NormSiLU, an activation function that normalizes inputs prior to SiLU operators, producing a more stable trend of routed-expert activation ratio and a higher intrinsic sparsity level. We also identify an empirical advantage in using non-gated MLP experts with ReLU-based routing, indicating the possibility of MoE architecture simplification. Experiments demonstrate that DECO, activating only 20% of routed experts, matches dense performance and outperforms established MoE baselines. Our specialized acceleration kernel delivers a 2.93$\times$ speedup on Jetson AGX Orin compared with dense inference. Code and checkpoints are available at https://github.com/thunlp/DECO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。