跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

LEAP:通过前视早收敛令牌检测解锁扩散语言模型并行性

文章摘要

扩散语言模型(dLLM)因其高度并行处理的潜力而备受关注。然而,现有方法依赖高置信度阈值来保证准确性,严重限制了并行扩展性。本文通过令牌级统计分析发现,大量令牌在去噪早期就已收敛到正确预测,但未达到标准置信度阈值,表明现有标准过于保守。为此,提出LEAP(前视早收敛令牌检测加速并行解码),一种无需训练、即插即用的方法,通过未来上下文过滤和多序列叠加检测早收敛令牌,实现可靠早期解码。实验表明,LEAP显著降低推理延迟和解码步数,较置信度解码平均减少约30%去噪步数,在GSM8K数据集上与dParallel结合达到每秒7.2个令牌的处理速度,同时保持模型精度。

来源arXiv Machine Learning作者: Haohui Zhang, Zhiye Wang, Xiaoying Gan, Xinbing Wang, Bo Jiang
LEAP:通过前视早收敛令牌检测解锁扩散语言模型并行性
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

扩散语言模型(Diffusion Language Models, dLLMs)因其在高度并行处理方面的潜力而受到广泛关注。与自回归模型逐步生成不同,dLLMs通过迭代去噪过程同时生成多个令牌,从而有望大幅提升解码速度。然而,现有dLLMs的并行能力依赖于条件独立假设,该假设要求模型在解码时对每个令牌具有高置信度,以确保边际分布与联合分布之间的差异可忽略不计。这种严格的置信度阈值虽然保证了生成质量,但严重限制了并行解码的扩展性——因为只有少数令牌能在早期满足高置信度条件,大部分令牌仍需逐步去噪,导致并行优势无法充分发挥。

为了解决这一问题,Haohui Zhang等研究者对令牌级动态进行了系统的统计分析。他们发现,在去噪过程中,相当大比例的令牌在早期迭代中就已经收敛到了正确的预测结果,但它们的置信度分数却远低于标准阈值。这表明现有的置信度标准过于保守,错失了提前解码的机会。基于这一洞察,团队提出了LEAP(Lookahead Early-Convergence Token Detection for Accelerated Parallel Decoding)方法。

LEAP是一种无需训练、即插即用的技术,核心思想是识别那些尽管置信度不高但实际上已经正确收敛的令牌。它通过两种机制实现:一是利用未来上下文过滤,即参考后续去噪步骤中的信息来验证当前令牌的收敛性;二是采用多序列叠加,通过比较多个独立去噪序列中令牌的一致性来增强检测可靠性。LEAP首先检测出早收敛令牌,然后允许它们提前解码,而无需等待所有令牌达到高置信度。实验验证表明,早收敛令牌与最终正确令牌之间具有高度一致性,保证了早期解码的可靠性。

在多个领域的基准测试中,LEAP展示了显著的性能提升。与传统的置信度基准解相比,使用LEAP后平均去噪步数减少了约30%。在数学推理数据集GSM8K上,将LEAP与现有的dParallel方法结合,解码速度提升至每秒7.2个令牌,同时保持了模型精度。这一成果不仅直接降低了推理延迟,还挑战了“高置信度是保证准确性的必要条件”这一传统观念,为并行解码开辟了新范式。LEAP的研究表明,通过更精细的令牌级分析和上下文感知的检测策略,可以更早地释放并行潜力,而无需牺牲生成质量。未来,该工作有望在实时应用和大规模部署中发挥重要作用。

展开要点与分析

文章情报

工程师进阶

要点

  • 发现大量令牌在去噪早期正确收敛但未达高置信度阈值,现有置信度标准过于保守
  • 提出LEAP方法,无需训练,通过未来上下文过滤和多序列叠加检测早收敛令牌
  • LEAP减少约30%去噪步数,与dParallel结合在GSM8K上达到7.2令牌/步
  • 打破对高置信度先验的依赖,为并行解码提供新范式

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。