English

When Does Sparse MoE Help in Vision? The Role of Backbone Compute Leverage in Sparse Routing

Computer Vision and Pattern Recognition 2026-05-18 v1 Machine Learning

Abstract

Mixture-of-Experts (MoE) networks promise favorable accuracy-compute trade-offs, yet practical vision deployments are hindered by expert collapse and limited end-to-end efficiency gains. We study when sparse top-kk routing with hard capacity constraints helps in vision classification, evaluated under multi-seed protocols on four benchmarks (CIFAR-10/100, Tiny-ImageNet, ImageNet-1K). We observe a \emph{compute-leverage pattern}: positive accuracy gaps require a substantial fraction ρ\rho of total FLOPs to be routed; at ImageNet scale this is necessary but not sufficient, as multi-expert routing (k2k \geq 2) is additionally required. Two controlled experiments isolate these factors. A hidden-size sweep on CIFAR-10 yields both predicted sign reversals across standard and depthwise backbones, ruling out backbone family as the active variable. An ImageNet-1K ablation that varies only top-kk -- holding architecture, initialization, and ρ\rho fixed -- reverses the gap from positive to negative across all five seeds. A per-sample variant of Soft MoE that softmaxes over experts rather than the batch rescues CIFAR-100 above the dense baseline, identifying batch-axis dispatch as the dominant failure mode in per-sample CNN settings. Code and aggregate results: https://github.com/libophd/sparse-moe-vision-rho.

Keywords

Cite

@article{arxiv.2605.15484,
  title  = {When Does Sparse MoE Help in Vision? The Role of Backbone Compute Leverage in Sparse Routing},
  author = {Libo Sun and Po-wei Harn and Peixiong He and Xiao Qin},
  journal= {arXiv preprint arXiv:2605.15484},
  year   = {2026}
}

Comments

24 pages (main + appendix), 8 figures, 18 tables. Under review at TMLR. Code and aggregate results: https://github.com/libophd/sparse-moe-vision-rho

R2 v1 2026-07-22T07:13:30.171Z