Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Ke Lin, Yiyang Luo, Zhaolong Su, Yunya Song, Anyi Rao• 2026

Related benchmarks

TaskDatasetResultRank
Physical Commonsense ReasoningPIQA
Accuracy73.5
696
ReasoningARC Easy--
233
Graduate-level Question AnsweringGPQA
Accuracy25.6
215
ReasoningARC Challenge
Accuracy61.6
100
Reading ComprehensionRACE
Accuracy49.7
75
Showing 5 of 5 rows

Other info

GitHub

Follow for update