LLM Reasoning 相关度: 7/10

Swordsman: Entropy-Driven Adaptive Block Partition for Efficient Diffusion Language Models

Yu Zhang, Xinchen Li, Jialei Zhou, Hongnan Ma, Zhongwei Wan, Yiwei Shi, Duoqian Miao, Qi Zhang, Longbing Cao
arXiv: 2602.04399v1 发布: 2026-02-04 更新: 2026-02-04

AI 摘要

Swordsman提出了一种基于熵驱动的自适应分块解码框架,提高了扩散语言模型的效率和性能。

主要贡献

  • 提出熵驱动的自适应分块解码框架Swordsman
  • 通过熵分析识别语义或句法成分边界
  • 动态调整掩码阈值,提高效率和稳定性

方法论

通过熵分析识别token边界,自适应划分block,并根据block内部掩码状态动态调整阈值,无需训练。

原文摘要

Block-wise decoding effectively improves the inference speed and quality in diffusion language models (DLMs) by combining inter-block sequential denoising and intra-block parallel unmasking. However, existing block-wise decoding methods typically partition blocks in a rigid and fixed manner, which inevitably fragments complete semantic or syntactic constituents, leading to suboptimal performance. Inspired by the entropy reduction hypothesis (ERH), we recognize that constituent boundaries offer greater opportunities for uncertainty reduction, which motivates us to employ entropy analysis for identifying constituent boundaries. Therefore, we propose Swordsman, an entropy-driven adaptive block-wise decoding framework for DLMs. Swordsman adaptively partitions blocks by identifying entropy shifts between adjacent tokens to better align with semantic or syntactic constituent boundaries. In addition, Swordsman dynamically adjusts unmasking thresholds conditioned on the real-time unmasking status within a block, further improving both efficiency and stability. As a training-free framework, supported by KV Cache, Swordsman demonstrates state-of-the-art performance across extensive evaluations.

标签

diffusion language models entropy block-wise decoding adaptive partition

arXiv 分类

cs.CL