← Search

Zilei Shao

3 accepted papers

2026

Breaking the Factorization Barrier in Diffusion Language Models

ICML 2026poster

Diffusion language models theoretically allow for efficient parallel generation but are practically hindered by the "factorization barrier": the assumption that simultaneously predicted tokens are independent. This limitation forces a trade-off: models must either sacrifice speed by resolving depend…

Cited by 0SourceScholar
2026

SIMoE: A Probabilistic Framework for Cardinality-Constrained Routing in Mixture-of-Experts

ICML 2026poster

Mixture-of-Experts (MoE) models scale by activating only a small subset of experts per token, but standard deterministic top-$k$ routing is non-differentiable and trained using surrogate gradients that ignore the discrete expert selection used at inference. We introduce SIMoE routing by modeling exp…

Cited by 0SourceScholar