English

Block-Based Double Decoders

Machine Learning 2026-05-20 v1 Artificial Intelligence

Abstract

Encoder-decoder models offer substantial inference-time savings over decoder-only models, but their pretraining objectives suffer from sparse supervision and dynamic sequence lengths, keeping them out of practice at scale. We propose block-based double decoders, a novel transformer architecture that utilizes doubly-causal block-based attention masks to train with full loss supervision and static sequence packing, combining decoder-only training efficiency with encoder-decoder inference efficiency. In scaling law experiments, block-based double decoders strongly outperform encoder-decoders and closely track decoder-only models across scales. At inference time, they cut KV-cache memory and per-token compute by at least 2/3 without sacrificing prefill caching or other existing inference optimizations available to decoder-only models.

Keywords

Cite

@article{arxiv.2605.18807,
  title  = {Block-Based Double Decoders},
  author = {Asher Labovich and Benjamin Bradley and Vanessa Alexander and Chaitanya Harsha},
  journal= {arXiv preprint arXiv:2605.18807},
  year   = {2026}
}

Comments

8 pages main, 13 pages total