Search papers, labs, and topics across Lattice.
Standard token-wise MoE architectures applied to video diffusion suffer from a "uniformity trap," where strict load-balancing disperses spatiotemporally coherent visual patches across disparate experts and degrades generation quality. To resolve this, the authors develop SplitMoE, which decouples the expert pool into high-level semantic experts and residual generic experts governed by prototype-guided routing and pull-push regularization. Under identical active-parameter budgets, the approach eliminates structural routing fragmentation, accelerates training convergence, and outperforms conventional load-balanced MoEs across standard video generation benchmarks.
Enforcing uniform load balancing across video tokens fundamentally breaks visual coherence, but bifurcating experts into semantic and generic tiers resolves patch fragmentation while driving emergent coarse-to-fine denoising.
Mixture-of-Experts (MoE), popularized by large language models, is a promising paradigm for scaling visual generative models. However, conventional token-wise MoE routes tokens independently within a homogeneous expert pool and regularizes expert usage toward uniformity, making it poorly matched to video data that is spatiotemporally redundant and semantically long-tailed. We show that existing visual MoEs fall into a uniformity trap: semantically under-organized routing, compounded by uniform expert-usage regularization, scatters coherent patches across disparate experts, causing routing fragmentation and structural distortion. To address this, we propose SplitMoE, a split-role sparse architecture that breaks the shackles of uniformity. To accommodate the inherent semantic imbalance, we explicitly bifurcate the expert pool into semantic experts and generic experts, with semantic experts capturing high-level semantic abstraction and generic experts preserving residual visual information and flexible generative capacity. Leveraging prototype-guided routing and pull-push regularization, SplitMoE enables tokens to cluster naturally by semantic attributes rather than arbitrary balancing constraints. Extensive results show that under an equivalent activated-parameter budget, SplitMoE outperforms traditional load-balanced MoEs in convergence speed, routing coherence, and video generation quality across standard benchmarks. By revealing an emergent coarse-to-fine denoising logic, SplitMoE provides the community with a modality-aware scaling path, serving as a critical reference for building large-scale video world models.