arXiv:2510.05781cs.CL2025-10被引 5

提出细粒度专家选择机制,显著提升MoE模型参数利用率与推理效率。

Mixture of Neuron Experts

  • 基于神经元级激活值筛选高贡献专家,仅保留活跃部分参数。
  • 仅激活50%参数即达到传统MoE性能,且相同激活量下表现更优。
  • 无需额外路由开销,适合部署在资源受限场景的高效MoE系统。

本文研究MoE层在推理时参数激活是否仍保持高度稀疏。对多个代表性MoE模型进行稀疏化实验发现,逐个修剪激活参数子集,最多去除60%时任务性能几乎不变,超过90%才出现显著下降。进一步将专家分解为神经元粒度并可视化激活值,发现多数神经元激活接近零。据此提出Mixture of Neuron Experts(MoNE):通过每个专家内简单Top-k选择实现神经元级专家选取,不引入额外路由参数或跨专家通信,延迟可忽略。大量实验表明,MoNE在仅激活50%的MoE层参数时即可匹配传统MoE性能,并在相同激活参数数量下持续优于传统方法,验证了其在提升参数利用效率和推理效率方面的实用性。

原文摘要 · Abstract (English)

In this work, we first explore whether the parameters activated by the MoE layer remain highly sparse at inference. We perform a sparsification study on several representative MoE models. For each expert, we rank parameters by the magnitude of their activations from the gate projection and progressively prune the activated subset. Pruning up to 60% of parameters within that subset causes only negligible task-performance degradation; substantial drops occur only after more than 90% are removed. We further decompose experts into neuron-granular MoE and visualize their activation values, finding that most neuron activations are near zero. This observation motivates us to select only high-activation neuron experts during pretraining. Based on this insight, we propose Mixture of Neuron Experts (MoNE). MoNE achieves neuron-granular expert selection by only applying a simple top-k selection within each expert, incurs negligible latency, and requires no additional routing parameters or inter-expert communication. Extensive experiments demonstrate that MoNE matches traditional MoE performance while activating only 50% of the MoE-layer parameters, and it consistently outperforms traditional MoE when compared at equal numbers of activated parameters. These results suggest that MoNE is a practical approach to improving parameter utilization and inference efficiency in MoE-like models.

MoE神经元级稀疏激活推理优化

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