mlsanityData to 7 Oct 2026

MoRE: Scaling mixture of experts with hardware-aware low-rank routing

arXiv 2609.36301 ↗First posted 28 Sep 2026cs.LG cs.AI cs.CLcited by 0

On the map: Mixture-of-experts → Expert routing strategies

Abstract (arXiv)

Mixture-of-Experts (MoE) layers are central to frontier language models, and recent architectures push toward more and smaller experts. In this regime, the standard linear router becomes a bottleneck: with M experts and hidden dimension h, its per-token cost Theta(Mh) dominates the MoE layer once M is large. We introduce MoRE (Mixture of Rank-reduced-routed Experts), which factorizes the router weight matrix at rank r and reduces the routing cost to O((h + M)r). We prove that rank logarithmic in M suffices for routing expressivity when the number of active experts is fixed, and is necessary up to precision factors. We also prove that logarithmic rank preserves load balance in a Gaussian memorization model, and training on a synthetic phonebook task shows that low rank does not hurt memorization. At matched active FLOPs, the factorization allows a factor of Theta(h/r) more experts. To realize this gain in wall-clock time, we design a fused Triton kernel at inference that avoids expensive memory operations on HBM. Empirically, MoRE improves memorization on the phonebook task and performance on knowledge-intensive Q&A benchmarks after pretraining, while matching reasoning ability. Code available at https://github.com/Matheart/MoREcode.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 45 of its 71 references that are in the map. First posted on arXiv 28 Sep 2026.

Builds on

45 of 71 references.

Closest ancestors · 6

Same direction, newest first

  1. 2026Slicing and Dicing: Configuring Optimal Mixtures of Experts12 May
  2. 2025Secret mixtures of experts inside your LLM20 Dec
  3. 2024Ultra-Sparse Memory Network19 Nov
  4. 2024Mixture of A Million Experts4 Jul
  5. 2022On the Representation Collapse of Sparse Mixture of Experts20 Apr
  6. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan

Other references · 39

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2025Qwen3 Technical Reportcited by 8,195
  3. 2020Scaling Laws for Neural Language Modelscited by 6,411
  4. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  5. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  6. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  7. 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
  8. 2024Mixtral of Expertscited by 1,785
  9. 2020GLU Variants Improve Transformercited by 1,724
  10. 2017Decoupled Weight Decay Regularizationcited by 1,722
  11. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
  12. 2025gpt-oss-120b & gpt-oss-20b Model Cardcited by 1,258
27 more
  1. 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
  2. 2024DeepSeek-V3 Technical Reportcited by 724
  3. 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
  4. 2022Efficiently Scaling Transformer Inferencecited by 529
  5. 2025Kimi K2: Open Agentic Intelligencecited by 523
  6. 2024LoRA+: Efficient Low Rank Adaptation of Large Modelscited by 365
  7. 2023Physics of Language Models: Part 3.1, Knowledge Storage and Extractioncited by 264
  8. 2022Unified Scaling Laws for Routed Language Modelscited by 249
  9. 2021Hash Layers For Large Sparse Modelscited by 236
  10. 2024Scaling Laws for Fine-Grained Mixture of Expertscited by 166
  11. 2018Online normalizer calculation for softmaxcited by 154
  12. 2024Physics of Language Models: Part 3.3, Knowledge Capacity Scaling Lawscited by 137
  13. 2024Liger Kernel: Efficient Triton Kernels for LLM Trainingcited by 123
  14. 2023Pre-gated MoE: An Algorithm-System Co-Design for Fast and Scalable Mixture-of-Expert Inferencecited by 101
  15. 2026Conditional Memory via Scalable Lookup: A New Axis of Sparsity for Large Language Modelscited by 78
  16. 2026FlashAttention-4: Algorithm and Kernel Pipelining Co-Design for Asymmetric Hardware Scalingcited by 58
  17. 2024MoE-Infinity: Efficient MoE Inference on Personal Machines with Sparsity-Aware Expert Cachecited by 48
  18. 2025Parameters vs FLOPs: Scaling Laws for Optimal Sparsity for Mixture-of-Experts Language Modelscited by 48
  19. 2024Memory Layers at Scalecited by 39
  20. 2024Mixture of Parrots: Experts improve memorization more than reasoningcited by 20
  21. 2025SonicMoE: Accelerating MoE with IO and Tile-aware Optimizationscited by 20
  22. 2024ExpertFlow: Efficient Mixture-of-Experts Inference via Predictive Expert Caching and Token Schedulingcited by 18
  23. 2026Scalable Training of Mixture-of-Experts Models with Megatron Corecited by 16
  24. 2024Scattered Mixture-of-Experts Implementationcited by 15
  25. 2026Hyperparameter Transfer with Mixture-of-Expert Layerscited by 10
  26. 2020Memorizing Gaussians with no over-parameterizaion via gradient decent on neural networkscited by 7
  27. 2026L2R: Low-Rank and Lipschitz-Controlled Routing for Mixture-of-Expertscited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Ban&Pick: Enhancing Performance and Efficiency of MoE-LLMs via Smarter Routing
  2. 2026Hierarchical Mixture-of-Experts with Two-Stage Optimization
  3. 2026MoE Routing Testbed: Studying Expert Specialization and Routing Behavior at Small Scale
  4. 2026MoRA: MoE Pruning via Router Bias Learning and Expert Approximation
  5. 2026Path-Constrained Mixture-of-Experts
  6. 2025Load Balancing Mixture of Experts with Similarity Preserving Routers
  7. 2026Equifinality in Mixture of Experts: Routing Topology Does Not Determine Language Modeling Quality
  8. 2025Maximum Score Routing For Mixture-of-Experts