arXiv:2604.13508cs.CV2026-04中稿 · CVPR

通过语义聚类优化专家模型初始化,提升早期专业化与多样性。

Enhancing Mixture-of-Experts Specialization via Cluster-Aware Upcycling

论文配图:Enhancing Mixture-of-Experts Specialization via Cluster-Aware Upcycling
图 1 · 摘自论文原文
  • 基于输入激活聚类,用截断SVD初始化专家子空间。
  • 在CLIP ViT-B/32和B/16上优于现有方法,零样本与少样本均提升。
  • 引入专家集成自蒸馏损失,增强路由稳定性与专家差异性。

稀疏复用(Sparse Upcycling)为从预训练稠密权重初始化混合专家(MoE)模型提供了高效途径,但因所有专家初始权重相同且路由器随机初始化,存在专家对称性问题,早期专业化能力有限。本文提出聚类感知复用(Cluster-aware Upcycling),将语义结构融入初始化过程:首先将稠密模型的输入激活划分为语义聚类,再通过截断SVD使用对应聚类的子空间表示初始化各专家,并将路由器初始权重设为聚类中心。该策略打破专家对称性,促进与数据分布一致的早期专业化。此外,引入专家集成自蒸馏损失,在训练中提供可靠的路由指导。在CLIP ViT-B/32和ViT-B/16上的实验表明,该方法在零样本与少样本基准上持续优于现有方法,生成更具多样性与解耦性的专家表征,降低专家间相似度,提升路由置信度。

原文摘要 · Abstract (English)

Sparse Upcycling provides an efficient way to initialize a Mixture-of-Experts (MoE) model from pretrained dense weights instead of training from scratch. However, since all experts start from identical weights and the router is randomly initialized, the model suffers from expert symmetry and limited early specialization. We propose Cluster-aware Upcycling, a strategy that incorporates semantic structure into MoE initialization. Our method first partitions the dense model's input activations into semantic clusters. Each expert is then initialized using the subspace representations of its corresponding cluster via truncated SVD, while setting the router's initial weights to the cluster centroids. This cluster-aware initialization breaks expert symmetry and encourages early specialization aligned with the data distribution. Furthermore, we introduce an expert-ensemble self-distillation loss that stabilizes training by providing reliable routing guidance using an ensemble teacher. When evaluated on CLIP ViT-B/32 and ViT-B/16, Cluster-aware Upcycling consistently outperforms existing methods across both zero-shot and few-shot benchmarks. The proposed method also produces more diverse and disentangled expert representations, reduces inter-expert similarity, and leads to more confident routing behavior. Project page: https://sanghyeokchu.github.io/cluster-aware-upcycling/

混合专家模型初始化自蒸馏聚类

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