arXiv:2605.08696cs.CLcs.LG2026-05

提出可切换训练并行与推理循环的结构化递归混合器,提升生成效率与吞吐。

Structured Recurrent Mixers for Massively Parallelized Sequence Generation

论文配图:Structured Recurrent Mixers for Massively Parallelized Sequence Generation
图 1 · 摘自论文原文
  • 训练时并行处理序列,推理时转为递归结构,无需特殊内核
  • 在相同计算量下,推理吞吐提升12倍,并发性提高170倍
  • 适合长序列语言建模与强化学习训练,尤其擅长批量扩展

过去二十年,语言建模从主要使用序列处理的递归架构转向训练时并行处理的非递归模型,虽提升了训练效率与稳定性,但降低了推理吞吐。本文提出结构化递归混合器(SRM),可在训练时采用并行序列表示,推理时转换为递归表示,无需专用内核或设备级内存管理。实验表明,该双表示机制显著提升训练效率、输入信息容量,并实现更大推理吞吐与并发性。我们推测,递归模型不适用于长序列扩展(针对信息密集的语言输入),但适合样本维度(批量)扩展,因其每样本内存恒定。我们提供了Mojo/MAX实现的SRM,在vLLM上相比同等能力的Transformer实现,推理吞吐提升12倍,并发性提升170倍;其PyTorch实现使计算量恒定下的GSM8k Pass@k提升30%。最后,我们证明了SRM在强化学习训练中的有效性。

原文摘要 · Abstract (English)

Over the last two decades, language modeling has experienced a shift from the use of predominantly recurrent architectures that process tokens sequentially during training and inference to non-recurrent models that process sequence elements in parallel during training, which results in greater training efficiency and stability at the expense of lower inference throughput. Here we introduce the Structured Recurrent Mixer, an architecture that allows for algebraic conversion between a sequence parallel representation at train time and a recurrent representation at inference, notably without the need for specialized kernels or device-specific memory management. We show experimentally that this dual representation allows for greater training efficiency, higher input information capacity, and larger inference throughput and concurrency when compared to other linear complexity models. We postulate that recurrent models are poorly suited to extended sequence length scaling for information-rich inputs typical of language, but are well suited to scaling in the sample (batch) dimension due to their constant memory per sample. We provide Mojo/MAX inference implementations of SRMs exhibiting 12x the throughput and 170x the concurrency of similarly powerful Transformers inferenced on vLLM, increases characteristic of Pytorch implementations resulting in a 30\% increase in compute-constant GSM8k Pass@k. We conclude by demonstrating that SRMs are effective reinforcement learning training candidates.

序列生成递归模型推理优化强化学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。