Bifocal Diffusion Language Models: Asymmetric Bidirectional Context for Parallel Generation
About
Discrete diffusion language models (dLLMs) recover masked tokens in parallel, offering significant speedups over autoregressive (AR) generation. However, such promising frameworks face a fundamental architectural design dilemma: \ding{182} Adopting bidirectional attention achieves strong generation quality by allowing each position to access the full context, but is inherently incompatible with KV caching, limiting inference throughput in batch-serving scenarios; \ding{183} Conversely, causal attention enables efficient cached inference but loses all right-side context, substantially degrading generation quality. This paper introduces Bifocal dLLMs, a new paradigm that resolves this dilemma through \emph{asymmetric bidirectional context}. Analogous to bifocal lenses, we instantiate the paradigm as \textbf{R2LM} (Right-to-Left Mamba), which combines two complementary mechanisms: $a$) standard causal attention providing precise left-context with full KV cache compatibility, while $b$) a lightweight reverse Mamba SSM sidecar supplying compressed right-side context without breaking cacheability. Comprehensive experiments on continued pretraining of Qwen3-1.7B with 60B tokens demonstrate that R2LM achieves $2.4\times$ to $12.9\times$ higher throughput than bidirectional dLLMs and $1.9\times$ to $2.9\times$ speedup over AR baselines in batch serving through parallel decoding with KV caching, while exceeding the causal baseline on most benchmarks and surpassing the bidirectional dLLM on average.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multiple-choice Question Answering | MMLU | Accuracy37 | 222 | |
| Multiple-choice Question Answering | HellaSwag | Accuracy43 | 212 | |
| Multiple-choice Question Answering | PIQA | Accuracy67.2 | 67 | |
| Multiple-choice Question Answering | WinoG | Accuracy60.2 | 52 | |
| Multiple-choice Question Answering | BoolQ | MC Accuracy0.673 | 50 | |
| Multiple-choice Question Answering | OpenBookQA | Accuracy27.4 | 30 | |
| Multiple-choice Question Answering | ARC-C | Accuracy40.9 | 22 |