Triplet-Block Diffusion RWKV
About
Causal Transformer language models suffer from strictly sequential decoding and a quadratic per-step attention cost. While linear-time causal models and discrete diffusion models each address these weaknesses, their integration remains inherently inconsistent: diffusion requires bidirectional attention, while causal models are unidirectional. To unify these architectures, we propose $B^3D-RWKV$, a diffusion RWKV variant that integrates the model's $O(L)$ inference efficiency with parallel, bidirectional discrete-diffusion through a \emph{triplet-block layout} method. $B^3D-RWKV-7.2B$ reaches comparable accuracy on an 8-task suite versus existing models while significantly outperforming baselines in decoding throughput with an average of $\mathbf{1.6\times}$ speedup.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Physical Commonsense Reasoning | PIQA | Accuracy73.5 | 696 | |
| Reasoning | ARC Easy | -- | 233 | |
| Graduate-level Question Answering | GPQA | Accuracy25.6 | 215 | |
| Reasoning | ARC Challenge | Accuracy61.6 | 100 | |
| Reading Comprehension | RACE | Accuracy49.7 | 75 |