# E-MoE: Enhanced Mixture-of-Experts for Non-Factorized Diffusion Language Models

> Source: <https://www.machinebrief.com/news/e-moe-enhanced-mixture-of-experts-for-non-factorized-diffusi-j1un>
> Published: 2026-09-30 04:00:00+00:00

arXiv:2609.37533v1 Announce Type: new 
Abstract: Masked diffusion models (MDMs) generate sequences by progressively unmasking several tokens per denoising step, but their reverse process is typically factorized over positions, limiting sample quality in the few-step regime where diffusion's speed advantage over autoregressive decoding matters most. A recent line of work introduces a continuous Gaussian latent, trained as a variational autoencoder, to capture correlations across positions, but such approaches are prone to posterior collapse, where the latent is silently ignored. We propose Enhanced Mixture-of-Experts (E-MoE), which builds the reverse process as a mixture of factorized distributions over a discrete shared latent given by the expert-routing decisions of a Mixture-of-Experts (MoE) backbone, without increasing active parameters over the factorized baseline. Across synthetic multi-modal benchmarks, binarized MNIST, and LM1B, E-MoE improves few-step generation over factorized baselines.