Pre-norm Might Cause MoE Load Imbalance

SeunghyunSEO7 · x · 2026-07-10

A post summarizes insights from a poster on high-dimensional learning dynamics: in pre-norm architectures, representation collapse could harm the expert load balancing of MoE. This is because similar hidden representations are more likely to be routed to a select few experts after self-attention.

Related event: Pre-norm Architecture May Disrupt MoE Load Balancing(2 posts)→

Original post →

More from Research

Research channel →