arXiv:2411.18322cs.CVcs.LG2024-11被引 12

探索图像分类中专家混合模型的最佳参数激活量

Mixture of Experts in Image Classification: What's the Sweet Spot?

  • 通过系统实验发现,每样本适度激活参数能实现性能与效率的平衡
  • 小到中等规模模型使用专家混合提升显著,大规模模型收益递减
  • 简单线性路由优于复杂路由,数据量大时可支持更多专家

专家混合(MoE)模型在跨领域参数高效扩展方面展现出潜力,但其在图像分类中的应用仍受限,通常需百亿级数据集才能具备竞争力。本文在开放数据集上系统研究了将MoE层融入图像分类架构的效果,分析不同配置与模型规模。结果表明,每样本适度激活参数提供最佳性能与效率权衡;随着激活参数增多,MoE优势减弱。实践启示包括:MoE最有效增强小型与中型模型,大型网络增益趋缓,无法重定义ImageNet顶尖性能;最后两层放置策略在多数架构中表现鲁棒,每两层放置在ViT中略优;更大数据集(如ImageNet-21k)允许最多16个专家被有效利用,无需改变位置,因数据增多减少过拟合并促进专家专精;简单线性路由器表现最优,说明额外路由复杂性无一致收益。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) models have shown promising potential for parameter-efficient scaling across domains. However, their application to image classification remains limited, often requiring billion-scale datasets to be competitive. In this work, we explore the integration of MoE layers into image classification architectures using open datasets. We conduct a systematic analysis across different MoE configurations and model scales. We find that moderate parameter activation per sample provides the best trade-off between performance and efficiency. However, as the number of activated parameters increases, the benefits of MoE diminish. Our analysis yields several practical insights for vision MoE design. First, MoE layers most effectively strengthen tiny and mid-sized models, while gains taper off for large-capacity networks and do not redefine state-of-the-art ImageNet performance. Second, a Last-2 placement heuristic offers the most robust cross-architecture choice, with Every-2 slightly better for Vision Transform (ViT), and both remaining effective as data and model scale increase. Third, larger datasets (e.g., ImageNet-21k) allow more experts, up to 16, for ConvNeXt to be utilized effectively without changing placement, as increased data reduces overfitting and promotes broader expert specialization. Finally, a simple linear router performs best, suggesting that additional routing complexity yields no consistent benefit.

专家混合图像分类模型效率视觉Transformer

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