English

ButterflyMoE: Sub-Linear Ternary Experts via Structured Butterfly Orbits

Machine Learning 2026-03-06 v4 Artificial Intelligence

Abstract

Linear memory scaling stores NN independent expert weight matrices requiring O(Nd2)\mathcal{O}(N \cdot d^2) memory, which exceeds edge devices memory budget. Current compression methods like quantization, pruning and low-rank factorization reduce constant factors but leave the scaling bottleneck unresolved. We introduce ButterflyMoE, a method that treats experts not as independent weight matrices but as geometric reorientations of a unified shared quantized substrate. Diversity among experts arises from viewing different angles of shared capacity, not from redundant storage. By applying learned rotations to a shared ternary prototype, each expert yields O(d2+Ndlogd)\mathcal{O}(d^2 + N \cdot d \log d) memory,sub-linear in the number of experts. The key insight: training these rotations with quantization reduces activation outliers and stabilizes extreme low bit training, where static methods collapse. Across language modeling benchmarks, ButterflyMoE achieves 150×\times memory reduction at 256 experts with negligible accuracy loss. ButterflyMoE allows multiple experts to fit on edge-constrained devices showing that geometric parameterization breaks linear scaling.

Keywords

Cite

@article{arxiv.2601.13563,
  title  = {ButterflyMoE: Sub-Linear Ternary Experts via Structured Butterfly Orbits},
  author = {Aryan Karmore},
  journal= {arXiv preprint arXiv:2601.13563},
  year   = {2026}
}
R2 v1 2026-07-01T09:11:46.434Z