Dual-Bias Load Balancing: Mitigating Expert-Bias Staleness in Auxiliary-Loss-Free Mixture-of-Experts

Yuxin Tian, Hailin Zhao, Jia Liu, Zhiqiang Zhang


Abstract
Auxiliary-loss-free load balancing (ALF-LB) achieves promising performance in Mixture-of-Experts large language model (MoE-LLM) pre-training as it decouples load balancing from the main objective. However, we observe an unexplored phenomenon: routing imbalance is concentrated in shallow MoE layers. We attribute this phenomenon to the interaction between representation homogeneity, which makes shallow-layer routing decisions highly sensitive, and expert-bias staleness, which applies stale load corrections to the current router state. To break the cycle, we propose Dual-Bias Load Balancing (DLB), which augments the standard global expert bias with an intra-step micro-bias residual. DLB introduces the effective bias 𝐛eff=𝐛g+𝛥𝐛, where 𝐛g is updated once per gradient step and 𝛥𝐛 is updated across micro-batches within the same step. DLB does not add forward or backward passes and integrates naturally into existing large-scale MoE training pipelines. Experiments on large-scale MoE pre-training show that DLB mitigates expert-bias staleness, alleviates shallow-layer imbalance, and yields lower test perplexity and stronger average downstream performance.
Anthology ID:
2026.findings-emnlp.1786
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
38193–38206
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1786/
DOI:
Bibkey:
Cite (ACL):
Yuxin Tian, Hailin Zhao, Jia Liu, and Zhiqiang Zhang. 2026. Dual-Bias Load Balancing: Mitigating Expert-Bias Staleness in Auxiliary-Loss-Free Mixture-of-Experts. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 38193–38206, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Dual-Bias Load Balancing: Mitigating Expert-Bias Staleness in Auxiliary-Loss-Free Mixture-of-Experts (Tian et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1786.pdf