arXiv:2609.35817v1 Announce Type: new Abstract: Although uniform diffusion language models (UDLMs) represent a promising diffusion paradigm, scaling them remains challenging. We identify the core obstacle as an over-uniform training objective and condition-target confusion during sampling. To address these, we propose Less Uniform Diffusion (LUDI), a novel UDLM framework. Specifically, we (i) introduce a less uniform loss that directs each reverse transition toward the clean token, and (ii) equip the model with per-token time embeddings that supply token-level corruption hints, enabling confidence-based few-step sampling. Experiments across scales show that LUDI yields cleaner supervision and improves few-step generation. We further continue-train a 7B autoregressive model into LUDI-7B, resulting in a UDLM capable of complex reasoning. It achieves a 3-token-per-step speedup over AR decoding and competitive performance compared with masked diffusion baselines, revealing that the full potential of UDLMs for complex generation remains to be unlocked.
Less Uniform Discrete Diffusion is More Powerful and Scalable
Researchers introduced Less Uniform Diffusion (LUDI), a uniform diffusion language model framework that addresses an over-uniform training objective and condition-target confusion during sampling by directing each reverse transition toward the clean token and adding per-token time embeddings for confidence-based few-step sampling. The team continue-trained a 7B autoregressive model into LUDI-7B, which achieved a 3-token-per-step speedup over autoregressive decoding and competitive performance against masked diffusion baselines. The work is published as arXiv:2609.35817v1.
Run your AI side-project on zahid.host
EU-based hosting, git-push deploys, automatic HTTPS, no cold starts. Free tier with a custom domain — perfect for shipping the agent you just read about.