English

Triplet-Block Diffusion RWKV

Computation and Language 2026-05-26 v1

Abstract

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 B3DRWKVB^3D-RWKV, a diffusion RWKV variant that integrates the model's O(L)O(L) inference efficiency with parallel, bidirectional discrete-diffusion through a \emph{triplet-block layout} method. B3DRWKV7.2BB^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 1.6×\mathbf{1.6\times} speedup.

Cite

@article{arxiv.2605.25969,
  title  = {Triplet-Block Diffusion RWKV},
  author = {Ke Lin and Yiyang Luo and Zhaolong Su and Yunya Song and Anyi Rao},
  journal= {arXiv preprint arXiv:2605.25969},
  year   = {2026}
}