arXiv:2604.21330cs.CV2026-04

用预训练教师模型指导路由,让视觉MoE更稳定高效

Teacher-Guided Routing for Sparse Vision Mixture-of-Experts

论文配图:Teacher-Guided Routing for Sparse Vision Mixture-of-Experts
图 1 · 摘自论文原文
  • 用教师模型中间表示构建伪路由监督信号
  • 在ImageNet-1K上提升准确率并减少路由波动
  • 适合需要稳定稀疏专家模型的研究者

深度学习的进展依赖于日益庞大的模型,但计算成本已成为关键瓶颈。稀疏混合专家(MoE)通过每次仅激活少量专家,实现了高可扩展性且不牺牲推理速度。然而,稀疏MoE训练存在优化难题:路由器仅通过前向传播中选中的专家获得梯度信息,导致梯度阻塞,难以从未选路径获取有效反馈。这种局部化反馈使路由器难以学习合适的专家选择得分,常引发路由动态不稳定,如训练中专家分配频繁波动。为此,本文提出TGR-MoE:面向稀疏视觉混合专家的教师引导路由方法。该方法利用预训练稠密教师模型的中间表示构建教师路由器,并将其路由输出作为学生路由器的伪监督信号,抑制训练过程中的频繁路由波动,实现从训练初期起的知识引导专家选择。在ImageNet-1K和CIFAR-100上的大量实验表明,TGR-MoE持续提升准确率与路由一致性,即使在高度稀疏配置下仍保持稳定训练。

原文摘要 · Abstract (English)

Recent progress in deep learning has been driven by increasingly large-scale models, but the resulting computational cost has become a critical bottleneck. Sparse Mixture of Experts (MoE) offers an effective solution by activating only a small subset of experts for each input, achieving high scalability without sacrificing inference speed. Although effective, sparse MoE training exhibits characteristic optimization difficulties. Because the router receives informative gradients only through the experts selected in the forward pass, it suffers from gradient blocking and obtains little information from unselected routes. This limited, highly localized feedback makes it difficult for the router to learn appropriate expert-selection scores and often leads to unstable routing dynamics, such as fluctuating expert assignments during training. To address this issue, we propose TGR-MoE: Teacher-Guided Routing for Sparse Vision Mixture-of-Experts, a simple yet effective method that stabilizes router learning using supervision derived from a pretrained dense teacher model. TGR-MoE constructs a teacher router from the teacher's intermediate representations and uses its routing outputs as pseudo-supervision for the student router, suppressing frequent routing fluctuations during training and enabling knowledge-guided expert selection from the early stages of training. Extensive experiments on ImageNet-1K and CIFAR-100 demonstrate that TGR consistently improves both accuracy and routing consistency, while maintaining stable training even under highly sparse configurations.

视觉模型MoE路由优化教师引导

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