arXiv:2504.09265cs.LGcs.CL2025-04被引 1

通过分组稀疏正则化提升专家多样性,让混合专家模型更高效

Mixture of Group Experts for Learning Invariant Representations

  • 用分组稀疏约束路由输入,间接增强专家分化
  • 在图像分类和语言建模任务中性能超越原始MoE,且开销极小
  • 适合需要大规模专家且追求低冗余的模型设计者

稀疏激活的混合专家(MoE)模型可在保持每标记计算成本稳定的同时显著增加参数量。然而,原始MoE模型通常存在专家间多样性与专业化不足的问题,尤其在专家数量增加时更为明显。本文受稀疏表示理论启发,提出一种基于top-$k$路由的新视角。在此基础上,我们引入群体稀疏正则化方法,称为分组混合专家(MoGE)。MoGE通过对路由输入施加结构约束来间接正则化专家,同时保持原有MoE架构不变。此外,我们将路由输入组织为二维拓扑图,对邻近元素进行空间分组,使模型能够捕捉对微小变换不变的表示,显著提升专家多样性与专业化。在多种Transformer模型上的综合评估表明,MoGE在图像分类与语言建模任务中显著优于基准MoE,且额外内存与计算开销极低。该方法为扩展专家数量、降低冗余提供了简单有效的解决方案。源代码已附于补充材料,并将公开发布。

原文摘要 · Abstract (English)

Sparsely activated Mixture-of-Experts (MoE) models effectively increase the number of parameters while maintaining consistent computational costs per token. However, vanilla MoE models often suffer from limited diversity and specialization among experts, constraining their performance and scalability, especially as the number of experts increases. In this paper, we present a novel perspective on vanilla MoE with top-$k$ routing inspired by sparse representation. This allows us to bridge established theoretical insights from sparse representation into MoE models. Building on this foundation, we propose a group sparse regularization approach for the input of top-$k$ routing, termed Mixture of Group Experts (MoGE). MoGE indirectly regularizes experts by imposing structural constraints on the routing inputs, while preserving the original MoE architecture. Furthermore, we organize the routing input into a 2D topographic map, spatially grouping neighboring elements. This structure enables MoGE to capture representations invariant to minor transformations, thereby significantly enhancing expert diversity and specialization. Comprehensive evaluations across various Transformer models for image classification and language modeling tasks demonstrate that MoGE substantially outperforms its MoE counterpart, with minimal additional memory and computation overhead. Our approach provides a simple yet effective solution to scale the number of experts and reduce redundancy among them. The source code is included in the supplementary material and will be publicly released.

混合专家稀疏表示模型压缩不变表示

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