mlsanityData to 7 Oct 2026

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

arXiv 2411.18322 ↗First posted 27 Nov 2024cs.CV cs.LGcited by 11

On the map: Mixture-of-experts → Vision mixture-of-experts

Abstract (arXiv)

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.

Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Placed in Vision mixture-of-experts, one of 12 directions in Mixture-of-experts. Builds on 36 of its 41 references that are in the map. First posted on arXiv 27 Nov 2024.

Builds on

36 of 41 references.

Closest ancestors · 4

Same direction, newest first

  1. 2023From Sparse to Soft Mixtures of Experts2 Aug
  2. 2022Towards Understanding Mixture of Experts in Deep Learning4 Aug
  3. 2021Cross-token Modeling with Conditional Computation5 Sep
  4. 2021Scaling Vision with Sparse Mixture of Experts10 Jun

Other references · 32

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  4. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
  5. 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
  6. 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
  7. 2022A ConvNet for the 2020scited by 3,545
  8. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  9. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  10. 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,875
  11. 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,261
  12. 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,842
20 more
  1. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
  2. 2019Do ImageNet Classifiers Generalize to ImageNet?cited by 1,507
  3. 2021MLP-Mixer: An all-MLP Architecture for Visioncited by 1,441
  4. 2019Natural Adversarial Examplescited by 1,401
  5. 2022No Language Left Behind: Scaling Human-Centered Machine Translationcited by 1,290
  6. 2022CoCa: Contrastive Captioners are Image-Text Foundation Modelscited by 1,265
  7. 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,242
  8. 2017Random Erasing Data Augmentationcited by 1,235
  9. 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,010
  10. 2021Scaling Vision Transformerscited by 993
  11. 2022MaxViT: Multi-Axis Vision Transformercited by 476
  12. 2022DeiT III: Revenge of the ViTcited by 417
  13. 2022Sparse Upcycling: Training Mixture-of-Experts from Dense Checkpointscited by 207
  14. 2022Multimodal Contrastive Learning with LIMoE: the Language-Image Mixture of Expertscited by 202
  15. 2022Tutel: Adaptive Mixture-of-Experts at Scalecited by 199
  16. 2022On the Representation Collapse of Sparse Mixture of Expertscited by 156
  17. 2023Getting ViT in Shape: Scaling Laws for Compute-Optimal Model Designcited by 87
  18. 2024MoMa: Efficient Early-Fusion Pre-training with Mixture of Modality-Aware Expertscited by 61
  19. 2022Is a Modular Architecture Enough?cited by 35
  20. 2021Mixture-of-Variational-Experts for Continual Learningcited by 7

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 4; 2026: 7.

11 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Efficient Data Driven Mixture-of-Expert Extraction from Trained Networks
  2. 2024Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
  3. 2026Mixture of Experts with Soft Nearest Neighbor Loss: Resolving Expert Collapse via Representation Disentanglement
  4. 2026Mixture-of-Experts Models in Vision: Routing, Optimization, and Generalization
  5. 2024ViMoE: An Empirical Study of Designing Vision Mixture-of-Experts
  6. 2026Domain-Specialized Object Detection via Model-Level Mixtures of Experts
  7. 2024How Lightweight Can A Vision Transformer Be
  8. 2026MoE-ViE: Mixture of Experts Vision Encoder for Efficient Image and Video Understanding