Triplet-Block Diffusion RWKV

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


Abstract
Causal Transformer language models suffer from strictly sequential decoding and a quadratic per-step attention cost, resulting in expensive inference and an inability to revise previous tokens during text generation. While linear-time causal models and discrete diffusion models each address these weaknesses, their integration remains inherently inconsistent and underdeveloped. In this work, we propose B3D-RWKV, a diffusion-based RWKV variant that integrates the model’s 𝒪(L) inference efficiency with parallel, bidirectional discrete-diffusion through a triplet-block layout method. B3D-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× speedup. Code is available at https://github.com/leonardodalinky/B3D-RWKV.
Anthology ID:
2026.findings-emnlp.940
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
19816–19829
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.940/
DOI:
Bibkey:
Cite (ACL):
Ke Lin, Yiyang Luo, Zhaolong Su, Yunya Song, and Anyi Rao. 2026. Triplet-Block Diffusion RWKV. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 19816–19829, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Triplet-Block Diffusion RWKV (Lin et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.940.pdf