Search papers, labs, and topics across Lattice.
The paper investigates the underperformance of continuous diffusion language models (DLMs) compared to discrete DLMs, identifying token rounding as a key bottleneck. To address this, they introduce CoDAR, a two-stage framework that maintains continuous diffusion in the embedding space while employing a context-conditional autoregressive Transformer decoder for improved token rounding. Experiments on LM1B and OpenWebText show that CoDAR achieves generation quality comparable to strong discrete DLMs and provides a mechanism for controlling the fluency-diversity trade-off.
Continuous diffusion LMs can rival discrete models by fixing the token-rounding bottleneck with a contextual autoregressive decoder, unlocking a fluency-diversity knob in the process.
We study why continuous diffusion language models (DLMs) have lagged behind discrete diffusion approaches despite their appealing continuous generative dynamics. Under a controlled token--recovery study, we identify token rounding, the final projection from denoised embeddings to tokens, as a primary bottleneck. Building on these insights, we propose CoDAR (Continuous Diffusion with Contextual AutoRegressive Decoder), a two--stage framework that keeps diffusion entirely continuous in an embedding space while learning a strong, context--conditional discretizer: an autoregressive Transformer decoder that cross--attends to the denoised embedding sequence and performs contextualized rounding to tokens. Experiments on LM1B and OpenWebText demonstrate that CoDAR substantially improves generation quality over latent diffusion and becomes competitive with strong discrete DLMs, while exposing a simple decoder--temperature knob to navigate the fluency--diversity trade off.
