New MoE paper: EQB exact global quantile balancing beats Kimi K3's approximate method

ChengleiSi · x · 2026-09-27

A new paper tackles MoE load balancing at two scales. EQB (Exact Quantile Balancing) computes exact global-batch BF16 quantiles with negligible communication, outperforming the approximate histogram variant used by Kimi K3. LEI (Load-Error Injection) injects local load errors into router-score gradients, beating the GShard auxiliary loss at comparable quality. Validated on 7.5B MoEs trained up to 500B tokens.

Original post →

More from Research

Research channel →