arXiv:2606.08411cs.CL2026-06

让语言模型生成更快:异步解码不等完再继续

AsyncLane: Decoupling Refinement from Advancement in Diffusion Language Model Decoding

论文配图:AsyncLane: Decoupling Refinement from Advancement in Diffusion Language Model Decoding
图 1 · 摘自论文原文
  • 发现分块生成中可提前开启新块,无需等前块完全完成
  • 实验显示最高提速3.04倍,长文本生成时优势更明显
  • 无需重训练,直接替换现有模型即可使用

块式半自回归解码是扩散语言模型的标准推理范式,但要求块间严格依赖:下一区块无法启动,直到当前区块完全解码或去噪预算耗尽。我们观察到,一旦某区块出现可靠分隔符边界或稳定语义前缀,后续生成无需等待所有残余词元解析完成。为此提出AsyncLane,一种无需训练的解码调度器,将精炼与推进解耦。在检测到分隔符处,系统分叉出一个精炼车道和一个继续生成车道:前缀仍可编辑,而后续生成可提前推进,无需等待前缀精炼结束。生成路径以车道树形式记录依赖关系与输出顺序,执行在活跃车道集合上并行进行。为提升双向注意力下的效率,AsyncLane结合共享前缀车道批处理、前瞻草稿复用、级联终止机制及紧凑缓存刷新与刷新逻辑复用,避免调用成本随车道数线性增长。AsyncLane可作为现成替代品直接部署于块式扩散语言模型采样器,无需重新训练。数学推理与代码生成实验表明,AsyncLane在保持竞争力质量的同时持续提升吞吐量。在LLaDA与Dream骨干网络上,其在所有评估长度设置下均达到最高每秒生成词数(TPS);相较最快基线,最大提速达2.95倍(LLaDA)与3.04倍(Dream),尤其在较长生成预算下表现突出。

原文摘要 · Abstract (English)

Block-wise semi-autoregressive decoding is the standard inference paradigm for diffusion large language models (DLMs), but it imposes a strict dependency between blocks: the next block cannot begin until the current block is fully decoded or its denoising budget is exhausted. We observe that once a block exposes a reliable delimiter boundary or stable semantic prefix, continuation generation need not wait for every residual token to be resolved. We propose AsyncLane, a training-free decoding scheduler that decouples refinement from advancement. AsyncLane forks a generate lane at observed delimiter boundaries into a refine lane and a continuation generate lane: the prefix remains editable, while the continuation advances before prefix refinement finishes. The resulting lane tree records decoding dependencies and output order, while execution proceeds over the active lane set. To make this asynchronous schedule efficient under bidirectional attention, AsyncLane combines shared-prefix lane batching, lookahead draft reuse, cascading termination, and compact cache refresh with refresh-logit reuse, preventing model-call cost from scaling directly with the number of lanes. AsyncLane is a drop-in replacement for block-wise DLM samplers and requires no retraining. Experiments on mathematical reasoning and code generation show that AsyncLane consistently improves throughput while maintaining competitive quality. Across LLaDA and Dream backbones, AsyncLane achieves the highest TPS in all evaluated benchmark-length settings; relative to the fastest competing baseline, it reaches peak speedups of 2.95x on LLaDA and 3.04x on Dream, with especially large gains under longer generation budgets.

扩散模型生成加速异步解码

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