让专家模型的路由更精准,提升训练速度和抗干扰能力。
Tight Clusters Make Specialized Experts
- 基于聚类优化设计自适应特征加权,提升潜在簇识别能力。
- 在语言与图像任务中,收敛更快且对数据污染更鲁棒。
- 适合追求高效专家模型的开发者,尤其在噪声环境下表现优异。
稀疏混合专家(MoE)架构通过解耦模型容量与计算成本成为有前景的方法。其核心是路由器,负责学习输入分布的潜在聚类结构,将输入标记分配给合适的专家。然而,在高维空间中潜在聚类可能难以识别,导致收敛缓慢、易受数据污染影响,并整体降低表示质量。本文从聚类优化视角分析路由器,推导出能最大化识别潜在聚类的最优特征权重。利用这些权重,在自适应变换空间中计算标记-专家路由分配,促进簇间分离,从而更准确匹配每个标记的最佳专家。具体而言,针对每个专家簇,计算一组特征缩放权重,依据该专家是否沿特定特征紧密聚集。提出的新型路由器称为自适应聚类(AC)路由器。实验表明,该方法使MoE模型实现三个连贯优势:1)更快收敛;2)更强数据扰动鲁棒性;3)整体性能提升,因专家专注于输入空间中语义区分的区域。我们在多种语言建模与图像识别任务的MoE骨干网络上,验证了其在干净与污染设置下均优于基线路由方法。
原文摘要 · Abstract (English)
Sparse Mixture-of-Experts (MoE) architectures have emerged as a promising approach to decoupling model capacity from computational cost. At the core of the MoE model is the router, which learns the underlying clustering structure of the input distribution in order to send input tokens to appropriate experts. However, latent clusters may be unidentifiable in high dimension, which causes slow convergence, susceptibility to data contamination, and overall degraded representations as the router is unable to perform appropriate token-expert matching. We examine the router through the lens of clustering optimization and derive optimal feature weights that maximally identify the latent clusters. We use these weights to compute the token-expert routing assignments in an adaptively transformed space that promotes well-separated clusters, which helps identify the best-matched expert for each token. In particular, for each expert cluster, we compute a set of weights that scales features according to whether that expert clusters tightly along that feature. We term this novel router the Adaptive Clustering (AC) router. Our AC router enables the MoE model to obtain three connected benefits: 1) faster convergence, 2) better robustness to data corruption, and 3) overall performance improvement, as experts are specialized in semantically distinct regions of the input space. We empirically demonstrate the advantages of our AC router over baseline routing methods when applied on a variety of MoE backbones for language modeling and image recognition tasks in both clean and corrupted settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。