DiffuMamba: High-Throughput Diffusion LMs with Mamba Backbone
About
Diffusion language models (DLMs) have emerged as a promising alternative to autoregressive (AR) generation, yet their reliance on Transformer backbones limits inference efficiency due to quadratic attention or KV-cache overhead. We introduce DiffuMamba, a masked diffusion language model built on a bidirectional Mamba backbone that combines the diffusion objective with linear-time sequence modeling, and DiffuMamba-H, a hybrid variant with interleaved attention. Across scales up to 1.3B parameters, our models match Transformer-based diffusion in downstream performance while achieving up to 8.2x and 4.3x higher inference throughput, respectively, on long sequences. We further present a systematic analysis of inference efficiency across modern DLM variants combining asymptotic complexity with empirical measurements. Notably, cache-efficient block diffusion with Mamba mixers emerges as the only strategy that scales linearly with sequence length and achieves the strongest performance across all baselines, suggesting a promising direction for future diffusion-based generation systems.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multiple-choice Question Answering | MCQA | Accuracy37.32 | 29 | |
| Language Modeling | C4 en (val) | Perplexity83.73 | 28 | |
| Training Throughput | Synthetic 8192-token context (train) | Training Throughput (tok/s)2.24e+5 | 4 | |
| Language Modeling | Paloma-C4 (val) | Perplexity80.42 | 4 | |
| Inference Throughput | 700M random-initialized models | Throughput (Seq Len 256)470 | 4 |